60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
|
// Copyright (C) 2021 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information
|
||
|
|
|
||
|
|
package geoip_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"storj.io/common/storj/location"
|
||
|
|
"storj.io/storj/satellite/geoip"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEmptyIP2CountryMock(t *testing.T) {
|
||
|
|
ipLookup := geoip.MockIPToCountry{}
|
||
|
|
{
|
||
|
|
co, err := ipLookup.LookupISOCountryCode("127.0.0.1")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, location.CountryCode(0), co)
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
co, err := ipLookup.LookupISOCountryCode("[2001:0db8:85a3:0000:0000:8a2e:0370:7334]")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, location.CountryCode(0), co)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIP2CountryMock(t *testing.T) {
|
||
|
|
ipLookup := geoip.MockIPToCountry{location.UnitedStates, location.Germany, location.France}
|
||
|
|
|
||
|
|
cases := []struct {
|
||
|
|
name string
|
||
|
|
address string
|
||
|
|
country location.CountryCode
|
||
|
|
errExpected bool
|
||
|
|
}{
|
||
|
|
{"first IP in the pool", "127.0.0.1:1234", location.Germany, false},
|
||
|
|
{"second IP in the pool", "127.0.0.2:1234", location.France, false},
|
||
|
|
{"third IP in the pool", "127.0.0.3:1234", location.UnitedStates, false},
|
||
|
|
{"ipv6", "[2001:0db8:85a3:0000:0000:8a2e:0370:7334]:1234", location.Germany, false},
|
||
|
|
{"not an ip address", "not at all", location.CountryCode(0), true},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tc := range cases {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
co, err := ipLookup.LookupISOCountryCode(tc.address)
|
||
|
|
if tc.errExpected {
|
||
|
|
assert.Error(t, err)
|
||
|
|
} else {
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, tc.country, co)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|