@@ -329,18 +329,20 @@ async def device_test(self, smile=pw_smile.Smile, testdata=None):
329329 assert data [measure_key ] == measure_assert
330330
331331 @pytest .mark .asyncio
332- async def tinker_relay (self , smile , dev_ids = None , members = None , unhappy = False ):
333- """Switch a relay on and off to test functionality."""
334- _LOGGER .info ("Asserting modifying settings for relay devices:" )
332+ async def tinker_switch (
333+ self , smile , dev_ids = None , members = None , model = None , unhappy = False
334+ ):
335+ """Turn a Switch on and off to test functionality."""
336+ _LOGGER .info ("Asserting modifying settings for switch devices:" )
335337 for dev_id in dev_ids :
336338 _LOGGER .info ("- Devices (%s):" , dev_id )
337339 for new_state in [False , True , False ]:
338340 _LOGGER .info ("- Switching %s" , new_state )
339341 try :
340- relay_change = await smile .set_relay_state (
341- dev_id , members , new_state
342+ switch_change = await smile .set_switch_state (
343+ dev_id , members , model , new_state
342344 )
343- assert relay_change
345+ assert switch_change
344346 _LOGGER .info (" + worked as intended" )
345347 except (
346348 pw_exceptions .ErrorSendingCommandError ,
@@ -878,7 +880,7 @@ async def test_connect_adam_plus_anna(self):
878880 await self .tinker_thermostat (
879881 smile , "009490cc2f674ce6b576863fbb64f867" , good_schemas = ["Weekschema" ]
880882 )
881- await self .tinker_relay (smile , ["aa6b0002df0a46e1b1eb94beb61eddfe" ])
883+ await self .tinker_switch (smile , ["aa6b0002df0a46e1b1eb94beb61eddfe" ])
882884 await smile .close_connection ()
883885 await self .disconnect (server , client )
884886
@@ -889,7 +891,7 @@ async def test_connect_adam_plus_anna(self):
889891 good_schemas = ["Weekschema" ],
890892 unhappy = True ,
891893 )
892- await self .tinker_relay (
894+ await self .tinker_switch (
893895 smile , ["aa6b0002df0a46e1b1eb94beb61eddfe" ], unhappy = True
894896 )
895897 await smile .close_connection ()
@@ -913,12 +915,14 @@ async def test_connect_adam_plus_anna_new(self):
913915 _LOGGER .info (" # Assert version" )
914916 assert smile .smile_version [0 ] == "3.2.4"
915917
916- await self .tinker_relay (
918+ await self .tinker_switch (
917919 smile ,
918920 ["b83f9f9758064c0fab4af6578cba4c6d" ],
919921 ["aa6b0002df0a46e1b1eb94beb61eddfe" , "f2be121e4a9345ac83c6e99ed89a98be" ],
920922 )
921-
923+ await self .tinker_switch (
924+ smile , ["2743216f626f43948deec1f7ab3b3d70" ], model = "dhw_cm_switch"
925+ )
922926 await self .device_test (smile , testdata )
923927 await smile .close_connection ()
924928 await self .disconnect (server , client )
@@ -986,7 +990,7 @@ async def test_connect_adam_zone_per_device(self):
986990 await self .tinker_thermostat (
987991 smile , "82fa13f017d240daa0d0ea1775420f24" , good_schemas = ["CV Jessie" ]
988992 )
989- await self .tinker_relay (smile , ["675416a629f343c495449970e2ca37b5" ])
993+ await self .tinker_switch (smile , ["675416a629f343c495449970e2ca37b5" ])
990994 await smile .close_connection ()
991995 await self .disconnect (server , client )
992996
@@ -1069,7 +1073,7 @@ async def test_connect_adam_multiple_devices_per_zone(self):
10691073 await self .tinker_thermostat (
10701074 smile , "82fa13f017d240daa0d0ea1775420f24" , good_schemas = ["CV Jessie" ]
10711075 )
1072- await self .tinker_relay (smile , ["675416a629f343c495449970e2ca37b5" ])
1076+ await self .tinker_switch (smile , ["675416a629f343c495449970e2ca37b5" ])
10731077 await smile .close_connection ()
10741078 await self .disconnect (server , client )
10751079
@@ -1360,7 +1364,7 @@ async def test_connect_stretch_v23(self):
13601364 _LOGGER .info (" # Assert legacy" )
13611365 assert smile ._smile_legacy # pylint: disable=protected-access
13621366
1363- await self .tinker_relay (smile , ["2587a7fcdd7e482dab03fda256076b4b" ])
1367+ await self .tinker_switch (smile , ["2587a7fcdd7e482dab03fda256076b4b" ])
13641368
13651369 await self .device_test (smile , testdata )
13661370
0 commit comments