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.
This commit is contained in:
kshitijk4poor
2026-09-11 12:14:06 +05:30
committed by kshitij
parent c133095cc5
commit c71a411fe5
2 changed files with 2 additions and 2 deletions

View File

@@ -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

View File

@@ -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