iterator_test.go

package resolver

import (
	"context"
	"testing"

	"github.com/miekg/dns"

	"darvaza.org/resolver/pkg/exdns"
)

func TestRootLookup(t *testing.T) {
	root, err := NewRootLookuper("")
	if err != nil {
		t.Fatal(err)
	}
	root.DisableAAAA() // for github sake

	// Simple
	testRootTypeA(t, root, "karasz.im", "95.216.149.141")
	// Complex
	testRootTypeA(t, root, "fda.my.salesforce.com", "")
	// EDU
	testRootTypeA(t, root, "www.seas.upenn.edu", "")
}

func TestRootLookupFrom(t *testing.T) {
	root, err := NewRootLookuper(roots["a.root-servers.net"])
	if err != nil {
		t.Fatal(err)
	}
	root.DisableAAAA() // for github sake

	// Simple
	testRootTypeA(t, root, "karasz.im", "95.216.149.141")
}

func testRootTypeA(t *testing.T, h Lookuper, name, address string) {
	z, err := h.Lookup(context.TODO(), name, dns.TypeA)
	if err != nil {
		t.Errorf("%s: %s", name, err.Error())
		return
	}

	rr := exdns.GetFirstAnswer[*dns.A](z)
	if rr == nil {
		if address != "" {
			t.Errorf("%s: no answer (expected %s)", name, address)
		} else {
			t.Errorf("%s: no answer", name)
		}
		return
	}

	first := rr.A.String()

	if address != "" {
		if first != address {
			t.Errorf("%s: %s (expected %s)", name, first, address)
			return
		}
	}

	t.Logf("%s: %s", name, first)
}