package netboxdns

import (
	"context"
	"fmt"
	"net/http"
	"net/url"
	"testing"

	"github.com/coredns/coredns/plugin/pkg/dnstest"
	"github.com/coredns/coredns/plugin/test"
	"github.com/doubleu-labs/coredns-netbox-plugin-dns/internal/netbox"
	"github.com/miekg/dns"
)

func TestPluginName(t *testing.T) {
	netboxdns := &NetboxDNS{}
	name := netboxdns.Name()
	if name != pluginName {
		t.Errorf("plugin name is wrong. how did this happen...")
	}
}

type testFamily int

const (
	testFamilyV4 testFamily = iota
	testFamilyV6
)

var testFamilyToString map[testFamily]string = map[testFamily]string{
	testFamilyV4: "V4",
	testFamilyV6: "V6",
}

const (
	testInstanceToken   string = "w5pgWXPqZVmngLN4w4XwuPvZfUC72ytDxnnHgEmI"
	testInstanceUrlHost string = "localhost:9999"
	testInstanceUrlPath string = "/api/plugins/netbox-dns/"
)

var netboxdnsPlugin NetboxDNS = NetboxDNS{
	Next:  test.ErrorHandler(),
	zones: []string{"."},
	requestClient: &netbox.APIRequestClient{
		Client: &http.Client{
			Timeout: defaultHTTPClientTimeout,
		},
		NetboxURL: &url.URL{
			Scheme: "http",
			Host:   testInstanceUrlHost,
			Path:   testInstanceUrlPath,
		},
		Token: testInstanceToken,
	},
}

func RunTestLookup(t *testing.T, tcs []test.Case, family testFamily) {
	for _, tc := range tcs {
		tcName := fmt.Sprintf(
			"%s %s %s",
			tc.Qname,
			dns.TypeToString[tc.Qtype],
			testFamilyToString[family],
		)
		t.Run(tcName, func(t *testing.T) {
			msg := tc.Msg()
			respWriter := GetTestResponseWriter(family)
			rec := dnstest.NewRecorder(respWriter)
			_, err := netboxdnsPlugin.ServeDNS(context.Background(), rec, msg)
			if err != nil {
				t.Errorf("expected no error, got %v", err)
				return
			}
			resp := rec.Msg
			if resp == nil {
				t.Fatal("got nil response message")
			}
			if ok := RunTestLookupCheckCNAME(t, tc, resp); !ok {
				return
			}
			if err := test.SortAndCheck(resp, tc); err != nil {
				t.Logf("%s\n", rec.Msg)
				t.Error(err)
			}
		})
	}
}

func GetTestResponseWriter(family testFamily) dns.ResponseWriter {
	var respWriter dns.ResponseWriter
	switch family {
	case testFamilyV4:
		respWriter = &test.ResponseWriter{}
	case testFamilyV6:
		respWriter = &test.ResponseWriter6{}
	}
	return respWriter
}

func RunTestLookupCheckCNAME(t *testing.T, tc test.Case, resp *dns.Msg) bool {
	if err := test.CNAMEOrder(resp); err != nil {
		t.Errorf("cname response out of order")
	}
	if tc.Qtype == dns.TypeCNAME {
		if err := test.Header(tc, resp); err != nil {
			t.Error(err)
		}
		if err := test.Section(tc, test.Answer, resp.Answer); err != nil {
			t.Error(err)
		}
		if err := test.Section(tc, test.Ns, resp.Ns); err != nil {
			t.Error(err)
		}
		if err := test.Section(tc, test.Extra, resp.Extra); err != nil {
			t.Error(err)
		}
		return false
	}
	return true
}

var (
	exampledotcomName          string = "example.com."
	subdotexampledotcomName    string = "sub.example.com."
	subtwodotexampledotcomName string = "subtwo.example.com."

	exampledotcomNS []dns.RR = []dns.RR{
		test.NS("example.com. 3600 IN NS dns01.example.com."),
		test.NS("example.com. 3600 IN NS dns02.example.com."),
	}
	subdotexampledotcomNS []dns.RR = []dns.RR{
		test.NS("sub.example.com. 3600 IN NS dns01.example.com."),
		test.NS("sub.example.com. 3600 IN NS dns02.example.com."),
	}
	subtwodotexampledotcomNS []dns.RR = []dns.RR{
		test.NS("subtwo.example.com. 3600 IN NS dns01.example.com."),
		test.NS("subtwo.example.com. 3600 IN NS dns02.example.com."),
	}
	exampledotcomNS1Record4 dns.RR   = test.A("dns01.example.com. 3600 IN A 10.0.0.10")
	exampledotcomNS2Record4 dns.RR   = test.A("dns02.example.com. 3600 IN A 10.0.0.11")
	exampledotcomNSAddr4    []dns.RR = []dns.RR{
		exampledotcomNS1Record4,
		exampledotcomNS2Record4,
	}
	exampledotcomNS1Record6 dns.RR   = test.AAAA("dns01.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:10")
	exampledotcomNS2Record6 dns.RR   = test.AAAA("dns02.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:11")
	exampledotcomNSAddr6    []dns.RR = []dns.RR{
		exampledotcomNS1Record6,
		exampledotcomNS2Record6,
	}
)

var (
	testLookupForwardZonesCasesv4 []test.Case = []test.Case{
		{
			Qname: exampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns:    exampledotcomNS,
			Extra: exampledotcomNSAddr4,
		},
		{
			Qname: subdotexampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("sub.example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("sub.example.com. 3600 IN NS dns01.example.com"),
				test.NS("sub.example.com. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr4,
		},
		{
			Qname: subtwodotexampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("subtwo.example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("subtwo.example.com. 3600 IN NS dns01.example.com"),
				test.NS("subtwo.example.com. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr4,
		},
	}

	testLookupReverseZonesCasesv4 []test.Case = []test.Case{
		{
			Qname: "0.0.10.in-addr.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("0.0.10.in-addr.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("0.0.10.in-addr.arpa. 3600 IN NS dns01.example.com"),
				test.NS("0.0.10.in-addr.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr4,
		},
		{
			Qname: "1.0.10.in-addr.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("1.0.10.in-addr.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("1.0.10.in-addr.arpa. 3600 IN NS dns01.example.com"),
				test.NS("1.0.10.in-addr.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr4,
		},
		{
			Qname: "2.0.10.in-addr.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("2.0.10.in-addr.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("2.0.10.in-addr.arpa. 3600 IN NS dns01.example.com"),
				test.NS("2.0.10.in-addr.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr4,
		},
	}

	testLookupForwardZonesCasesv6 []test.Case = []test.Case{
		{
			Qname: exampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns:    exampledotcomNS,
			Extra: exampledotcomNSAddr6,
		},
		{
			Qname: subdotexampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("sub.example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("sub.example.com. 3600 IN NS dns01.example.com"),
				test.NS("sub.example.com. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr6,
		},
		{
			Qname: subtwodotexampledotcomName, Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("subtwo.example.com. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("subtwo.example.com. 3600 IN NS dns01.example.com"),
				test.NS("subtwo.example.com. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr6,
		},
	}

	testLookupReverseZonesCasesv6 []test.Case = []test.Case{
		{
			Qname: "1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns01.example.com"),
				test.NS("1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr6,
		},
		{
			Qname: "2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns01.example.com"),
				test.NS("2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr6,
		},
		{
			Qname: "3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypeSOA,
			Answer: []dns.RR{
				test.SOA("3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 86400 IN SOA dns01.example.com. admin.example.com. 1 43200 7200 2419200 3600"),
			},
			Ns: []dns.RR{
				test.NS("3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns01.example.com"),
				test.NS("3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN NS dns02.example.com"),
			},
			Extra: exampledotcomNSAddr6,
		},
	}
)

func TestLookupZones(t *testing.T) {
	RunTestLookup(t, testLookupForwardZonesCasesv4, testFamilyV4)
	RunTestLookup(t, testLookupReverseZonesCasesv4, testFamilyV4)
	RunTestLookup(t, testLookupForwardZonesCasesv6, testFamilyV6)
	RunTestLookup(t, testLookupReverseZonesCasesv6, testFamilyV6)
}

var (
	testLookupRecordV4 []test.Case = []test.Case{
		{
			Qname: exampledotcomName, Qtype: dns.TypeNS,
			Answer: exampledotcomNS,
			Extra:  exampledotcomNSAddr4,
		},
		{
			Qname: "dns01.example.com", Qtype: dns.TypeA,
			Answer: []dns.RR{
				exampledotcomNS1Record4,
			},
		},
		{
			Qname: "dns02.example.com", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("dns02.example.com. 3600 IN A 10.0.0.11"),
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeA,
			Ns:    exampledotcomNS,
			Extra: exampledotcomNSAddr4,
		},
		{
			Qname: "aservice.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("aservice.example.com. 3600 IN A 10.0.0.12"),
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeMX,
			Answer: []dns.RR{
				test.MX("example.com. 3600 IN MX 10 mail.example.com."),
			},
			Extra: []dns.RR{
				test.A("mail.example.com. 3600 IN A 10.0.0.13"),
			},
		},
		{
			Qname: "mail.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("mail.example.com. 3600 IN A 10.0.0.13"),
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeTXT,
			Answer: []dns.RR{
				test.TXT(`example.com. 3600 IN TXT "my value" "second my value" "third my value"`),
				test.TXT(`example.com. 3600 IN TXT "newline record" "second value"`),
				test.TXT(`example.com. 3600 IN TXT "some value" "another value"`),
				test.TXT(`example.com. 3600 IN TXT "v=DMARC1;p=none;sp=quarantine;pct=100;rua=admin@example.com;"`),
				test.TXT(`example.com. 3600 IN TXT "v=spf1 ip4:10.0.0.13 ip6:2001:db8:dead:beef::1:13 a -all"`),
			},
		},
		{
			Qname: "puppet-server-a.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("puppet-server-a.example.com. 3600 IN A 10.0.0.15"),
			},
		},
		{
			Qname: "puppet-server-b.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("puppet-server-b.example.com. 3600 IN A 10.0.0.16"),
			},
		},
		{
			Qname: "_x-puppet._tcp.example.com.", Qtype: dns.TypeSRV,
			Answer: []dns.RR{
				test.SRV("_x-puppet._tcp.example.com. 3600 IN SRV 0 5 8140 puppet-server-a.example.com."),
				test.SRV("_x-puppet._tcp.example.com. 3600 IN SRV 0 5 8140 puppet-server-b.example.com."),
			},
			Extra: []dns.RR{
				test.A("puppet-server-a.example.com. 3600 IN A 10.0.0.15"),
				test.A("puppet-server-b.example.com. 3600 IN A 10.0.0.16"),
			},
		},
		{
			Qname: "web.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("web.example.com. 3600 IN A 10.0.0.17"),
			},
		},
		{
			Qname: "www.example.com.", Qtype: dns.TypeCNAME,
			Answer: []dns.RR{
				test.CNAME("www.example.com. 3600 IN CNAME web.example.com."),
				test.A("web.example.com. 3600 IN A 10.0.0.17"),
			},
		},
		{
			Qname: subdotexampledotcomName, Qtype: dns.TypeNS,
			Answer: subdotexampledotcomNS,
			Extra:  exampledotcomNSAddr4,
		},
		{
			Qname: "myservice.sub.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("myservice.sub.example.com. 3600 IN A 10.0.1.10"),
			},
		},
		{
			Qname: subtwodotexampledotcomName, Qtype: dns.TypeNS,
			Answer: subtwodotexampledotcomNS,
			Extra:  exampledotcomNSAddr4,
		},
		{
			Qname: "myotherservice.subtwo.example.com.", Qtype: dns.TypeA,
			Answer: []dns.RR{
				test.A("myotherservice.subtwo.example.com. 3600 IN A 10.0.2.10"),
			},
		},
	}

	testLookupPTRV4 []test.Case = []test.Case{
		{
			Qname: "10.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("10.0.0.10.in-addr.arpa. 3600 IN PTR dns01.example.com."),
			},
		},
		{
			Qname: "11.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("11.0.0.10.in-addr.arpa. 3600 IN PTR dns02.example.com."),
			},
		},
		{
			Qname: "12.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("12.0.0.10.in-addr.arpa. 3600 IN PTR aservice.example.com."),
			},
		},
		{
			Qname: "13.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("13.0.0.10.in-addr.arpa. 3600 IN PTR mail.example.com."),
			},
		},
		{
			Qname: "15.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("15.0.0.10.in-addr.arpa. 3600 IN PTR puppet-server-a.example.com."),
			},
		},
		{
			Qname: "16.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("16.0.0.10.in-addr.arpa. 3600 IN PTR puppet-server-b.example.com."),
			},
		},
		{
			Qname: "17.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("17.0.0.10.in-addr.arpa. 3600 IN PTR web.example.com."),
			},
		},
		{
			Qname: "10.1.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("10.1.0.10.in-addr.arpa. 3600 IN PTR myservice.sub.example.com."),
			},
		},
		{
			Qname: "10.2.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("10.2.0.10.in-addr.arpa. 3600 IN PTR myotherservice.subtwo.example.com."),
			},
		},
	}

	testLookupRecordV6 []test.Case = []test.Case{
		{
			Qname: exampledotcomName, Qtype: dns.TypeNS,
			Answer: exampledotcomNS,
			Extra:  exampledotcomNSAddr6,
		},
		{
			Qname: "dns01.example.com", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				exampledotcomNS1Record6,
			},
		},
		{
			Qname: "dns02.example.com", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				exampledotcomNS2Record6,
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeAAAA,
			Ns:    exampledotcomNS,
			Extra: exampledotcomNSAddr6,
		},
		{
			Qname: "aservice.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("aservice.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:12"),
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeMX,
			Answer: []dns.RR{
				test.MX("example.com. 3600 IN MX 10 mail.example.com."),
			},
			Extra: []dns.RR{
				test.AAAA("mail.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:13"),
			},
		},
		{
			Qname: "mail.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("mail.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:13"),
			},
		},
		{
			Qname: exampledotcomName, Qtype: dns.TypeTXT,
			Answer: []dns.RR{
				test.TXT(`example.com. 3600 IN TXT "my value" "second my value" "third my value"`),
				test.TXT(`example.com. 3600 IN TXT "newline record" "second value"`),
				test.TXT(`example.com. 3600 IN TXT "some value" "another value"`),
				test.TXT(`example.com. 3600 IN TXT "v=DMARC1;p=none;sp=quarantine;pct=100;rua=admin@example.com;"`),
				test.TXT(`example.com. 3600 IN TXT "v=spf1 ip4:10.0.0.13 ip6:2001:db8:dead:beef::1:13 a -all"`),
			},
		},
		{
			Qname: "puppet-server-a.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("puppet-server-a.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:15"),
			},
		},
		{
			Qname: "puppet-server-b.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("puppet-server-b.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:16"),
			},
		},
		{
			Qname: "_x-puppet._tcp.example.com.", Qtype: dns.TypeSRV,
			Answer: []dns.RR{
				test.SRV("_x-puppet._tcp.example.com. 3600 IN SRV 0 5 8140 puppet-server-a.example.com."),
				test.SRV("_x-puppet._tcp.example.com. 3600 IN SRV 0 5 8140 puppet-server-b.example.com."),
			},
			Extra: []dns.RR{
				test.AAAA("puppet-server-a.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:15"),
				test.AAAA("puppet-server-b.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:16"),
			},
		},
		{
			Qname: "web.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("web.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:17"),
			},
		},
		{
			Qname: "www.example.com.", Qtype: dns.TypeCNAME,
			Answer: []dns.RR{
				test.CNAME("www.example.com. 3600 IN CNAME web.example.com."),
				test.AAAA("web.example.com. 3600 IN AAAA 2001:db8:dead:beef::1:17"),
			},
		},
		{
			Qname: subdotexampledotcomName, Qtype: dns.TypeNS,
			Answer: subdotexampledotcomNS,
			Extra:  exampledotcomNSAddr6,
		},
		{
			Qname: "myservice.sub.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("myservice.sub.example.com. 3600 IN AAAA 2001:db8:dead:beef::2:10"),
			},
		},
		{
			Qname: subtwodotexampledotcomName, Qtype: dns.TypeNS,
			Answer: subtwodotexampledotcomNS,
			Extra:  exampledotcomNSAddr6,
		},
		{
			Qname: "myotherservice.subtwo.example.com.", Qtype: dns.TypeAAAA,
			Answer: []dns.RR{
				test.AAAA("myotherservice.subtwo.example.com. 3600 IN AAAA 2001:db8:dead:beef::3:10"),
			},
		},
	}

	testLookupPTRV6 []test.Case = []test.Case{
		{
			Qname: "0.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("0.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR dns01.example.com."),
			},
		},
		{
			Qname: "1.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("1.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR dns02.example.com."),
			},
		},
		{
			Qname: "2.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("2.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR aservice.example.com."),
			},
		},
		{
			Qname: "3.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("3.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR mail.example.com."),
			},
		},
		{
			Qname: "5.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("5.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR puppet-server-a.example.com."),
			},
		},
		{
			Qname: "6.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("6.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR puppet-server-b.example.com."),
			},
		},
		{
			Qname: "7.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("7.1.0.0.1.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR web.example.com."),
			},
		},
		{
			Qname: "0.1.0.0.2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("0.1.0.0.2.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR myservice.sub.example.com."),
			},
		},
		{
			Qname: "0.1.0.0.3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa.", Qtype: dns.TypePTR,
			Answer: []dns.RR{
				test.PTR("0.1.0.0.3.0.0.0.0.0.0.0.0.0.0.0.f.e.e.b.d.a.e.d.8.b.d.0.1.0.0.2.ip6.arpa. 3600 IN PTR myotherservice.subtwo.example.com."),
			},
		},
	}
)

func TestLookupRecords(t *testing.T) {
	RunTestLookup(t, testLookupRecordV4, testFamilyV4)
	RunTestLookup(t, testLookupPTRV4, testFamilyV4)
	RunTestLookup(t, testLookupRecordV6, testFamilyV6)
	RunTestLookup(t, testLookupPTRV6, testFamilyV6)
}

var (
	testUnknownRecords []test.Case = []test.Case{
		{
			Qname: "noop.com.", Qtype: dns.TypeSOA,
			Rcode: dns.RcodeNameError,
		},
	}

	testUnknownRecordsV4 []test.Case = []test.Case{
		{
			Qname: "noop.example.com.", Qtype: dns.TypeA,
			Rcode: dns.RcodeNameError,
		},
	}

	testUnknownRecordsV6 []test.Case = []test.Case{
		{
			Qname: "noop.example.com.", Qtype: dns.TypeAAAA,
			Rcode: dns.RcodeNameError,
		},
	}
)

func TestLookupUnknown(t *testing.T) {
	RunTestLookup(t, testUnknownRecords, testFamilyV4)
	RunTestLookup(t, testUnknownRecordsV4, testFamilyV4)
	RunTestLookup(t, testUnknownRecordsV6, testFamilyV6)
}

func TestOffline(t *testing.T) {
	netboxdns := NetboxDNS{
		Next:  test.ErrorHandler(),
		zones: []string{"."},
		requestClient: &netbox.APIRequestClient{
			Client: &http.Client{
				Timeout: defaultHTTPClientTimeout,
			},
			NetboxURL: &url.URL{
				Scheme: "http",
				Host:   "localhost:9876",
				Path:   testInstanceUrlPath,
			},
			Token: testInstanceToken,
		},
	}
	tc := test.Case{
		Qname: exampledotcomName, Qtype: dns.TypeA,
	}
	msg := tc.Msg()
	rec := dnstest.NewRecorder(&test.ResponseWriter{})
	_, err := netboxdns.ServeDNS(context.Background(), rec, msg)
	if err == nil {
		t.Error("expected connection error, got none")
	}
}

func TestUnauthorized(t *testing.T) {
	netboxdns := NetboxDNS{
		Next:  test.ErrorHandler(),
		zones: []string{"."},
		requestClient: &netbox.APIRequestClient{
			Client: &http.Client{
				Timeout: defaultHTTPClientTimeout,
			},
			NetboxURL: &url.URL{
				Scheme: "http",
				Host:   testInstanceUrlHost,
				Path:   testInstanceUrlPath,
			},
			Token: "noop",
		},
	}
	tc := test.Case{
		Qname: exampledotcomName, Qtype: dns.TypeA,
	}
	msg := tc.Msg()
	rec := dnstest.NewRecorder(&test.ResponseWriter{})
	_, err := netboxdns.ServeDNS(context.Background(), rec, msg)
	if err == nil {
		t.Error("expected connection error, got none")
	}
}

func TestFallthrough(t *testing.T) {
	netboxdns := NetboxDNS{
		Next:  test.ErrorHandler(),
		zones: []string{exampledotcomName},
		requestClient: &netbox.APIRequestClient{
			Client: &http.Client{
				Timeout: defaultHTTPClientTimeout,
			},
			NetboxURL: &url.URL{
				Scheme: "http",
				Host:   testInstanceUrlHost,
				Path:   testInstanceUrlPath,
			},
			Token: testInstanceToken,
		},
	}
	netboxdns.fall.SetZonesFromArgs([]string{"out.example.com"})
	tc := test.Case{
		Qname: "a.out.example.com.", Qtype: dns.TypeA,
	}
	msg := tc.Msg()
	rec := dnstest.NewRecorder(&test.ResponseWriter{})
	_, err := netboxdns.ServeDNS(context.Background(), rec, msg)
	if err != nil {
		t.Errorf("expected fallthrough, got %v", err)
	}
}

func TestUnhandledZone(t *testing.T) {
	netboxdns := NetboxDNS{
		Next:  test.ErrorHandler(),
		zones: []string{exampledotcomName},
		requestClient: &netbox.APIRequestClient{
			Client: &http.Client{
				Timeout: defaultHTTPClientTimeout,
			},
			NetboxURL: &url.URL{
				Scheme: "http",
				Host:   testInstanceUrlHost,
				Path:   testInstanceUrlPath,
			},
			Token: testInstanceToken,
		},
	}
	tc := test.Case{
		Qname: "www.example.net.", Qtype: dns.TypeA,
	}
	msg := tc.Msg()
	rec := dnstest.NewRecorder(&test.ResponseWriter{})
	_, err := netboxdns.ServeDNS(context.Background(), rec, msg)
	if err != nil {
		t.Errorf("expected no error, got %v", err)
	}
}