From fcd80dd972f1c7687050af3fb42d4d88f7af24b6 Mon Sep 17 00:00:00 2001
From: Conrad Irwin <conrad.irwin@gmail.com>
Date: Thu, 15 Jan 2026 11:39:06 -0700
Subject: [PATCH] Restore run_with_clock_advancement (#46930)

Release Notes:

- N/A
---
 crates/scheduler/src/test_scheduler.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/crates/scheduler/src/test_scheduler.rs b/crates/scheduler/src/test_scheduler.rs
index fcc6e5c1e7d567cc5157cf703467aea472ff3379..e69d3419cf46f8707bd2aa12d9a448dda62e547e 100644
--- a/crates/scheduler/src/test_scheduler.rs
+++ b/crates/scheduler/src/test_scheduler.rs
@@ -177,6 +177,12 @@ impl TestScheduler {
         }
     }
 
+    pub fn run_with_clock_advancement(&self) {
+        while self.step() || self.advance_clock_to_next_timer() {
+            // Continue until no work remains
+        }
+    }
+
     /// Execute one tick of the scheduler, processing expired timers and running
     /// at most one task. Returns true if any work was done.
     ///