322394fe63
* feat: use go1.16 embedding for resources We want to embed everything that can be easily embedded. We should, at a minimum, replace the downloading of resources and bindata. Ref: https://github.com/ooni/probe/issues/1367. * fix: get rid of bindata and use go embed instead * fix: start unbreaking some automatic tests * fix: fetch resources as part of the mobile build * fix: convert more stuff to go1.16 I still expect many breakages, but we'll fix them. * fix: make the windows CI green * fix: get resources before running QA * fix: go1.16 uses modules by default * hopefully fix all other outstanding issues * fix(QA/telegram.py): add another DC IP address * Apply suggestions from code review
96 lines
2.0 KiB
Go
96 lines
2.0 KiB
Go
package geolocate
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ooni/probe-cli/v3/internal/engine/resourcesmanager"
|
|
)
|
|
|
|
const (
|
|
asnDBPath = "../testdata/asn.mmdb"
|
|
countryDBPath = "../testdata/country.mmdb"
|
|
ipAddr = "35.204.49.125"
|
|
)
|
|
|
|
func maybeFetchResources(t *testing.T) {
|
|
c := &resourcesmanager.CopyWorker{DestDir: "../testdata/"}
|
|
if err := c.Ensure(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestLookupASN(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
asn, org, err := LookupASN(asnDBPath, ipAddr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if asn <= 0 {
|
|
t.Fatal("unexpected ASN value")
|
|
}
|
|
if len(org) <= 0 {
|
|
t.Fatal("unexpected org value")
|
|
}
|
|
}
|
|
|
|
func TestLookupASNInvalidFile(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
asn, org, err := LookupASN("/nonexistent", ipAddr)
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if asn != DefaultProbeASN {
|
|
t.Fatal("expected a zero ASN")
|
|
}
|
|
if org != DefaultProbeNetworkName {
|
|
t.Fatal("expected an empty org")
|
|
}
|
|
}
|
|
|
|
func TestLookupASNInvalidIP(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
asn, org, err := LookupASN(asnDBPath, "xxx")
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if asn != DefaultProbeASN {
|
|
t.Fatal("expected a zero ASN")
|
|
}
|
|
if org != DefaultProbeNetworkName {
|
|
t.Fatal("expected an empty org")
|
|
}
|
|
}
|
|
|
|
func TestLookupCC(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
cc, err := (mmdbLookupper{}).LookupCC(countryDBPath, ipAddr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(cc) != 2 {
|
|
t.Fatal("does not seem a country code")
|
|
}
|
|
}
|
|
|
|
func TestLookupCCInvalidFile(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
cc, err := (mmdbLookupper{}).LookupCC("/nonexistent", ipAddr)
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if cc != DefaultProbeCC {
|
|
t.Fatal("expected an empty cc")
|
|
}
|
|
}
|
|
|
|
func TestLookupCCInvalidIP(t *testing.T) {
|
|
maybeFetchResources(t)
|
|
cc, err := (mmdbLookupper{}).LookupCC(asnDBPath, "xxx")
|
|
if err == nil {
|
|
t.Fatal("expected an error here")
|
|
}
|
|
if cc != DefaultProbeCC {
|
|
t.Fatal("expected an empty cc")
|
|
}
|
|
}
|