mirror of
https://github.com/folbricht/routedns.git
synced 2026-05-07 08:29:36 -05:00
Frank Olbricht
acc8842fad
* Support naming blocklists to help with logging * Support naming of lists in response blocklists too * Add list name to client-blocklist as well
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
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.ParseIP("127.0.0.1")},
|
|
{"domain2.com.", dns.TypeA, true, nil},
|
|
{"domain4.com.", dns.TypeA, true, nil},
|
|
{"domain5.com.", dns.TypeA, true, nil},
|
|
{"domain6.com.", dns.TypeA, true, net.ParseIP("192.168.1.1")},
|
|
{"domain6.com.", dns.TypeAAAA, true, 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.Equal(t, test.ip, ip, "query: %s", test.q)
|
|
require.Equal(t, "testlist", match.List)
|
|
}
|
|
}
|