fix unit test

prebid · Feb 24, 2025 · 3241c6f · 3241c6f
1 parent ce2e6c2
commit 3241c6f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/analytics/pubxai/pubxai_module_test.go b/analytics/pubxai/pubxai_module_test.go
@@ -197,10 +197,11 @@ func TestPubxaiModule_start(t *testing.T) {
 
 			if tt.expectUpdate {
 				mockAuctionBidsQueue.On("UpdateConfig", tt.newConfig.BufferInterval, tt.newConfig.BufferSize).Return(nil).Run(func(args mock.Arguments) {
-					// Signal done after the update is complete
 					done <- struct{}{}
 				})
-				mockWinBidsQueue.On("UpdateConfig", tt.newConfig.BufferInterval, tt.newConfig.BufferSize).Return(nil)
+				mockWinBidsQueue.On("UpdateConfig", tt.newConfig.BufferInterval, tt.newConfig.BufferSize).Return(nil).Run(func(args mock.Arguments) {
+					done <- struct{}{}
+				})
 			}
 
 			go p.start(configChan)