mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 14:26:36 +08:00
ryan
aa4faddade
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"golint_vetx_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"vitest_failed":0,"vitest_total":116,"measure_s":81}
103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package geoip
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
type fakeProvider struct {
|
|
calls int
|
|
}
|
|
|
|
func (f *fakeProvider) Name() string {
|
|
return "fake"
|
|
}
|
|
|
|
func (f *fakeProvider) GetGeoInfo(ip net.IP) (*GeoInfo, error) {
|
|
f.calls++
|
|
return &GeoInfo{
|
|
ISOCode: "CN",
|
|
Name: "China",
|
|
}, nil
|
|
}
|
|
|
|
func (f *fakeProvider) UpdateDatabase() error {
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeProvider) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func TestGetGeoInfoCachesByProviderAndIP(t *testing.T) {
|
|
originalProvider := CurrentProvider
|
|
geoCache.Flush()
|
|
fake := &fakeProvider{}
|
|
CurrentProvider = fake
|
|
defer func() {
|
|
CurrentProvider = originalProvider
|
|
}()
|
|
|
|
ip := net.ParseIP("8.8.8.8")
|
|
record, err := GetGeoInfo(ip)
|
|
if err != nil {
|
|
t.Fatalf("expected nil error, got %v", err)
|
|
}
|
|
if record == nil || record.ISOCode != "CN" {
|
|
t.Fatalf("expected cached record, got %#v", record)
|
|
}
|
|
|
|
_, err = GetGeoInfo(ip)
|
|
if err != nil {
|
|
t.Fatalf("expected nil error on second call, got %v", err)
|
|
}
|
|
if fake.calls != 1 {
|
|
t.Fatalf("expected provider to be called once, got %d", fake.calls)
|
|
}
|
|
}
|
|
|
|
func TestUnicodeEmoji(t *testing.T) {
|
|
emoji := GetRegionUnicodeEmoji("CN")
|
|
if emoji != "🇨🇳" {
|
|
t.Errorf("expected emoji for CN, got %s", emoji)
|
|
}
|
|
}
|
|
|
|
func TestIsValidProvider(t *testing.T) {
|
|
cases := map[string]bool{
|
|
"disabled": true,
|
|
"mmdb": true,
|
|
"ip-api": true,
|
|
"geojs": true,
|
|
"ipinfo": true,
|
|
"unknown": false,
|
|
}
|
|
|
|
for provider, want := range cases {
|
|
if got := IsValidProvider(provider); got != want {
|
|
t.Fatalf("provider %s validity mismatch: want %v, got %v", provider, want, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLookupGeoInfoWithProviderUsesTemporaryProvider(t *testing.T) {
|
|
previousFactory := providerFactory
|
|
providerFactory = func(provider string) (Service, error) {
|
|
return &fakeProvider{}, nil
|
|
}
|
|
defer func() {
|
|
providerFactory = previousFactory
|
|
}()
|
|
|
|
info, err := LookupGeoInfoWithProvider("ipinfo", net.ParseIP("8.8.8.8"))
|
|
if err != nil {
|
|
t.Fatalf("expected lookup to succeed, got %v", err)
|
|
}
|
|
if info == nil || info.ISOCode != "CN" || info.Name != "China" {
|
|
t.Fatalf("unexpected geo info: %#v", info)
|
|
}
|
|
}
|