From 455e1d63740e388800390c97e08676d9fc494392 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Mon, 7 Sep 2026 11:09:49 +0200 Subject: [PATCH] [api] Assert the switch frame count instead of reading for it separately --- tests/integration/test_api_empty_message_roundtrip.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/integration/test_api_empty_message_roundtrip.py b/tests/integration/test_api_empty_message_roundtrip.py index ec0c95ab48..3cd5b6cd9e 100644 --- a/tests/integration/test_api_empty_message_roundtrip.py +++ b/tests/integration/test_api_empty_message_roundtrip.py @@ -27,10 +27,11 @@ async def test_api_empty_message_roundtrip( await client.send_message(api_pb2.DeviceInfoRequest()) await client.read_until_frame(MESSAGE_TYPE_OF[api_pb2.DeviceInfoResponse]) await client.send_message(api_pb2.ListEntitiesRequest()) - await client.read_until_frame( - MESSAGE_TYPE_OF[api_pb2.ListEntitiesSwitchResponse] - ) await client.read_until_frame(MESSAGE_TYPE_OF[api_pb2.ListEntitiesDoneResponse]) + assert ( + client.frame_counts[MESSAGE_TYPE_OF[api_pb2.ListEntitiesSwitchResponse]] + == 1 + ) await client.send_message(api_pb2.DisconnectRequest()) await client.read_until_frame(MESSAGE_TYPE_OF[api_pb2.DisconnectResponse])