all: resync with master

internal/querylog/search_internal_test.go

@@ -0,0 +1,102 @@
+package querylog
+
+import (
+	"net"
+	"testing"
+	"time"
+
+	"github.com/AdguardTeam/golibs/logutil/slogutil"
+	"github.com/AdguardTeam/golibs/testutil"
+	"github.com/AdguardTeam/golibs/timeutil"
+	"github.com/miekg/dns"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestQueryLog_Search_findClient(t *testing.T) {
+	const knownClientID = "client-1"
+	const knownClientName = "Known Client 1"
+	const unknownClientID = "client-2"
+
+	knownClient := &Client{
+		Name: knownClientName,
+	}
+
+	findClientCalls := 0
+	findClient := func(ids []string) (c *Client, _ error) {
+		defer func() { findClientCalls++ }()
+
+		if len(ids) == 0 {
+			return nil, nil
+		}
+
+		if ids[0] == knownClientID {
+			return knownClient, nil
+		}
+
+		return nil, nil
+	}
+
+	l, err := newQueryLog(Config{
+		Logger:            slogutil.NewDiscardLogger(),
+		FindClient:        findClient,
+		BaseDir:           t.TempDir(),
+		RotationIvl:       timeutil.Day,
+		MemSize:           100,
+		Enabled:           true,
+		FileEnabled:       true,
+		AnonymizeClientIP: false,
+	})
+	require.NoError(t, err)
+
+	ctx := testutil.ContextWithTimeout(t, testTimeout)
+	testutil.CleanupAndRequireSuccess(t, func() (err error) {
+		return l.Shutdown(ctx)
+	})
+
+	q := &dns.Msg{
+		Question: []dns.Question{{
+			Name: "example.com",
+		}},
+	}
+
+	l.Add(&AddParams{
+		Question: q,
+		ClientID: knownClientID,
+		ClientIP: net.IP{1, 2, 3, 4},
+	})
+
+	// Add the same thing again to test the cache.
+	l.Add(&AddParams{
+		Question: q,
+		ClientID: knownClientID,
+		ClientIP: net.IP{1, 2, 3, 4},
+	})
+
+	l.Add(&AddParams{
+		Question: q,
+		ClientID: unknownClientID,
+		ClientIP: net.IP{1, 2, 3, 5},
+	})
+
+	sp := &searchParams{
+		// Add some time to the "current" one to protect against
+		// low-resolution timers on some Windows machines.
+		//
+		// TODO(a.garipov): Use some kind of timeSource interface
+		// instead of relying on time.Now() in tests.
+		olderThan: time.Now().Add(10 * time.Second),
+		limit:     3,
+	}
+	entries, _ := l.search(ctx, sp)
+	assert.Equal(t, 2, findClientCalls)
+
+	require.Len(t, entries, 3)
+
+	assert.Nil(t, entries[0].client)
+
+	gotClient := entries[2].client
+	require.NotNil(t, gotClient)
+
+	assert.Equal(t, knownClientName, gotClient.Name)
+}