1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
package autoconfig
import (
"context"
"errors"
"net"
"testing"
"time"
"github.com/stretchr/testify/assert"
)
func TestMX(t *testing.T) {
tests := []struct {
address string
correctConfig *Config
}{
{
address: "poldrack.dev",
correctConfig: &Config{
Found: ProtocolIMAP,
IMAP: Credentials{
Encryption: EncryptionSTARTTLS,
Address: "mail.moritz.sh",
Port: 143,
Username: "john@poldrack.dev",
},
SMTP: Credentials{
Encryption: EncryptionSTARTTLS,
Address: "mail.moritz.sh",
Port: 587,
Username: "john@poldrack.dev",
},
},
},
}
netDial = mxTestDialer
lookupMX = mxTestLookup
defer func() {
netDial = net.Dial
lookupMX = net.LookupMX
}()
for _, test := range tests {
t.Run(test.address, func(t *testing.T) {
result := make(chan *Config)
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
go guessMX(ctx, "john", test.address, result)
select {
case res := <-result:
if res == nil {
t.Log("no result")
t.FailNow()
}
assert.Equal(t, test.correctConfig, res)
case <-ctx.Done():
t.Error("retrieval timed out!")
}
})
}
}
func mxTestLookup(address string) ([]*net.MX, error) {
switch address {
case "poldrack.dev":
return []*net.MX{
{Host: "mail.moritz.sh", Pref: 1},
}, nil
default:
return nil, errors.New("unknown address")
}
}
|
