@@ -46,7 +46,9 @@ async def test_legacy_query_param_valid_key():
4646 ws = _make_mock_websocket ()
4747 with patch ("openhands.agent_server.sockets.get_default_config" ) as mock_config :
4848 mock_config .return_value .session_api_keys = ["sk-oh-valid" ]
49- result = await _accept_authenticated_websocket (ws , session_api_key = "sk-oh-valid" )
49+ result = await _accept_authenticated_websocket (
50+ ws , session_api_key = "sk-oh-valid"
51+ )
5052
5153 assert result is True
5254 ws .accept .assert_called_once ()
@@ -58,7 +60,9 @@ async def test_legacy_query_param_invalid_key():
5860 ws = _make_mock_websocket ()
5961 with patch ("openhands.agent_server.sockets.get_default_config" ) as mock_config :
6062 mock_config .return_value .session_api_keys = ["sk-oh-valid" ]
61- result = await _accept_authenticated_websocket (ws , session_api_key = "sk-oh-wrong" )
63+ result = await _accept_authenticated_websocket (
64+ ws , session_api_key = "sk-oh-wrong"
65+ )
6266
6367 assert result is False
6468 ws .close .assert_called_once_with (code = 4001 , reason = "Authentication failed" )
@@ -202,8 +206,7 @@ async def test_events_socket_first_message_auth_e2e():
202206 from openhands .agent_server .sockets import events_socket
203207
204208 ws = _make_mock_websocket ()
205- # First call: auth message (read by _accept_authenticated_websocket via receive_text)
206- # Then receive_json calls: disconnect
209+ # Auth via receive_text, then receive_json raises disconnect.
207210 ws .receive_text .return_value = json .dumps (
208211 {"type" : "auth" , "session_api_key" : "sk-oh-valid" }
209212 )
@@ -220,9 +223,7 @@ async def test_events_socket_first_message_auth_e2e():
220223 patch ("openhands.agent_server.sockets.get_default_config" ) as mock_config ,
221224 ):
222225 mock_config .return_value .session_api_keys = ["sk-oh-valid" ]
223- mock_conv_service .get_event_service = AsyncMock (
224- return_value = mock_event_service
225- )
226+ mock_conv_service .get_event_service = AsyncMock (return_value = mock_event_service )
226227
227228 await events_socket (uuid4 (), ws , session_api_key = None )
228229
0 commit comments