From 8c2b8f0df9f94446e0d0816db9ecffd0086406b4 Mon Sep 17 00:00:00 2001
From: Greg Hendrickson <greg@hndrx.co>
Date: Tue, 27 Jan 2026 20:56:31 +0000
Subject: [PATCH] Fix test assertions - mode is 4th arg not 3rd

---
 tests/test_ssh_server.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_ssh_server.py b/tests/test_ssh_server.py
index 2a94bd0..07a3893 100644
--- a/tests/test_ssh_server.py
+++ b/tests/test_ssh_server.py
@@ -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: