From fc69dec062db8f197af7a5ec6f2d7ecf9d69e2a9 Mon Sep 17 00:00:00 2001 From: Ashwanth Date: Mon, 30 Jun 2025 19:36:04 +0530 Subject: [PATCH] chore(tailer): fix a flaky tests in tail/querier_test.go (#18282) --- pkg/querier/tail/querier_test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pkg/querier/tail/querier_test.go b/pkg/querier/tail/querier_test.go index b0f5d113b0..8818b16be8 100644 --- a/pkg/querier/tail/querier_test.go +++ b/pkg/querier/tail/querier_test.go @@ -60,6 +60,7 @@ func TestQuerier_Tail_QueryTimeoutConfigFlag(t *testing.T) { } ingester.On("Tail", mock.Anything, &request).Return(clients, nil) ingester.On("TailersCount", mock.Anything).Return([]uint32{0}, nil) + ingester.On("TailDisconnectedIngesters", mock.Anything, mock.Anything, mock.Anything).Return(map[string]logproto.Querier_TailClient{}, nil).Maybe() // Setup log selector mock logSelector := newMockTailLogSelector() @@ -153,6 +154,7 @@ func TestQuerier_concurrentTailLimits(t *testing.T) { } ingester.On("Tail", mock.Anything, &request).Return(clients, testData.ingesterError) ingester.On("TailersCount", mock.Anything).Return([]uint32{testData.tailersCount}, testData.ingesterError) + ingester.On("TailDisconnectedIngesters", mock.Anything, mock.Anything, mock.Anything).Return(map[string]logproto.Querier_TailClient{}, nil).Maybe() // Setup log selector mock logSelector := newMockTailLogSelector()