whatismyip/router/setup_test.go

52 lines
1.4 KiB
Go

package router
import (
"os"
"testing"
"github.com/dcarrillo/whatismyip/models"
"github.com/gin-gonic/gin"
)
type testIPs struct {
ipv4 string
ipv4ASN string
ipv6 string
ipv6ASN string
}
type contentTypes struct {
html string
text string
json string
}
var (
app *gin.Engine
testIP = testIPs{
ipv4: "81.2.69.192",
ipv4ASN: "82.99.17.64",
ipv6: "2a02:9000::1",
ipv6ASN: "2a02:a800::1",
}
contentType = contentTypes{
html: "content-type: text/html; charset=utf-8",
text: "text/plain; charset=utf-8",
json: "application/json; charset=utf-8",
}
jsonIPv4 = `{"client_port":"1000","ip":"81.2.69.192","ip_version":4,"country":"United Kingdom","country_code":"GB","city":"London","latitude":51.5142,"longitude":-0.0931,"postal_code":"","time_zone":"Europe/London","asn":0,"asn_organization":"","host":"test","headers":{"X-Real-Ip":["81.2.69.192"]}}`
jsonIPv6 = `{"asn":3352, "asn_organization":"TELEFONICA DE ESPANA", "city":"", "client_port":"1000", "country":"", "country_code":"", "headers":{"X-Real-Ip":["2a02:9000::1"]}, "host":"test", "ip":"2a02:9000::1", "ip_version":6, "latitude":0, "longitude":0, "postal_code":"", "time_zone":""}`
)
const trustedHeader = "X-Real-IP"
func TestMain(m *testing.M) {
app = gin.Default()
app.TrustedPlatform = trustedHeader
models.Setup("../test/GeoIP2-City-Test.mmdb", "../test/GeoLite2-ASN-Test.mmdb")
Setup(app)
defer models.CloseDBs()
os.Exit(m.Run())
}