async def test_turn_on_off_via_power_state_message(
mock_aio_protocol, caplog: pytest.LogCaptureFixture):
"""Test we can turn on and off via power state message."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\x25\x23\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xde")
await task
task = asyncio.create_task(light.async_turn_off())
# Wait for the future to get added
await asyncio.sleep(0)
light._ignore_next_power_state_update = False
light._aio_protocol.data_received(b"\x0F\x71\x24\xA4")
await asyncio.sleep(0)
assert light.is_on is False
await task
task = asyncio.create_task(light.async_turn_on())
await asyncio.sleep(0)
light._ignore_next_power_state_update = False
light._aio_protocol.data_received(b"\x0F\x71\x23\xA3")
await asyncio.sleep(0)
assert light.is_on is True
await task
async def test_turn_on_off_via_assessable_state_message(
mock_aio_protocol, caplog: pytest.LogCaptureFixture):
"""Test we can turn on and off via addressable state message."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\xA3#\x25\x01\x10\x64\x00\x00\x00\x04\x00\xf0\xd5")
await task
task = asyncio.create_task(light.async_turn_off())
# Wait for the future to get added
await asyncio.sleep(0)
light._ignore_next_power_state_update = False
light._aio_protocol.data_received(
b"\xB0\xB1\xB2\xB3\x00\x01\x01\x23\x00\x0E\x81\xA3\x24\x25\xFF\x47\x64\xFF\xFF\x00\x01\x00\x1E\x34\x61"
)
await asyncio.sleep(0)
assert light.is_on is False
await task
task = asyncio.create_task(light.async_turn_on())
await asyncio.sleep(0)
light._ignore_next_power_state_update = False
light._aio_protocol.data_received(
b"\xB0\xB1\xB2\xB3\x00\x01\x01\x24\x00\x0E\x81\xA3\x23\x25\x5F\x21\x64\xFF\xFF\x00\x01\x00\x1E\x6D\xD4"
)
await asyncio.sleep(0)
assert light.is_on is True
await task
async def test_turn_on_off(mock_aio_protocol,
caplog: pytest.LogCaptureFixture):
"""Test we can turn on and off."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\x25\x23\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xde")
await task
task = asyncio.create_task(light.async_turn_off())
# Wait for the future to get added
await asyncio.sleep(0)
light._aio_protocol.data_received(
b"\x81\x25\x24\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xdf")
await asyncio.sleep(0)
assert light.is_on is False
await task
task = asyncio.create_task(light.async_turn_on())
await asyncio.sleep(0)
light._aio_protocol.data_received(
b"\x81\x25\x23\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xde")
await asyncio.sleep(0)
assert light.is_on is True
await task
await asyncio.sleep(0)
caplog.clear()
caplog.set_level(logging.DEBUG)
# Handle the failure case
with patch.object(aiodevice, "POWER_STATE_TIMEOUT", 0.05):
await asyncio.create_task(light.async_turn_off())
assert light.is_on is True
assert "Failed to turn off (1/3)" in caplog.text
assert "Failed to turn off (2/3)" in caplog.text
assert "Failed to turn off (3/3)" in caplog.text
with patch.object(aiodevice, "POWER_STATE_TIMEOUT", 0.05):
task = asyncio.create_task(light.async_turn_off())
# Do NOT wait for the future to get added, we know the retry logic works
light._aio_protocol.data_received(
b"\x81\x25\x24\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xdf")
await asyncio.sleep(0)
assert light.is_on is False
await task
await asyncio.sleep(0)
caplog.clear()
caplog.set_level(logging.DEBUG)
# Handle the failure case
with patch.object(aiodevice, "POWER_STATE_TIMEOUT", 0.05):
await asyncio.create_task(light.async_turn_on())
assert light.is_on is False
assert "Failed to turn on (1/3)" in caplog.text
assert "Failed to turn on (2/3)" in caplog.text
assert "Failed to turn on (3/3)" in caplog.text