31e478b04e
* fix(pkg.go.dev): import a subpackage containing the assets We're trying to fix this issue that pkg.go.dev does not build. Thanks to @hellais for this very neat idea! Let's keep our fingers crossed and see whether it fixes! * feat: use embedded geoip databases Closes https://github.com/ooni/probe/issues/1372. Work done as part of https://github.com/ooni/probe/issues/1369. * fix(assetsx): add tests * feat: simplify and just vendor uncompressed DBs * remove tests that seems not necessary anymore * fix: run go mod tidy * Address https://github.com/ooni/probe-cli/pull/260/files#r605181364 * rewrite a test in a better way * fix: gently cleanup the legacy assetsdir Do not remove the whole directory with brute force. Just zap the files whose name we know. Then attempt to delete the legacy directory as well. If not empty, just fail. This is fine because it means the user has stored other files inside the directory. * fix: create .miniooni if missing
330 lines
8.7 KiB
Go
330 lines
8.7 KiB
Go
package geolocate
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
type taskProbeIPLookupper struct {
|
|
ip string
|
|
err error
|
|
}
|
|
|
|
func (c taskProbeIPLookupper) LookupProbeIP(ctx context.Context) (string, error) {
|
|
return c.ip, c.err
|
|
}
|
|
|
|
func TestLocationLookupCannotLookupProbeIP(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{err: expected},
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if !errors.Is(err, expected) {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != DefaultProbeASN {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != DefaultProbeCC {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != DefaultProbeNetworkName {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != DefaultProbeIP {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatal("invalid ResolverASN value")
|
|
}
|
|
if out.ResolverIP != DefaultResolverIP {
|
|
t.Fatal("invalid ResolverIP value")
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
type taskASNLookupper struct {
|
|
err error
|
|
asn uint
|
|
name string
|
|
}
|
|
|
|
func (c taskASNLookupper) LookupASN(ip string) (uint, string, error) {
|
|
return c.asn, c.name, c.err
|
|
}
|
|
|
|
func TestLocationLookupCannotLookupProbeASN(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{err: expected},
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if !errors.Is(err, expected) {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != DefaultProbeASN {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != DefaultProbeCC {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != DefaultProbeNetworkName {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatal("invalid ResolverASN value")
|
|
}
|
|
if out.ResolverIP != DefaultResolverIP {
|
|
t.Fatal("invalid ResolverIP value")
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
type taskCCLookupper struct {
|
|
err error
|
|
cc string
|
|
}
|
|
|
|
func (c taskCCLookupper) LookupCC(ip string) (string, error) {
|
|
return c.cc, c.err
|
|
}
|
|
|
|
func TestLocationLookupCannotLookupProbeCC(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{asn: 1234, name: "1234.com"},
|
|
countryLookupper: taskCCLookupper{cc: "US", err: expected},
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if !errors.Is(err, expected) {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != 1234 {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != DefaultProbeCC {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != "1234.com" {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatal("invalid ResolverASN value")
|
|
}
|
|
if out.ResolverIP != DefaultResolverIP {
|
|
t.Fatal("invalid ResolverIP value")
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
type taskResolverIPLookupper struct {
|
|
ip string
|
|
err error
|
|
}
|
|
|
|
func (c taskResolverIPLookupper) LookupResolverIP(ctx context.Context) (string, error) {
|
|
return c.ip, c.err
|
|
}
|
|
|
|
func TestLocationLookupCannotLookupResolverIP(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{asn: 1234, name: "1234.com"},
|
|
countryLookupper: taskCCLookupper{cc: "IT"},
|
|
resolverIPLookupper: taskResolverIPLookupper{err: expected},
|
|
enableResolverLookup: true,
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if err != nil {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != 1234 {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != "IT" {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != "1234.com" {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.DidResolverLookup != true {
|
|
t.Fatal("invalid DidResolverLookup value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatal("invalid ResolverASN value")
|
|
}
|
|
if out.ResolverIP != DefaultResolverIP {
|
|
t.Fatal("invalid ResolverIP value")
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
func TestLocationLookupCannotLookupResolverNetworkName(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{asn: 1234, name: "1234.com"},
|
|
countryLookupper: taskCCLookupper{cc: "IT"},
|
|
resolverIPLookupper: taskResolverIPLookupper{ip: "4.3.2.1"},
|
|
resolverASNLookupper: taskASNLookupper{err: expected},
|
|
enableResolverLookup: true,
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if err != nil {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != 1234 {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != "IT" {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != "1234.com" {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.DidResolverLookup != true {
|
|
t.Fatal("invalid DidResolverLookup value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatalf("invalid ResolverASN value: %+v", out.ResolverASN)
|
|
}
|
|
if out.ResolverIP != "4.3.2.1" {
|
|
t.Fatalf("invalid ResolverIP value: %+v", out.ResolverIP)
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
func TestLocationLookupSuccessWithResolverLookup(t *testing.T) {
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{asn: 1234, name: "1234.com"},
|
|
countryLookupper: taskCCLookupper{cc: "IT"},
|
|
resolverIPLookupper: taskResolverIPLookupper{ip: "4.3.2.1"},
|
|
resolverASNLookupper: taskASNLookupper{asn: 4321, name: "4321.com"},
|
|
enableResolverLookup: true,
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if err != nil {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != 1234 {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != "IT" {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != "1234.com" {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.DidResolverLookup != true {
|
|
t.Fatal("invalid DidResolverLookup value")
|
|
}
|
|
if out.ResolverASN != 4321 {
|
|
t.Fatalf("invalid ResolverASN value: %+v", out.ResolverASN)
|
|
}
|
|
if out.ResolverIP != "4.3.2.1" {
|
|
t.Fatalf("invalid ResolverIP value: %+v", out.ResolverIP)
|
|
}
|
|
if out.ResolverNetworkName != "4321.com" {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
func TestLocationLookupSuccessWithoutResolverLookup(t *testing.T) {
|
|
op := Task{
|
|
probeIPLookupper: taskProbeIPLookupper{ip: "1.2.3.4"},
|
|
probeASNLookupper: taskASNLookupper{asn: 1234, name: "1234.com"},
|
|
countryLookupper: taskCCLookupper{cc: "IT"},
|
|
resolverIPLookupper: taskResolverIPLookupper{ip: "4.3.2.1"},
|
|
resolverASNLookupper: taskASNLookupper{asn: 4321, name: "4321.com"},
|
|
}
|
|
ctx := context.Background()
|
|
out, err := op.Run(ctx)
|
|
if err != nil {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
if out.ASN != 1234 {
|
|
t.Fatal("invalid ASN value")
|
|
}
|
|
if out.CountryCode != "IT" {
|
|
t.Fatal("invalid CountryCode value")
|
|
}
|
|
if out.NetworkName != "1234.com" {
|
|
t.Fatal("invalid NetworkName value")
|
|
}
|
|
if out.ProbeIP != "1.2.3.4" {
|
|
t.Fatal("invalid ProbeIP value")
|
|
}
|
|
if out.DidResolverLookup != false {
|
|
t.Fatal("invalid DidResolverLookup value")
|
|
}
|
|
if out.ResolverASN != DefaultResolverASN {
|
|
t.Fatalf("invalid ResolverASN value: %+v", out.ResolverASN)
|
|
}
|
|
if out.ResolverIP != DefaultResolverIP {
|
|
t.Fatalf("invalid ResolverIP value: %+v", out.ResolverIP)
|
|
}
|
|
if out.ResolverNetworkName != DefaultResolverNetworkName {
|
|
t.Fatal("invalid ResolverNetworkName value")
|
|
}
|
|
}
|
|
|
|
func TestSmoke(t *testing.T) {
|
|
config := Config{
|
|
EnableResolverLookup: true,
|
|
}
|
|
task := Must(NewTask(config))
|
|
result, err := task.Run(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if result == nil {
|
|
t.Fatal("expected non nil result")
|
|
}
|
|
// we already checked above that the returned
|
|
// value is okay for all codepaths.
|
|
}
|
|
|
|
func TestASNStringWorks(t *testing.T) {
|
|
r := Results{ASN: 1234}
|
|
if r.ASNString() != "AS1234" {
|
|
t.Fatal("unexpected result")
|
|
}
|
|
}
|