From 69dbe664c9abc1ff719d898e99e6854d8f3618ea Mon Sep 17 00:00:00 2001 From: Adrien Guillo Date: Fri, 9 Feb 2024 14:45:49 -0500 Subject: [PATCH] Attempt to fix flaky GCP PubSub source test --- quickwit/quickwit-indexing/src/source/gcp_pubsub_source.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quickwit/quickwit-indexing/src/source/gcp_pubsub_source.rs b/quickwit/quickwit-indexing/src/source/gcp_pubsub_source.rs index e7413a0b391..14c48846aa6 100644 --- 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); }