-
Notifications
You must be signed in to change notification settings - Fork 62
/
blocklistdb-hosts_test.go
48 lines (42 loc) · 1.25 KB
/
blocklistdb-hosts_test.go
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
package rdns
import (
"net"
"testing"
"github.com/miekg/dns"
"github.com/stretchr/testify/require"
)
func TestHostsDB(t *testing.T) {
loader := NewStaticLoader([]string{
"# some comment",
" ",
"127.0.0.1 domain1.com",
"0.0.0.0 domain2.com",
"0.0.0.0 domain3.com domain4.com. ",
":: domain5.com",
"::1 domain6.com",
"192.168.1.1 domain6.com",
})
m, err := NewHostsDB("testlist", loader)
require.NoError(t, err)
tests := []struct {
q string
typ uint16
match bool
ip []net.IP
}{
{"domain1.com.", dns.TypeA, true, []net.IP{net.ParseIP("127.0.0.1")}},
{"domain2.com.", dns.TypeA, true, []net.IP{nil}},
{"domain4.com.", dns.TypeA, true, []net.IP{nil}},
{"domain5.com.", dns.TypeA, true, []net.IP(nil)},
{"domain6.com.", dns.TypeA, true, []net.IP{net.ParseIP("192.168.1.1")}},
{"domain6.com.", dns.TypeAAAA, true, []net.IP{net.ParseIP("::1")}},
{"domainX.com.", dns.TypeA, false, nil},
}
for _, test := range tests {
q := dns.Question{Name: test.q, Qtype: test.typ, Qclass: dns.ClassINET}
ip, _, match, ok := m.Match(q)
require.Equal(t, test.match, ok, "query: %s", test.q)
require.EqualValues(t, test.ip, ip, "query: %s", test.q)
require.Equal(t, "testlist", match.List)
}
}