mirror of
https://github.com/tobychui/zoraxy.git
synced 2025-08-05 12:56:48 +02:00
73ab9ca778
+ Added unset subdomain custom redirection feature #46 + Optimized memory usage by space time tradeoff in geoip lookup to fix #52 + Replaced all stori/go.uuid to google/uuid for security reasons #55
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package geodb_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"imuslab.com/zoraxy/mod/geodb"
|
|
)
|
|
|
|
/*
|
|
func TestTrieConstruct(t *testing.T) {
|
|
tt := geodb.NewTrie()
|
|
data := [][]string{
|
|
{"1.0.16.0", "1.0.31.255", "JP"},
|
|
{"1.0.32.0", "1.0.63.255", "CN"},
|
|
{"1.0.64.0", "1.0.127.255", "JP"},
|
|
{"1.0.128.0", "1.0.255.255", "TH"},
|
|
{"1.1.0.0", "1.1.0.255", "CN"},
|
|
{"1.1.1.0", "1.1.1.255", "AU"},
|
|
{"1.1.2.0", "1.1.63.255", "CN"},
|
|
{"1.1.64.0", "1.1.127.255", "JP"},
|
|
{"1.1.128.0", "1.1.255.255", "TH"},
|
|
{"1.2.0.0", "1.2.2.255", "CN"},
|
|
{"1.2.3.0", "1.2.3.255", "AU"},
|
|
}
|
|
|
|
for _, entry := range data {
|
|
startIp := entry[0]
|
|
endIp := entry[1]
|
|
cc := entry[2]
|
|
tt.Insert(startIp, cc)
|
|
tt.Insert(endIp, cc)
|
|
}
|
|
|
|
t.Log(tt.Search("1.0.16.20"), "== JP") //JP
|
|
t.Log(tt.Search("1.2.0.122"), "== CN") //CN
|
|
t.Log(tt.Search("1.2.1.0"), "== CN") //CN
|
|
t.Log(tt.Search("1.0.65.243"), "== JP") //JP
|
|
t.Log(tt.Search("1.0.62.243"), "== CN") //CN
|
|
}
|
|
*/
|
|
|
|
func TestResolveCountryCodeFromIP(t *testing.T) {
|
|
// Create a new store
|
|
store, err := geodb.NewGeoDb(nil, &geodb.StoreOptions{
|
|
false,
|
|
false,
|
|
})
|
|
if err != nil {
|
|
t.Errorf("error creating store: %v", err)
|
|
return
|
|
}
|
|
|
|
// Test an IP address that should return a valid country code
|
|
ip := "8.8.8.8"
|
|
expected := "US"
|
|
info, err := store.ResolveCountryCodeFromIP(ip)
|
|
if err != nil {
|
|
t.Errorf("error resolving country code for IP %s: %v", ip, err)
|
|
return
|
|
}
|
|
if info.CountryIsoCode != expected {
|
|
t.Errorf("expected country code %s, but got %s for IP %s", expected, info.CountryIsoCode, ip)
|
|
}
|
|
|
|
// Test an IP address that should return an empty country code
|
|
ip = "127.0.0.1"
|
|
expected = ""
|
|
info, err = store.ResolveCountryCodeFromIP(ip)
|
|
if err != nil {
|
|
t.Errorf("error resolving country code for IP %s: %v", ip, err)
|
|
return
|
|
}
|
|
if info.CountryIsoCode != expected {
|
|
t.Errorf("expected country code %s, but got %s for IP %s", expected, info.CountryIsoCode, ip)
|
|
}
|
|
}
|