Function TestExtractDomains

internal/dmarc/evaluate_test.go:459–514

(t *testing.T)

}

func TestExtractDomains(t *testing.T) {
	type tCase struct {
		hdr string

		fromDomain string
	}
	test := func(i int, c tCase) {
		hdr, err := textproto.ReadHeader(bufio.NewReader(strings.NewReader(c.hdr + "\n\n")))
		if err != nil {
			panic(err)
		}

		domain, err := ExtractFromDomain(hdr)
		if c.fromDomain == "" && err == nil {
			t.Errorf("%d: expected failure, got fromDomain = %s", i, domain)
			return
		}
		if c.fromDomain != "" && err != nil {
			t.Errorf("%d: unexpected error: %v", i, err)
			return
		}
		if domain != c.fromDomain {
			t.Errorf("%d: want fromDomain = %v but got %s", i, c.fromDomain, domain)
		}
	}

	cases := []tCase{
		{
			hdr:        `From: <test@example.org>`,
			fromDomain: "example.org",
		},
		{
			hdr:        `From: <test@foo.example.org>`,
			fromDomain: "foo.example.org",
		},
		{
			hdr: `From: <test@foo.example.org>, <test@bar.example.org>`,
		},
		{
			hdr: `From: <test@foo.example.org>,
From: <test@bar.example.org>`,
		},
		{
			hdr: `From: <test@>`,
		},
		{
			hdr: `From: `,
		},
		{
			hdr: `From: foo`,
		},
	}
	for i, case_ := range cases {
		test(i, case_)
	}
}