82505566f8
Merge in DNS/adguard-home from 2846-cover-aghnet-vol.2 to master
Updates #2846.
Closes #4408.
Squashed commit of the following:
commit 8d62b29d5b5be875cb71e518e479e321d853eb1a
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 20:42:04 2022 +0300
home: recover panic
commit 1d98109e910830bec712c7aecbbbcb8f659d823d
Merge: ac11d751 9ce2a0fb
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 20:32:05 2022 +0300
Merge branch 'master' into 2846-cover-aghnet-vol.2
commit ac11d751fb7951e3dd0940bf425a893223c32789
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 20:29:41 2022 +0300
aghnet: use iotest
commit 7c923df7bafd5d4b91c4b4a01e75ab161944f949
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 20:17:19 2022 +0300
aghnet: cover more
commit 3bfd4d587e4b887b5527d60c0eb6027da15c7e37
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 14:13:59 2022 +0300
aghnet: cover arpdb more
commit cd5cf7bbdecceeab6d3abee10a5572e1e907cc67
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Wed Mar 23 13:05:35 2022 +0300
all: rm arpdb initial refresh
commit 0fb8d9e44a4d130ca4e8fc2ea5d595ec08555302
Author: Eugene Burkov <E.Burkov@AdGuard.COM>
Date: Tue Mar 22 21:13:16 2022 +0300
aghnet: cover arpdb
83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package aghnet
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func createTestSystemResolversImpl(
|
|
t *testing.T,
|
|
hostGenFunc HostGenFunc,
|
|
) (imp *systemResolvers) {
|
|
t.Helper()
|
|
|
|
sr := createTestSystemResolvers(t, hostGenFunc)
|
|
require.IsType(t, (*systemResolvers)(nil), sr)
|
|
|
|
return sr.(*systemResolvers)
|
|
}
|
|
|
|
func TestSystemResolvers_Refresh(t *testing.T) {
|
|
t.Run("expected_error", func(t *testing.T) {
|
|
sr := createTestSystemResolvers(t, nil)
|
|
|
|
assert.NoError(t, sr.refresh())
|
|
})
|
|
|
|
t.Run("unexpected_error", func(t *testing.T) {
|
|
_, err := NewSystemResolvers(func() string {
|
|
return "127.0.0.1::123"
|
|
})
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestSystemResolvers_DialFunc(t *testing.T) {
|
|
imp := createTestSystemResolversImpl(t, nil)
|
|
|
|
testCases := []struct {
|
|
want error
|
|
name string
|
|
address string
|
|
}{{
|
|
want: errFakeDial,
|
|
name: "valid_ipv4",
|
|
address: "127.0.0.1",
|
|
}, {
|
|
want: errFakeDial,
|
|
name: "valid_ipv6_port",
|
|
address: "[::1]:53",
|
|
}, {
|
|
want: errFakeDial,
|
|
name: "valid_ipv6_zone_port",
|
|
address: "[::1%lo0]:53",
|
|
}, {
|
|
want: errBadAddrPassed,
|
|
name: "invalid_split_host",
|
|
address: "127.0.0.1::123",
|
|
}, {
|
|
want: errUnexpectedHostFormat,
|
|
name: "invalid_ipv6_zone_port",
|
|
address: "[::1%%lo0]:53",
|
|
}, {
|
|
want: errBadAddrPassed,
|
|
name: "invalid_parse_ip",
|
|
address: "not-ip",
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
conn, err := imp.dialFunc(context.Background(), "", tc.address)
|
|
require.Nil(t, conn)
|
|
|
|
assert.ErrorIs(t, err, tc.want)
|
|
})
|
|
}
|
|
}
|