mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-12-16 11:52:58 +03:00
c9d2436d77
Merge in DNS/adguard-home from 2574-external-tests-3 to master Updates #2574. Squashed commit of the following: commit 29d429c65dee2621ca503710a7ba9522f14f55f9 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 20:06:57 2021 +0300 all: finally fix spacing commit 9e3a3be63b74852a7802e3f1832648444b58e4d0 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 19:59:09 2021 +0300 aghtest: polish spacing commit 8a984159fe813b95b989803f5b8b78d01a41bd39 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 18:44:47 2021 +0300 all: fix linux tests, imp code quality commit 0c1b42bacba1b23fa847e1fa032579c525b3eaa1 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 17:33:12 2021 +0300 all: mv testutil to aghtest package, imp tests
125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
package util
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
|
"github.com/miekg/dns"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
aghtest.DiscardLogOutput(m)
|
|
}
|
|
|
|
func prepareTestDir() string {
|
|
const dir = "./agh-test"
|
|
_ = os.RemoveAll(dir)
|
|
_ = os.MkdirAll(dir, 0o755)
|
|
return dir
|
|
}
|
|
|
|
func TestAutoHostsResolution(t *testing.T) {
|
|
ah := AutoHosts{}
|
|
|
|
dir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(dir) }()
|
|
|
|
f, _ := ioutil.TempFile(dir, "")
|
|
defer func() { _ = os.Remove(f.Name()) }()
|
|
defer f.Close()
|
|
|
|
_, _ = f.WriteString(" 127.0.0.1 host localhost # comment \n")
|
|
_, _ = f.WriteString(" ::1 localhost#comment \n")
|
|
|
|
ah.Init(f.Name())
|
|
|
|
// Existing host
|
|
ips := ah.Process("localhost", dns.TypeA)
|
|
assert.NotNil(t, ips)
|
|
assert.Len(t, ips, 1)
|
|
assert.Equal(t, net.ParseIP("127.0.0.1"), ips[0])
|
|
|
|
// Unknown host
|
|
ips = ah.Process("newhost", dns.TypeA)
|
|
assert.Nil(t, ips)
|
|
|
|
// Unknown host (comment)
|
|
ips = ah.Process("comment", dns.TypeA)
|
|
assert.Nil(t, ips)
|
|
|
|
// Test hosts file
|
|
table := ah.List()
|
|
names, ok := table["127.0.0.1"]
|
|
assert.True(t, ok)
|
|
assert.Equal(t, []string{"host", "localhost"}, names)
|
|
|
|
// Test PTR
|
|
a, _ := dns.ReverseAddr("127.0.0.1")
|
|
a = strings.TrimSuffix(a, ".")
|
|
hosts := ah.ProcessReverse(a, dns.TypePTR)
|
|
if assert.Len(t, hosts, 2) {
|
|
assert.Equal(t, hosts[0], "host")
|
|
}
|
|
|
|
a, _ = dns.ReverseAddr("::1")
|
|
a = strings.TrimSuffix(a, ".")
|
|
hosts = ah.ProcessReverse(a, dns.TypePTR)
|
|
if assert.Len(t, hosts, 1) {
|
|
assert.Equal(t, hosts[0], "localhost")
|
|
}
|
|
}
|
|
|
|
func TestAutoHostsFSNotify(t *testing.T) {
|
|
ah := AutoHosts{}
|
|
|
|
dir := prepareTestDir()
|
|
defer func() { _ = os.RemoveAll(dir) }()
|
|
|
|
f, _ := ioutil.TempFile(dir, "")
|
|
defer func() { _ = os.Remove(f.Name()) }()
|
|
defer f.Close()
|
|
|
|
// Init
|
|
_, _ = f.WriteString(" 127.0.0.1 host localhost \n")
|
|
ah.Init(f.Name())
|
|
|
|
// Unknown host
|
|
ips := ah.Process("newhost", dns.TypeA)
|
|
assert.Nil(t, ips)
|
|
|
|
// Stat monitoring for changes
|
|
ah.Start()
|
|
defer ah.Close()
|
|
|
|
// Update file
|
|
_, _ = f.WriteString("127.0.0.2 newhost\n")
|
|
_ = f.Sync()
|
|
|
|
// wait until fsnotify has triggerred and processed the file-modification event
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
// Check if we are notified about changes
|
|
ips = ah.Process("newhost", dns.TypeA)
|
|
assert.NotNil(t, ips)
|
|
assert.Len(t, ips, 1)
|
|
assert.True(t, net.IP{127, 0, 0, 2}.Equal(ips[0]))
|
|
}
|
|
|
|
func TestIP(t *testing.T) {
|
|
assert.True(t, net.IP{127, 0, 0, 1}.Equal(DNSUnreverseAddr("1.0.0.127.in-addr.arpa")))
|
|
assert.Equal(t, "::abcd:1234", DNSUnreverseAddr("4.3.2.1.d.c.b.a.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa").String())
|
|
assert.Equal(t, "::abcd:1234", DNSUnreverseAddr("4.3.2.1.d.c.B.A.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa").String())
|
|
|
|
assert.Nil(t, DNSUnreverseAddr("1.0.0.127.in-addr.arpa."))
|
|
assert.Nil(t, DNSUnreverseAddr(".0.0.127.in-addr.arpa"))
|
|
assert.Nil(t, DNSUnreverseAddr(".3.2.1.d.c.b.a.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa"))
|
|
assert.Nil(t, DNSUnreverseAddr("4.3.2.1.d.c.b.a.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0..ip6.arpa"))
|
|
assert.Nil(t, DNSUnreverseAddr("4.3.2.1.d.c.b. .0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa"))
|
|
}
|