mirror of
https://github.com/dcarrillo/whatismyip.git
synced 2026-07-23 21:35:47 +00:00
dcarrillo
2e32a20f60
* refactor: export Settings type, Setup returns value * refactor: resolver.Setup takes explicit Settings struct * refactor: GetHeadersWithoutTrustedHeaders takes explicit header params * refactor: server constructors take narrow config * refactor: Router struct with handler methods, remove geoSvc global * refactor: wire DI through main, remove setting.App references * refactor: remove App global, use returned Settings * refactor: update router tests for DI * chore: fix lint issues — rename ServerTimeouts to Timeouts, fix shadowed variables
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package router
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/dcarrillo/whatismyip/service"
|
|
"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":"1001","ip":"81.2.69.192","ip_version":4,"country":"United Kingdom","country_code":"GB","city":"London","latitude":51.5142,"longitude":-0.0931,"time_zone":"Europe/London","host":"test", "headers": {}}`
|
|
jsonIPv6 = `{"asn":3352,"asn_organization":"TELEFONICA DE ESPANA","client_port":"1001","host":"test","ip":"2a02:9000::1","ip_version":6,"headers": {}}`
|
|
jsonDNSIPv4 = `{"dns":{"ip":"81.2.69.192","country":"United Kingdom"}}`
|
|
plainDNSIPv4 = "81.2.69.192 (United Kingdom / )\n"
|
|
)
|
|
|
|
const (
|
|
trustedHeader = "X-Real-IP"
|
|
trustedPortHeader = "X-Real-Port"
|
|
domain = "dns.example.com"
|
|
)
|
|
|
|
var rt *Router
|
|
|
|
func TestMain(m *testing.M) {
|
|
app = gin.Default()
|
|
app.TrustedPlatform = trustedHeader
|
|
svc, _ := service.NewGeo(context.Background(), "../test/GeoIP2-City-Test.mmdb", "../test/GeoLite2-ASN-Test.mmdb")
|
|
rt = NewRouter(svc, trustedHeader, "", "", false)
|
|
Setup(app, rt)
|
|
|
|
os.Exit(m.Run())
|
|
}
|