Fix test assertions - mode is 4th arg not 3rd

This commit is contained in:
Greg Hendrickson
2026-01-27 20:56:31 +00:00
parent 8a5e1785dc
commit 8c2b8f0df9

View File

@@ -62,10 +62,10 @@ class TestModeSelection:
mock_menu.return_value = None
await handle_client(process)
# Verify mode was 'play'
# Verify mode was 'play' (4th arg: process, session, username, mode)
mock_menu.assert_called_once()
call_args = mock_menu.call_args
assert call_args[0][2] == "play" # mode argument
assert call_args[0][3] == "play" # mode is 4th positional arg
@pytest.mark.asyncio
async def test_learn_mode(self):
@@ -85,7 +85,7 @@ class TestModeSelection:
await handle_client(process)
call_args = mock_menu.call_args
assert call_args[0][2] == "tutorial"
assert call_args[0][3] == "tutorial" # mode is 4th positional arg
@pytest.mark.asyncio
async def test_watch_mode(self):
@@ -105,7 +105,7 @@ class TestModeSelection:
await handle_client(process)
call_args = mock_menu.call_args
assert call_args[0][2] == "spectate"
assert call_args[0][3] == "spectate" # mode is 4th positional arg
class TestChessBoard: