Fix aggregation job driver test (#3618)

divviup · Jan 23, 2025 · f96fe5e · f96fe5e
1 parent d78ae1a
commit f96fe5e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/aggregator/src/aggregator/aggregation_job_driver/tests.rs b/aggregator/src/aggregator/aggregation_job_driver/tests.rs
@@ -4000,7 +4000,7 @@ async fn async_aggregation_job_init_poll_to_finished() {
     assert_eq!(want_report_aggregation, got_report_aggregation);
     assert_eq!(want_batch_aggregations, got_batch_aggregations);
 
-    assert_task_aggregation_counter(&ds, *task.id(), TaskAggregationCounter::new_with_values(0))
+    assert_task_aggregation_counter(&ds, *task.id(), TaskAggregationCounter::new_with_values(1))
         .await;
 }
 

diff --git a/aggregator/src/aggregator/test_util.rs b/aggregator/src/aggregator/test_util.rs
@@ -171,6 +171,8 @@ pub async fn assert_task_aggregation_counter(
 ) {
     // We can't coordinate with the counter-update tasks, so we loop on polling them.
 
+    sleep(Duration::from_millis(100)).await;
+
     let end_instant = Instant::now() + Duration::from_secs(10);
     loop {
         let now = Instant::now();