@@ -792,12 +792,13 @@ async def test_node_relay_and_power(self, monkeypatch: pytest.MonkeyPatch) -> No
792792 # Test blocked async switching due to relay-lock active
793793 await stick .nodes ["0098765432101234" ].set_relay_lock (True )
794794 assert stick .nodes ["0098765432101234" ].relay_lock
795+ assert stick .nodes ["0098765432101234" ].relay_lock .lock_state
795796 with pytest .raises (pw_exceptions .NodeError ):
796797 await stick .nodes ["0098765432101234" ].set_relay (True )
797798 assert not stick .nodes ["0098765432101234" ].relay
798799 # Make sure to turn lock off for further testing
799800 await stick .nodes ["0098765432101234" ].set_relay_lock (False )
800- assert not stick .nodes ["0098765432101234" ].relay_lock
801+ assert not stick .nodes ["0098765432101234" ].relay_lock . lock_state
801802
802803 # Test async switching back from off to on
803804 self .test_relay_state_on = asyncio .Future ()
@@ -2512,7 +2513,7 @@ async def test_node_discovery_and_load(
25122513 assert state [pw_api .NodeFeature .INFO ].version == "070073"
25132514
25142515 assert state [pw_api .NodeFeature .RELAY ].state
2515- assert not state [pw_api .NodeFeature .RELAY_LOCK ]
2516+ assert not state [pw_api .NodeFeature .RELAY_LOCK ]. lock_state
25162517
25172518 # Check 1111111111111111
25182519 get_state_timestamp = dt .now (UTC ).replace (minute = 0 , second = 0 , microsecond = 0 )
@@ -2549,7 +2550,7 @@ async def test_node_discovery_and_load(
25492550 )
25502551 assert state [pw_api .NodeFeature .AVAILABLE ].state
25512552 assert state [pw_api .NodeFeature .RELAY ].state
2552- assert not state [pw_api .NodeFeature .RELAY_LOCK ]
2553+ assert not state [pw_api .NodeFeature .RELAY_LOCK ]. lock_state
25532554
25542555
25552556 # region Scan
0 commit comments