async def test_data_async(event_loop):
"""Test data and subscription with async 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()
async_on_connect = AsyncMock()
async_on_data = AsyncMock()
async_on_disconnect = AsyncMock()
async_on_subscribed = AsyncMock()
client.websocket.async_on_connect(async_on_connect)
client.websocket.async_on_data(async_on_data)
client.websocket.async_on_disconnect(async_on_disconnect)
client.websocket.async_on_subscribed(async_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", "/")
async_on_connect.assert_called_once()
await client.websocket._sio._trigger_event("data", "/", {"foo": "bar"})
async_on_data.assert_called_once()
await client.websocket._sio._trigger_event("subscribed", "/", {"foo": "bar"})
async_on_subscribed.assert_called()
await client.websocket.disconnect()
await client.websocket._sio._trigger_event("disconnect", "/")
async_on_disconnect.assert_called_once()
client.websocket._sio.eio.disconnect.assert_called_once_with(abort=True)
async def test_data_async():
"""Test data and subscription with async 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}
async_on_connect = AsyncMock()
async_on_data = AsyncMock()
async_on_disconnect = AsyncMock()
async_on_subscribed = AsyncMock()
websocket.async_on_connect(async_on_connect)
websocket.async_on_data(async_on_data)
websocket.async_on_disconnect(async_on_disconnect)
websocket.async_on_subscribed(async_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", "/")
async_on_connect.assert_called_once()
await websocket._sio._trigger_event("data", "/", {"foo": "bar"})
async_on_data.assert_called_once()
await websocket._sio._trigger_event("subscribed", "/", {"foo": "bar"})
async_on_subscribed.assert_called()
await websocket.disconnect()
await websocket._sio._trigger_event("disconnect", "/")
async_on_disconnect.assert_called_once()
websocket._sio.disconnect.assert_called_once()
