@@ -21,10 +21,12 @@ async def test_async_listen_events(async_websocket_client: WebsocketClient) -> N
2121 await async_websocket_client .async_fire_event (
2222 "async_test_event" , message = "Triggered by async websocket client"
2323 )
24- async for _ , event in zip (range (1 ), events ):
24+ # Typing breaks when using zip in an async context, so break instead
25+ async for event in events :
2526 assert event .origin == "LOCAL"
2627 assert event .event_type == "async_test_event"
2728 assert event .data ["message" ] == "Triggered by async websocket client"
29+ break
2830
2931
3032def test_listen_trigger (websocket_client : WebsocketClient ) -> None :
@@ -50,8 +52,10 @@ async def test_async_listen_trigger(async_websocket_client: WebsocketClient) ->
5052 async with async_websocket_client .async_listen_trigger (
5153 "time" , at = future .strftime ("%H:%M:%S" )
5254 ) as triggers :
53- async for _ , trigger in zip (range (1 ), triggers ):
55+ # Typing breaks when using zip in an async context, so break instead
56+ async for trigger in triggers :
5457 assert trigger ["trigger" ]["platform" ] == "time"
5558 assert datetime .fromisoformat (
5659 trigger ["trigger" ]["now" ]
5760 ).timestamp () == pytest .approx (future .timestamp (), abs = 1 )
61+ break
0 commit comments