From c71a411fe53397a6c3b045bcf90bd841b82b5eaa Mon Sep 17 00:00:00 2001
From: kshitijk4poor <82637225+kshitijk4poor@users.noreply.github.com>
Date: Fri, 11 Sep 2026 12:14:06 +0530
Subject: [PATCH] test(gateway): release-slot fakes accept the finalizer's
 run_generation kwarg

The turn finalizer now releases only its own generation
(`_release_running_agent_state(key, run_generation=...)`); two test doubles
were zero-kwarg lambdas and raised TypeError inside the finally.
---
 tests/gateway/test_gateway_command_dispatch_minimal.py | 2 +-
 tests/hermes_cli/test_pre_command_hook.py              | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/gateway/test_gateway_command_dispatch_minimal.py b/tests/gateway/test_gateway_command_dispatch_minimal.py
index d0a9b96593..fa8ad0a13b 100644
--- a/tests/gateway/test_gateway_command_dispatch_minimal.py
+++ b/tests/gateway/test_gateway_command_dispatch_minimal.py
@@ -98,7 +98,7 @@ def _make_runner():
     runner._should_send_telegram_lobby_reminder = lambda _source: False
     runner._check_slash_access = lambda _source, _command: None
     runner._begin_session_run_generation = lambda _key: 1
-    runner._release_running_agent_state = lambda key: runner._running_agents.pop(key, None)
+    runner._release_running_agent_state = lambda key, run_generation=None: runner._running_agents.pop(key, None)
     return runner, adapter
 
 
diff --git a/tests/hermes_cli/test_pre_command_hook.py b/tests/hermes_cli/test_pre_command_hook.py
index c3a4079f5d..1504354ca8 100644
--- a/tests/hermes_cli/test_pre_command_hook.py
+++ b/tests/hermes_cli/test_pre_command_hook.py
@@ -284,7 +284,7 @@ def _make_runner():
     runner._check_slash_access = lambda _source, _command: None
     runner._begin_session_run_generation = lambda _key: 1
     runner._release_running_agent_state = (
-        lambda key: runner._running_agents.pop(key, None)
+        lambda key, run_generation=None: runner._running_agents.pop(key, None)
     )
     return runner, adapter