Skip to content

Commit 6cf09d1

Browse files
committed
Assert switch_change at each test
To be able to assert a not switch_change
1 parent 24dbd2c commit 6cf09d1

File tree

1 file changed

+21
-12
lines changed

1 file changed

+21
-12
lines changed

tests/test_smile.py

Lines changed: 21 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -419,8 +419,6 @@ async def tinker_switch(
419419
switch_change = await smile.set_switch_state(
420420
dev_id, members, model, new_state
421421
)
422-
assert switch_change
423-
_LOGGER.info(" + worked as intended")
424422
except (
425423
pw_exceptions.ErrorSendingCommandError,
426424
pw_exceptions.ResponseError,
@@ -430,6 +428,7 @@ async def tinker_switch(
430428
else: # pragma: no cover
431429
_LOGGER.info(" - failed unexpectedly")
432430
raise self.UnexpectedError
431+
return switch_change
433432

434433
@pytest.mark.asyncio
435434
async def tinker_thermostat(self, smile, loc_id, good_schemas=None, unhappy=False):
@@ -968,7 +967,8 @@ async def test_connect_adam_plus_anna(self):
968967
await self.tinker_thermostat(
969968
smile, "009490cc2f674ce6b576863fbb64f867", good_schemas=["Weekschema"]
970969
)
971-
await self.tinker_switch(smile, "aa6b0002df0a46e1b1eb94beb61eddfe")
970+
switch_change = await self.tinker_switch(smile, "aa6b0002df0a46e1b1eb94beb61eddfe")
971+
assert switch_change
972972
await smile.close_connection()
973973
await self.disconnect(server, client)
974974

@@ -979,9 +979,10 @@ async def test_connect_adam_plus_anna(self):
979979
good_schemas=["Weekschema"],
980980
unhappy=True,
981981
)
982-
await self.tinker_switch(
982+
switch_change = await self.tinker_switch(
983983
smile, "aa6b0002df0a46e1b1eb94beb61eddfe", unhappy=True
984984
)
985+
assert switch_change
985986
await smile.close_connection()
986987
await self.disconnect(server, client)
987988

@@ -1010,18 +1011,22 @@ async def test_connect_adam_plus_anna_new(self):
10101011
await self.device_test(smile, testdata)
10111012
assert smile.active_device_present
10121013

1013-
await self.tinker_switch(
1014+
switch_change = await self.tinker_switch(
10141015
smile,
10151016
"b83f9f9758064c0fab4af6578cba4c6d",
10161017
["aa6b0002df0a46e1b1eb94beb61eddfe", "f2be121e4a9345ac83c6e99ed89a98be"],
10171018
)
1018-
await self.tinker_switch(
1019+
assert switch_change
1020+
switch_change = await self.tinker_switch(
10191021
smile, "2743216f626f43948deec1f7ab3b3d70", model="dhw_cm_switch"
10201022
)
1021-
await self.tinker_switch(
1023+
assert switch_change
1024+
switch_change = await self.tinker_switch(
10221025
smile, "40ec6ebe67844b21914c4a5382a3f09f", model="lock"
10231026
)
1024-
await self.tinker_switch(smile, "f2be121e4a9345ac83c6e99ed89a98be" )
1027+
assert switch_change
1028+
switch_change = await self.tinker_switch(smile, "f2be121e4a9345ac83c6e99ed89a98be" )
1029+
assert not switch_change
10251030
await smile.close_connection()
10261031
await self.disconnect(server, client)
10271032

@@ -1091,7 +1096,8 @@ async def test_connect_adam_zone_per_device(self):
10911096
await self.tinker_thermostat(
10921097
smile, "82fa13f017d240daa0d0ea1775420f24", good_schemas=["CV Jessie"]
10931098
)
1094-
await self.tinker_switch(smile, "675416a629f343c495449970e2ca37b5")
1099+
switch_change = await self.tinker_switch(smile, "675416a629f343c495449970e2ca37b5")
1100+
assert switch_change
10951101
await smile.close_connection()
10961102
await self.disconnect(server, client)
10971103

@@ -1185,7 +1191,8 @@ async def test_connect_adam_multiple_devices_per_zone(self):
11851191
await self.tinker_thermostat(
11861192
smile, "82fa13f017d240daa0d0ea1775420f24", good_schemas=["CV Jessie"]
11871193
)
1188-
await self.tinker_switch(smile, "675416a629f343c495449970e2ca37b5")
1194+
switch_change = await self.tinker_switch(smile, "675416a629f343c495449970e2ca37b5")
1195+
assert switch_change
11891196
await smile.close_connection()
11901197
await self.disconnect(server, client)
11911198

@@ -1486,12 +1493,14 @@ async def test_connect_stretch_v23(self):
14861493
_LOGGER.info(" # Assert no master thermostat")
14871494
assert smile.single_master_thermostat() is None # it's not a thermostat :)
14881495

1489-
await self.tinker_switch(smile, "2587a7fcdd7e482dab03fda256076b4b")
1490-
await self.tinker_switch(
1496+
switch_change = await self.tinker_switch(smile, "2587a7fcdd7e482dab03fda256076b4b")
1497+
assert switch_change
1498+
switch_change = await self.tinker_switch(
14911499
smile,
14921500
"f7b145c8492f4dd7a4de760456fdef3e",
14931501
["407aa1c1099d463c9137a3a9eda787fd"],
14941502
)
1503+
assert switch_change
14951504

14961505
smile.get_all_devices()
14971506
await self.device_test(smile, testdata)

0 commit comments

Comments
 (0)