package hosted

import "testing"

func TestMXMatchesExpected(t *testing.T) {
	if !MXMatchesExpected([]string{"mail.acme.com."}, []string{"mail.acme.com"}) {
		t.Fatal("exact mx match")
	}
	if !MXMatchesExpected([]string{"mx1.mail.ultisuite.fr"}, []string{"mail.ultisuite.fr"}) {
		t.Fatal("suffix mx match")
	}
	if MXMatchesExpected([]string{"aspmx.l.google.com"}, []string{"mail.ultisuite.fr"}) {
		t.Fatal("google mx should not match")
	}
}

func TestTXTContainsToken(t *testing.T) {
	if !TXTContainsToken([]string{"abc123", "other"}, "abc123") {
		t.Fatal("expected token match")
	}
	if TXTContainsToken([]string{"wrong"}, "abc123") {
		t.Fatal("unexpected token match")
	}
}
