Attempt to fix flaky GCP PubSub source test

quickwit-oss · Feb 9, 2024 · 69dbe66 · 69dbe66
1 parent e83bd52
commit 69dbe66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/quickwit/quickwit-indexing/src/source/gcp_pubsub_source.rs b/quickwit/quickwit-indexing/src/source/gcp_pubsub_source.rs
@@ -154,7 +154,7 @@ impl GcpPubSubSource {
     }
 
     fn should_exit(&self) -> bool {
-        self.backfill_mode_enabled && self.state.num_consecutive_empty_batches > 5
+        self.backfill_mode_enabled && self.state.num_consecutive_empty_batches >= 10
     }
 }
 
@@ -438,7 +438,7 @@ mod gcp_pubsub_emulator_tests {
             "num_bytes_processed": 54,
             "num_messages_processed": 6,
             "num_invalid_messages": 0,
-            "num_consecutive_empty_batches": 6,
+            "num_consecutive_empty_batches": 10,
         });
         assert_eq!(exit_state, expected_exit_state);
     }