async def test_data_sync(event_loop): """Test data and subscription with sync handlers.""" async with aiohttp.ClientSession(loop=event_loop) as session: client = Client(TEST_API_KEY, TEST_APP_KEY, session=session) client.websocket._sio.eio._trigger_event = AsyncMock() client.websocket._sio.eio.connect = AsyncMock() client.websocket._sio.eio.disconnect = AsyncMock() on_connect = MagicMock() on_data = MagicMock() on_disconnect = MagicMock() on_subscribed = MagicMock() client.websocket.on_connect(on_connect) client.websocket.on_data(on_data) client.websocket.on_disconnect(on_disconnect) client.websocket.on_subscribed(on_subscribed) await client.websocket.connect() client.websocket._sio.eio.connect.assert_called_once_with( f"https://api.ambientweather.net/?api=1&applicationKey={TEST_APP_KEY}", engineio_path="socket.io", headers={}, transports=["websocket"], ) await client.websocket._sio._trigger_event("connect", "/") on_connect.assert_called_once() await client.websocket._sio._trigger_event("data", "/", {"foo": "bar"}) on_data.assert_called_once() await client.websocket._sio._trigger_event("subscribed", "/", {"foo": "bar"}) on_subscribed.assert_called() await client.websocket.disconnect() await client.websocket._sio._trigger_event("disconnect", "/") on_disconnect.assert_called_once() client.websocket._sio.eio.disconnect.assert_called_once_with(abort=True)
async def test_data_sync(): """Test data and subscription with sync handlers.""" websocket = Websocket(TEST_API_KEY, TEST_APP_KEY) websocket._sio.connect = AsyncMock() websocket._sio.disconnect = AsyncMock() websocket._sio.eio._trigger_event = AsyncMock() websocket._sio.namespaces = {"/": 1} on_connect = MagicMock() on_data = MagicMock() on_disconnect = MagicMock() on_subscribed = MagicMock() websocket.on_connect(on_connect) websocket.on_data(on_data) websocket.on_disconnect(on_disconnect) websocket.on_subscribed(on_subscribed) await websocket.connect() websocket._sio.connect.assert_called_once_with( f"https://rt2.ambientweather.net/?api=1&applicationKey={TEST_APP_KEY}", transports=["websocket"], ) await websocket._sio._trigger_event("connect", "/") on_connect.assert_called_once() await websocket._sio._trigger_event("data", "/", {"foo": "bar"}) on_data.assert_called_once() await websocket._sio._trigger_event("subscribed", "/", {"foo": "bar"}) on_subscribed.assert_called() await websocket.disconnect() await websocket._sio._trigger_event("disconnect", "/") on_disconnect.assert_called_once() websocket._sio.disconnect.assert_called_once()