File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed
Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -9,3 +9,4 @@ python_binance.egg-info/
99* .egg-info /
1010.idea /
1111venv * /
12+ .vscode
Original file line number Diff line number Diff line change 88from binance .exceptions import BinanceWebsocketUnableToConnect
99from websockets import WebSocketClientProtocol # type: ignore
1010from websockets .protocol import State
11+ import asyncio
1112
1213try :
1314 from unittest .mock import AsyncMock # Python 3.8+
@@ -175,7 +176,7 @@ async def test_connect_fails_to_connect_on_enter_context():
175176async def test_connect_fails_to_connect_after_disconnect ():
176177 # Create mock WebSocket client
177178 mock_socket = create_autospec (WebSocketClientProtocol )
178- mock_socket .recv = AsyncMock (return_value = '{"e": "value"}' )
179+ mock_socket .recv = AsyncMock (side_effect = delayed_return )
179180 mock_socket .state = AsyncMock ()
180181
181182 # Create mock connect that succeeds first, then fails
@@ -200,3 +201,8 @@ async def test_connect_fails_to_connect_after_disconnect():
200201 # After retrying to reconnect, receive BinanceWebsocketUnableToConnect
201202 assert msg ["e" ] == "error"
202203 assert msg ["type" ] == "BinanceWebsocketUnableToConnect"
204+
205+
206+ async def delayed_return ():
207+ await asyncio .sleep (0.1 ) # 100 ms delay
208+ return '{"e": "value"}'
You can’t perform that action at this time.
0 commit comments