package main

import (
	"net"
	"testing"

	"github.com/miekg/dns"
)

func init() {
	start("test/rrConfig.yaml")
}

// Helper

func request(name string, rrType uint16) (*dns.Msg, error) {
	m := new(dns.Msg)
	m.SetQuestion(dns.Fqdn(name), rrType)
	return dns.Exchange(m, "127.0.0.1:8053")
}

func containsA(haystack []dns.RR, name, ip string) bool {
	searchIP := net.ParseIP(ip)
	for _, v := range haystack {
		if v.Header().Name == dns.Fqdn(name) && v.Header().Rrtype == dns.TypeA {
			if t, ok := v.(*dns.A); ok {
				if t.A.Equal(searchIP) {
					return true
				}
			}
		}
	}

	return false
}

func containsAAAA(haystack []dns.RR, name, ip string) bool {
	searchIP := net.ParseIP(ip)
	for _, v := range haystack {
		if v.Header().Name == dns.Fqdn(name) && v.Header().Rrtype == dns.TypeAAAA {
			if t, ok := v.(*dns.AAAA); ok {
				if t.AAAA.Equal(searchIP) {
					return true
				}
			}
		}
	}

	return false
}

func TestNormalA(t *testing.T) {
	res, err := request("example.com", dns.TypeA)
	if err != nil {
		t.Error(err)
	}

	if !containsA(res.Answer, "example.com", "1.2.3.1") || !containsA(res.Answer, "example.com", "1.2.3.2") {
		t.FailNow()
	}
}

func TestNormalAAAA(t *testing.T) {
	res, err := request("example.com", dns.TypeAAAA)
	if err != nil {
		t.Error(err)
	}

	if !containsAAAA(res.Answer, "example.com", "2001:db8:10::1") {
		t.FailNow()
	}
}

func TestNormalSOA(t *testing.T) {
	res, err := request("example.com", dns.TypeSOA)
	if err != nil {
		t.Error(err)
	}

	if len(res.Answer) != 1 {
		t.Fatalf("Should only be 1 SOA got %d", len(res.Answer))
	}

	if soa, ok := res.Answer[0].(*dns.SOA); ok {
		if soa.Ns != "ns.example.com." {
			t.Fatal("Wrong SOA rr")
		}
	} else {
		t.Fatal("Answer is not a SOA rr")
	}
}