Drain inflight requests after early stop

2026-04-25 16:57:01 +08:00
parent 2dc2815620
commit 2d7ebe50ee
3 changed files with 59 additions and 12 deletions
--- a/tests/test_core_flow.py
+++ b/tests/test_core_flow.py
@@ -2191,6 +2191,7 @@ class CoreFlowTests(unittest.TestCase):
                    max_lag_s=None,
                    max_elapsed_s=None,
                    evaluate_outcome=fake_evaluate,
+                    drain_inflight_on_early_stop=False,
                )

        self.assertEqual(submitted, ["r0", "r1"])
@@ -2254,6 +2255,7 @@ class CoreFlowTests(unittest.TestCase):
                        max_lag_s=None,
                        max_elapsed_s=1.0,
                        evaluate_outcome=fake_evaluate,
+                        drain_inflight_on_early_stop=False,
                    )

        self.assertEqual(submitted, ["r0"])