From 50cd1190ef2c87f9ae72e5360fe8227abcfc9fcd Mon Sep 17 00:00:00 2001 From: Teknium <127238744+teknium1@users.noreply.github.com> Date: Mon, 7 Sep 2026 07:18:12 -0700 Subject: [PATCH] test: pass snapshot listings to the catch-up restart budget probe Main now snapshots systemd unit listings before stopping old processes and passes them into _restart_systemd_gateway_units_best_effort; the catch-up budget test and eval call the new two-argument shape while still asserting the unit's stop+start budget reaches the client timeout. --- evals/update_unit_client_budget.py | 2 +- tests/hermes_cli/test_update_unit_client_budget.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/evals/update_unit_client_budget.py b/evals/update_unit_client_budget.py index 5297b154c1..ec5cd1633b 100644 --- a/evals/update_unit_client_budget.py +++ b/evals/update_unit_client_budget.py @@ -43,7 +43,7 @@ try: try: if catchup: failed = [] - fleet._restart_systemd_gateway_units_best_effort(failed) + fleet._restart_systemd_gateway_units_best_effort(failed, list(fleet._systemd_gateway_unit_listings())) record.update(failed_units=failed) else: result = _systemctl_reset_and_restart(cmd, unit) diff --git a/tests/hermes_cli/test_update_unit_client_budget.py b/tests/hermes_cli/test_update_unit_client_budget.py index ef61c1953e..c9db5169f6 100644 --- a/tests/hermes_cli/test_update_unit_client_budget.py +++ b/tests/hermes_cli/test_update_unit_client_budget.py @@ -30,7 +30,7 @@ def test_unit_transaction_budget_preserves_scope_and_health(monkeypatch, gracefu if catchup: monkeypatch.setattr(fleet, "_SYSTEMD_SCOPES", (("user", scope),)) failed = [] - fleet._restart_systemd_gateway_units_best_effort(failed) + fleet._restart_systemd_gateway_units_best_effort(failed, list(fleet._systemd_gateway_unit_listings())) assert not failed assert sum("restart" in cmd for cmd, _ in calls) == 1 return