package tmpl_test

import (
	"testing"

	"github.com/vasyakrg/dns-autoresolver/internal/model"
	"github.com/vasyakrg/dns-autoresolver/internal/store/dto"
	"github.com/vasyakrg/dns-autoresolver/internal/tmpl"
)

func TestMaterializeReplacesInNameAndValues(t *testing.T) {
	doc := dto.TemplateDoc{Records: []dto.RecordDTO{
		{Type: "TXT", Name: "_dmarc.{{domain_name}}.", TTL: 600, Values: []string{"v=DMARC1; p=quarantine"}},
		{Type: "MX", Name: "{{domain_name}}.", TTL: 600, Values: []string{"0 pmg2-mail.{{domain_name}}."}},
		{Type: "TXT", Name: "{{domain_name}}.", TTL: 600, Values: []string{"v=spf1 a:mail.{{domain_name}} ~all"}},
	}}
	recs := tmpl.Materialize(doc, "reconops.ru.") // trailing dot stripped
	if recs[0].Name != "_dmarc.reconops.ru." {
		t.Fatalf("name: %q", recs[0].Name)
	}
	if recs[1].Values[0] != "0 pmg2-mail.reconops.ru." {
		t.Fatalf("mx value: %q", recs[1].Values[0])
	}
	if recs[2].Values[0] != "v=spf1 a:mail.reconops.ru ~all" {
		t.Fatalf("spf value: %q", recs[2].Values[0])
	}
}

func TestParameterizeIsInverseForZoneOccurrences(t *testing.T) {
	recs := []model.Record{
		{Type: "TXT", Name: "_dmarc.reconops.ru.", TTL: 600, Values: []string{"v=DMARC1"}},
		{Type: "TXT", Name: "reconops.ru.", TTL: 600, Values: []string{"v=spf1 a:mail.reconops.ru ~all"}},
		{Type: "CNAME", Name: "mail.reconops.ru.", TTL: 600, Values: []string{"amail.amega.kz."}}, // external host untouched
	}
	doc := tmpl.Parameterize(recs, "reconops.ru.")
	if doc.Records[0].Name != "_dmarc.{{domain_name}}." {
		t.Fatalf("name: %q", doc.Records[0].Name)
	}
	if doc.Records[1].Values[0] != "v=spf1 a:mail.{{domain_name}} ~all" {
		t.Fatalf("spf: %q", doc.Records[1].Values[0])
	}
	if doc.Records[2].Values[0] != "amail.amega.kz." {
		t.Fatalf("external cname value must be untouched: %q", doc.Records[2].Values[0])
	}
}