sched: truthful executing marks in capacity test shortcuts; format wave files

src/wf_scheduling/resume_gate.py

@@ -124,5 +124,5 @@ class SchedulerResumeGate:
                     checkpoint_id=checkpoint_id,
                     now=service.clock(),
                 )
-            except (SecondOwnerError, OSError, ValueError):
+            except SecondOwnerError, OSError, ValueError:
                 return False

src/wf_scheduling/store.py

@@ -175,9 +175,7 @@ class FileScheduleStore:
             highest = -1
             for index, item in enumerate(entries):
                 ordinal = item.get("seq")
-                effective = (
-                    ordinal if isinstance(ordinal, int) else index
-                )
+                effective = ordinal if isinstance(ordinal, int) else index
                 if effective > highest:
                     highest = effective
             if record.seq is None:
@@ -310,9 +308,7 @@ class FileScheduleStore:
             last = page[-1]
             _, _, _, last_ordinal = page_keys[-1]
             iso = "" if last.resolved_at is None else last.resolved_at.isoformat()
-            next_cursor: str | None = (
-                f"{iso}|{last.occurrence_id}|{int(last_ordinal)}"
-            )
+            next_cursor: str | None = f"{iso}|{last.occurrence_id}|{int(last_ordinal)}"
         else:
             next_cursor = None
         return {

tests/scheduling/test_lifecycle.py

@@ -559,9 +559,7 @@ async def test_store_deployment_directory_contract(tmp_path: Path) -> None:
         directory.required_inputs("missing.deployment")
 
 
-async def _poll_two(
-    service: SchedulerService, intended: datetime
-) -> tuple[str, str]:
+async def _poll_two(service: SchedulerService, intended: datetime) -> tuple[str, str]:
     service.schedule_store.create_schedule(_sched_model("a", intended))
     service.schedule_store.create_schedule(_sched_model("b", intended))
     result = await service.poll_once(intended + timedelta(seconds=1))
@@ -644,9 +642,7 @@ async def test_live_settlement_failure_spares_healthy_sibling(
 
         def _flaky_settle(self: Any, run_id: str, state: Any, now: datetime) -> None:
             try:
-                owned_by_b = (
-                    self.run_store.get_admission(run_id).schedule_id == "b"
-                )
+                owned_by_b = self.run_store.get_admission(run_id).schedule_id == "b"
             except KeyError:
                 owned_by_b = False
             if owned_by_b and not fired["done"]:
@@ -707,11 +703,7 @@ async def test_live_torn_settlement_reconciles_genuine_result(
                 )
             except KeyError:
                 owned_by_b = False
-            if (
-                kwargs.get("kind") == "completed"
-                and owned_by_b
-                and not fired["done"]
-            ):
+            if kwargs.get("kind") == "completed" and owned_by_b and not fired["done"]:
                 fired["done"] = True
                 raise RuntimeError("injected history failure")
             original(self, **kwargs)

tests/scheduling/test_poll.py

@@ -193,7 +193,11 @@ def test_parallel_limits_and_interrupted_slots() -> None:
         sched.poll(t0)
         assert sched._task_load() == 1
         first = runs.list_runs()[0]
+        # A waiting interruption holds no task slot: production dispatch
+        # clears the executing mark when persisting the stopped result, so
+        # the simulated interruption clears it too.
         runs.save_run(first.model_copy(update={"status": StoredRunStatus.INTERRUPTED}))
+        runs.clear_executing(first.id)
         assert sched._task_load() == 0
         sched.poll(t0 + timedelta(minutes=5))
         assert len(runs.list_runs()) == 2
@@ -205,6 +209,7 @@ def test_parallel_limits_and_interrupted_slots() -> None:
         store.save_schedule(sched_model)
         second = [r for r in runs.list_runs() if r.id != first.id][0]
         runs.save_run(second.model_copy(update={"status": StoredRunStatus.INTERRUPTED}))
+        runs.clear_executing(second.id)
         sched.poll(t0 + timedelta(minutes=15))
         assert any(
             r["kind"] == "skipped-overlap" and "12:15" in str(r["resolved_at"])
@@ -287,6 +292,7 @@ def test_fairness_slow_schedule_not_starved(tmp_path: Path) -> None:
     )
     for run in runs.list_runs():
         runs.save_run(run.model_copy(update={"status": StoredRunStatus.COMPLETED}))
+        runs.clear_executing(run.id)
     sched.poll(t0 + timedelta(seconds=30))
     assert any(r["kind"] == "admitted" for r in _history(store, "slow"))
     sched.ownership.release()
@@ -390,9 +396,8 @@ def test_record_resumed_stopped_result_is_idempotent_and_attributed(
     tmp_path: Path,
 ) -> None:
     """Live resumed results record exactly once with schedule attribution (B3)."""
-    from wf_api.run_lifecycle import persist_admission
-
     from tests.scheduling.controlled import fixture_environment
+    from wf_api.run_lifecycle import persist_admission
 
     sched, store, runs, sources = _harness(tmp_path, script={"*": "interrupt"})
     t0 = ts(2026, 9, 8, 12, 0)

tests/scheduling/test_schedule_store.py

@@ -168,7 +168,9 @@ def test_history_pagination_over_resolved_utc(tmp_path: Path) -> None:
     assert "|" in str(page["next_cursor"])
     assert len(page["occurrences"]) == 2  # type: ignore[arg-type]
     second = store.list_occurrences(
-        "a", cursor=page["next_cursor"], limit=2  # type: ignore[arg-type]
+        "a",
+        cursor=page["next_cursor"],
+        limit=2,  # type: ignore[arg-type]
     )
     assert second["next_cursor"] is None
     assert len(second["occurrences"]) == 1  # type: ignore[arg-type]

tests/wf_server/test_scheduler_integration.py

@@ -1081,7 +1081,9 @@ async def test_scheduled_resume_rejected_when_capacity_saturated(
     _gate_open.clear()
     root = tmp_path / "store"
     server = build_local_static_workflow_server(root, extra_sources=_gate_sources())
-    await _seed(server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"])
+    await _seed(
+        server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"]
+    )
     await _seed(server, "gated", "gated.default", _gate_plan("gated"), ["ok"])
     store = FileScheduleStore(root)
     store.create_schedule(_one_shot("asker", "ask1.default", datetime.now(UTC)))
@@ -1277,7 +1279,9 @@ async def test_scheduled_resume_records_completion_live(tmp_path: Path) -> None:
     """
     root = tmp_path / "store"
     server = build_local_static_workflow_server(root)
-    await _seed(server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"])
+    await _seed(
+        server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"]
+    )
     store = FileScheduleStore(root)
     store.create_schedule(_one_shot("asker", "ask1.default", datetime.now(UTC)))
     service = _scheduler(server, capacity=2)
@@ -1398,7 +1402,9 @@ async def test_paused_and_deleted_schedules_keep_resume_history(
     """Pausing/deleting never suppresses retained resume history (B3)."""
     root = tmp_path / "store"
     server = build_local_static_workflow_server(root)
-    await _seed(server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"])
+    await _seed(
+        server, "ask1", "ask1.default", _single_interrupt_plan("ask1"), ["submitted"]
+    )
     store = FileScheduleStore(root)
     store.create_schedule(_one_shot("pausable", "ask1.default", datetime.now(UTC)))
     store.create_schedule(_one_shot("doomed", "ask1.default", datetime.now(UTC)))
@@ -1427,7 +1433,11 @@ async def test_paused_and_deleted_schedules_keep_resume_history(
                 run_id=rid, resume_payload={}, resume_outcome="submitted"
             )
             assert resumed["status"] == "completed"
-        assert await _kinds(root, "pausable") == ["admitted", "interrupted", "completed"]
+        assert await _kinds(root, "pausable") == [
+            "admitted",
+            "interrupted",
+            "completed",
+        ]
         assert await _kinds(root, "doomed") == ["admitted", "interrupted", "completed"]
     finally:
         await service.stop()

tests/wf_transport_rpc_http/test_schedules_rpc.py

@@ -306,7 +306,9 @@ async def test_rpc_schedule_occurrences_tied_entries_traverse_once(tmp_path) ->
         cursor: object = None
         while True:
             page = await client.list_schedule_occurrences(
-                schedule_id="s", cursor=cursor, limit=1  # type: ignore[arg-type]
+                schedule_id="s",
+                cursor=cursor,
+                limit=1,  # type: ignore[arg-type]
             )
             assert page["total"] == 2
             for row in page["occurrences"]: