53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
|
package app
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestIPRegionResolverLookupSpecialAddresses(t *testing.T) {
|
||
|
|
resolver := &IPRegionResolver{}
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
ip string
|
||
|
|
want string
|
||
|
|
}{
|
||
|
|
{name: "ipv4 loopback", ip: "127.0.0.1", want: "本机"},
|
||
|
|
{name: "ipv6 loopback", ip: "::1", want: "本机"},
|
||
|
|
{name: "private network", ip: "192.168.1.10", want: "内网IP"},
|
||
|
|
{name: "host port", ip: "10.0.0.8:443", want: "内网IP"},
|
||
|
|
{name: "invalid", ip: "not-an-ip", want: ""},
|
||
|
|
{name: "public disabled", ip: "8.8.8.8", want: ""},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
if got := resolver.Lookup(tt.ip); got != tt.want {
|
||
|
|
t.Fatalf("Lookup(%q) = %q, want %q", tt.ip, got, tt.want)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIPRegionResolverLookupFromXDB(t *testing.T) {
|
||
|
|
resolver, err := NewIPRegionResolver("", "", nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("NewIPRegionResolver() error = %v", err)
|
||
|
|
}
|
||
|
|
defer resolver.Close()
|
||
|
|
|
||
|
|
if got := resolver.Lookup("8.8.8.8"); got == "" {
|
||
|
|
t.Fatal("Lookup(8.8.8.8) returned empty region")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIPRegionResolverExternalPathFallsBackToEmbedWhenMissing(t *testing.T) {
|
||
|
|
resolver, err := NewIPRegionResolver("/path/not/exist/ip2region_v4.xdb", "/path/not/exist/ip2region_v6.xdb", nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("NewIPRegionResolver() error = %v", err)
|
||
|
|
}
|
||
|
|
defer resolver.Close()
|
||
|
|
|
||
|
|
if got := resolver.Lookup("8.8.8.8"); got == "" {
|
||
|
|
t.Fatal("Lookup(8.8.8.8) returned empty region")
|
||
|
|
}
|
||
|
|
}
|