daxi20/AdGuardHome
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

all: rewrites

Browse Source
This commit is contained in:
Dimitry Kolyshev
2022-12-15 12:13:14 +07:00
parent e657899c32
commit 526c358697
4 changed files with 53 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/dnsforward/dnsforward_test.go
View File

@@ -947,12 +947,11 @@ func TestRewrite(t *testing.T) {
reply, eerr = dns.Exchange(req, addr.String())
require.NoError(t, eerr)
// TODO (d.kolyshev): Investigate
// require.Len(t, reply.Answer, 2)
require.Len(t, reply.Answer, 2)
// assert.Equal(t, "test.com.", reply.Answer[0].(*dns.CNAME).Target)
// assert.True(t, net.IP{1, 2, 3, 4}.Equal(reply.Answer[1].(*dns.A).A))
assert.True(t, net.IP{1, 2, 3, 4}.Equal(reply.Answer[0].(*dns.A).A))
assert.Equal(t, "test.com.", reply.Answer[0].(*dns.CNAME).Target)
assert.Equal(t, dns.TypeA, reply.Answer[1].Header().Rrtype)
assert.True(t, net.IP{1, 2, 3, 4}.Equal(reply.Answer[1].(*dns.A).A))
req = createTestMessageWithType("my.alias.example.org.", dns.TypeA)
reply, eerr = dns.Exchange(req, addr.String())
@@ -963,11 +962,10 @@ func TestRewrite(t *testing.T) {
assert.Equal(t, "my.alias.example.org.", reply.Question[0].Name)
// TODO (d.kolyshev): Investigate
//require.Len(t, reply.Answer, 2)
//
//assert.Equal(t, "example.org.", reply.Answer[0].(*dns.CNAME).Target)
//assert.Equal(t, dns.TypeA, reply.Answer[1].Header().Rrtype)
require.Len(t, reply.Answer, 2)
assert.Equal(t, "example.org.", reply.Answer[0].(*dns.CNAME).Target)
assert.Equal(t, dns.TypeA, reply.Answer[1].Header().Rrtype)
}
for _, protect := range []bool{true, false} {