mirrors/AdGuardHome
1
0
Fork 0
mirror of https://github.com/AdguardTeam/AdGuardHome.git synced 2024-12-18 13:01:51 +03:00
Code Issues Projects Releases Packages Wiki Activity

dnsforward: imp tests

Browse source
This commit is contained in:
Stanislav Chzhen 2024-11-22 20:08:28 +03:00
parent 82851aa825
commit 1ca353423c
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
internal/dnsforward/dnsforward_test.go
View file

@ -500,7 +500,10 @@ func TestServerRace(t *testing.T) {
} }
func TestSafeSearch(t *testing.T) { func TestSafeSearch(t *testing.T) {
var upsCalledCounter uint32 var (
upsCalledCounter uint32
wantCNAMECounter uint32
)
safeSearchConf := filtering.SafeSearchConfig{ safeSearchConf := filtering.SafeSearchConfig{
Enabled: true, Enabled: true,
@ -615,6 +618,8 @@ func TestSafeSearch(t *testing.T) {
a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[1]) a := testutil.RequireTypeAssert[*dns.A](t, reply.Answer[1])
assert.NotEmpty(t, a.A) assert.NotEmpty(t, a.A)
atomic.AddUint32(&wantCNAMECounter, 1)
} else { } else {
require.Len(t, reply.Answer, 1) require.Len(t, reply.Answer, 1)
@ -625,7 +630,10 @@ func TestSafeSearch(t *testing.T) {
} }
t.Run("mock_upstream_is_used", func(t *testing.T) { t.Run("mock_upstream_is_used", func(t *testing.T) {
assert.Positive(t, atomic.LoadUint32(&upsCalledCounter)) called := atomic.LoadUint32(&upsCalledCounter)
want := atomic.LoadUint32(&wantCNAMECounter)
assert.Equal(t, want, called)
}) })
} }