package dnsrecord

import "testing"

func TestValidateRRSetAcceptsSupportedRecords(t *testing.T) {
	v := newTestValidator(t)

	cases := []struct {
		recordType string
		content    string
	}{
		{"A", "192.0.2.10"},
		{"AAAA", "2001:db8::1"},
		{"CAA", `0 issue "letsencrypt.org"`},
		{"CNAME", "target.example.org."},
		{"MX", "10 mail.example.org."},
		{"NS", "ns1.example.org."},
		{"SOA", "ns1.example.org. hostmaster.example.org. 2026010101 3600 600 604800 300"},
		{"SRV", "10 20 443 service.example.org."},
		{"TXT", `"v=spf1 -all"`},
	}

	for _, tc := range cases {
		t.Run(tc.recordType, func(t *testing.T) {
			if _, err := v.ValidateRRSet("www.example.org", tc.recordType, 300, []string{tc.content}); err != nil {
				t.Fatalf("ValidateRRSet returned error: %v", err)
			}
		})
	}
}

func TestValidateRRSetRejectsInvalidRecords(t *testing.T) {
	v := newTestValidator(t)

	cases := []struct {
		name       string
		recordType string
		content    string
	}{
		{"bad name", "A", "192.0.2.10"},
		{"www.example.org.", "A", "2001:db8::1"},
		{"www.example.org.", "AAAA", "192.0.2.10"},
		{"www.example.org.", "MX", "mail.example.org."},
		{"www.example.org.", "CNAME", "target.example.org"},
		{"www.example.org.", "TXT", "not quoted"},
		{"www.example.org.", "SOA", "ns1.example.org. hostmaster.example.org."},
		{"www.example.org.", "UNSUPPORTED", "value"},
	}

	for _, tc := range cases {
		t.Run(tc.recordType+" "+tc.content, func(t *testing.T) {
			if _, err := v.ValidateRRSet(tc.name, tc.recordType, 300, []string{tc.content}); err == nil {
				t.Fatal("expected validation error")
			}
		})
	}
}

func TestValidateRRSetRequiresTTL(t *testing.T) {
	v := newTestValidator(t)

	if _, err := v.ValidateRRSet("www.example.org.", "A", 0, []string{"192.0.2.10"}); err == nil {
		t.Fatal("expected ttl validation error")
	}
}

func newTestValidator(t *testing.T) *Validator {
	t.Helper()

	v, err := NewValidator()
	if err != nil {
		t.Fatalf("NewValidator returned error: %v", err)
	}
	return v
}