chore(tailer): fix a flaky tests in tail/querier_test.go (#18282)

pull/18284/head
Ashwanth 6 months ago committed by GitHub
parent 0ee8e76ba2
commit fc69dec062
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      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()

Loading…
Cancel
Save