@@ -77,8 +77,7 @@ async def test_dyad_api_query_values(dyad_api: DyadApi, fake_channel: FakeChanne
7777 assert message .protocol == RoborockMessageProtocol .RPC_REQUEST
7878 assert message .version == b"A01"
7979 payload_data = json .loads (unpad (message .payload , AES .block_size ))
80- assert payload_data ["dps" ] == {"10000" : "[209, 201, 207, 214, 215, 227, 229, 230, 222, 224]" }
81- assert "t" in payload_data
80+ assert payload_data == {"dps" : {"10000" : "[209, 201, 207, 214, 215, 227, 229, 230, 222, 224]" }}
8281
8382
8483@pytest .mark .parametrize (
@@ -175,8 +174,7 @@ async def test_zeo_api_query_values(zeo_api: ZeoApi, fake_channel: FakeChannel):
175174 assert message .protocol == RoborockMessageProtocol .RPC_REQUEST
176175 assert message .version == b"A01"
177176 payload_data = json .loads (unpad (message .payload , AES .block_size ))
178- assert payload_data ["dps" ] == {"10000" : "[203, 207, 226, 227, 224, 218]" }
179- assert "t" in payload_data
177+ assert payload_data == {"dps" : {"10000" : "[203, 207, 226, 227, 224, 218]" }}
180178
181179
182180@pytest .mark .parametrize (
@@ -247,8 +245,7 @@ async def test_dyad_api_set_value(dyad_api: DyadApi, fake_channel: FakeChannel):
247245 # decode the payload to verify contents
248246 payload_data = json .loads (unpad (message .payload , AES .block_size ))
249247 # A01 protocol expects values to be strings in the dps dict
250- assert payload_data ["dps" ] == {"209" : 1 }
251- assert "t" in payload_data
248+ assert payload_data == {"dps" : {"209" : 1 }}
252249
253250
254251async def test_zeo_api_set_value (zeo_api : ZeoApi , fake_channel : FakeChannel ):
@@ -264,5 +261,4 @@ async def test_zeo_api_set_value(zeo_api: ZeoApi, fake_channel: FakeChannel):
264261 # decode the payload to verify contents
265262 payload_data = json .loads (unpad (message .payload , AES .block_size ))
266263 # A01 protocol expects values to be strings in the dps dict
267- assert payload_data ["dps" ] == {"204" : "standard" }
268- assert "t" in payload_data
264+ assert payload_data == {"dps" : {"204" : "standard" }}
0 commit comments