ooni-probe-cli/internal/geoipx/geoipx_test.go

package geoipx

import (
	"testing"

	"github.com/ooni/probe-cli/v3/internal/model"
)

const ipAddr = "8.8.8.8"

func TestLookupASN(t *testing.T) {
	t.Run("with valid IP address", func(t *testing.T) {
		asn, org, err := LookupASN(ipAddr)
		if err != nil {
			t.Fatal(err)
		}
		if asn != 15169 {
			t.Fatal("unexpected ASN value", asn)
		}
		if org != "Google LLC" {
			t.Fatal("unexpected org value", org)
		}
	})

	t.Run("with invalid IP address", func(t *testing.T) {
		asn, org, err := LookupASN("xxx")
		if err == nil {
			t.Fatal("expected an error here")
		}
		if asn != model.DefaultProbeASN {
			t.Fatal("expected a zero ASN")
		}
		if org != model.DefaultProbeNetworkName {
			t.Fatal("expected an empty org")
		}
	})
}

func TestLookupCC(t *testing.T) {
	t.Run("with valid IP address", func(t *testing.T) {
		cc, err := LookupCC(ipAddr)
		if err != nil {
			t.Fatal(err)
		}
		if cc != "US" {
			t.Fatal("invalid country code", cc)
		}
	})

	t.Run("with invalid IP address", func(t *testing.T) {
		cc, err := LookupCC("xxx")
		if err == nil {
			t.Fatal("expected an error here")
		}
		if cc != model.DefaultProbeCC {
			t.Fatal("expected an empty cc")
		}
	})
}