Skip to content

Commit 39cabed

Browse files
mj23000GrandMoff100
authored andcommitted
Use breaks instead of zip(range(1), ...) in async event tests
1 parent 827a39b commit 39cabed

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

tests/test_events.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

3032
def 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

Comments
 (0)