mirror of
https://github.com/mjl-/mox.git
synced 2025-07-10 07:54:40 +03:00
mox!
This commit is contained in:
46
tlsrpt/lookup_test.go
Normal file
46
tlsrpt/lookup_test.go
Normal file
@ -0,0 +1,46 @@
|
||||
package tlsrpt
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/mjl-/mox/dns"
|
||||
)
|
||||
|
||||
func TestLookup(t *testing.T) {
|
||||
resolver := dns.MockResolver{
|
||||
TXT: map[string][]string{
|
||||
"_smtp._tls.basic.example.": {"v=TLSRPTv1; rua=mailto:tlsrpt@basic.example"},
|
||||
"_smtp._tls.one.example.": {"v=TLSRPTv1; rua=mailto:tlsrpt@basic.example", "other"},
|
||||
"_smtp._tls.multiple.example.": {"v=TLSRPTv1; rua=mailto:tlsrpt@basic.example", "v=TLSRPTv1; rua=mailto:tlsrpt@basic.example"},
|
||||
"_smtp._tls.malformed.example.": {"v=TLSRPTv1; bad"},
|
||||
"_smtp._tls.other.example.": {"other"},
|
||||
},
|
||||
Fail: map[dns.Mockreq]struct{}{
|
||||
{Type: "txt", Name: "_smtp._tls.temperror.example."}: {},
|
||||
},
|
||||
}
|
||||
|
||||
test := func(domain string, expRecord *Record, expErr error) {
|
||||
t.Helper()
|
||||
|
||||
d := dns.Domain{ASCII: domain}
|
||||
record, _, err := Lookup(context.Background(), resolver, d)
|
||||
if (err == nil) != (expErr == nil) || err != nil && !errors.Is(err, expErr) {
|
||||
t.Fatalf("lookup, got err %#v, expected %#v", err, expErr)
|
||||
}
|
||||
if err == nil && !reflect.DeepEqual(record, expRecord) {
|
||||
t.Fatalf("lookup, got %#v, expected %#v", record, expRecord)
|
||||
}
|
||||
}
|
||||
|
||||
test("basic.example", &Record{Version: "TLSRPTv1", RUAs: [][]string{{"mailto:tlsrpt@basic.example"}}}, nil)
|
||||
test("one.example", &Record{Version: "TLSRPTv1", RUAs: [][]string{{"mailto:tlsrpt@basic.example"}}}, nil)
|
||||
test("multiple.example", nil, ErrMultipleRecords)
|
||||
test("absent.example", nil, ErrNoRecord)
|
||||
test("other.example", nil, ErrNoRecord)
|
||||
test("malformed.example", nil, ErrRecordSyntax)
|
||||
test("temperror.example", nil, ErrDNS)
|
||||
}
|
Reference in New Issue
Block a user