mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-29 18:38:56 +03:00
3b2f5d7842
Merge in DNS/adguard-home from 2704-local-addresses-vol.1 to master Updates #2704. Updates #2829. Updates #2846. Squashed commit of the following: commit 9a49b3d27edcb30da7f16a065226907833b1dc81 Author: Eugene Burkov <e.burkov@adguard.com> Date: Mon Mar 22 15:39:17 2021 +0300 aghnet: imp docs and logging commit 74f95a29c55b9e732276601b0ecc63fb7c3a9f9e Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 20:56:51 2021 +0300 all: fix friday evening mistakes commit 0e2066bc5c16ed807fa601780b99e154502361a9 Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 20:51:15 2021 +0300 all: upd testify, imp code quality commit 8237c50b670c58361ccf7adec3ff2452b1196677 Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 20:19:29 2021 +0300 aghnet: imp test naming commit 14eb1e189339554c0a6d38e2ba7a93917774ebab Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 19:41:43 2021 +0300 aghnet: isolate windows-specific functionality commit d461ac8b18c187999da3e3aba116571b7ebe6785 Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 14:50:05 2021 +0300 aghnet: imp code quality commit d0ee01cb1f8613de2085c0f2f2f396e46beb52a5 Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Mar 19 11:59:10 2021 +0300 all: mv funcs to agherr, mk system resolvers getter
74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
// +build !windows
|
|
|
|
package aghnet
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func createTestSystemResolversImp(
|
|
t *testing.T,
|
|
refreshDur time.Duration,
|
|
hostGenFunc HostGenFunc,
|
|
) (imp *systemResolvers) {
|
|
t.Helper()
|
|
|
|
sr := createTestSystemResolvers(t, refreshDur, hostGenFunc)
|
|
|
|
var ok bool
|
|
imp, ok = sr.(*systemResolvers)
|
|
require.True(t, ok)
|
|
|
|
return imp
|
|
}
|
|
|
|
func TestSystemResolvers_Refresh(t *testing.T) {
|
|
t.Run("expected_error", func(t *testing.T) {
|
|
sr := createTestSystemResolvers(t, 0, nil)
|
|
|
|
assert.NoError(t, sr.Refresh())
|
|
})
|
|
|
|
t.Run("unexpected_error", func(t *testing.T) {
|
|
_, err := NewSystemResolvers(0, func() string {
|
|
return "127.0.0.1::123"
|
|
})
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestSystemResolvers_DialFunc(t *testing.T) {
|
|
imp := createTestSystemResolversImp(t, 0, nil)
|
|
|
|
testCases := []struct {
|
|
name string
|
|
address string
|
|
want error
|
|
}{{
|
|
name: "valid",
|
|
address: "127.0.0.1",
|
|
want: fakeDialErr,
|
|
}, {
|
|
name: "invalid_split_host",
|
|
address: "127.0.0.1::123",
|
|
want: badAddrPassedErr,
|
|
}, {
|
|
name: "invalid_parse_ip",
|
|
address: "not-ip",
|
|
want: badAddrPassedErr,
|
|
}}
|
|
|
|
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)
|
|
})
|
|
}
|
|
}
|