test: update mock assertions for conversation_history kwarg

The /branch and /resume flush tests asserted the old positional-only
call signature. Update to match the fix from #68480.
This commit is contained in:
kshitijk4poor 2026-07-22 09:49:41 +05:30 committed by kshitij
parent a46fbafe31
commit e57918ac80
2 changed files with 4 additions and 2 deletions

View file

@ -421,6 +421,7 @@ class TestResumeFlushesBeforeEndSession:
cli_obj._handle_resume_command("/resume target")
agent._flush_messages_to_session_db.assert_called_once_with(
[{"role": "user", "content": "hello"}, {"role": "assistant", "content": "hi"}]
[{"role": "user", "content": "hello"}, {"role": "assistant", "content": "hi"}],
conversation_history=[{"role": "user", "content": "hello"}, {"role": "assistant", "content": "hi"}],
)
cli_obj._session_db.end_session.assert_called_once()