async def test_async_set_levels(mock_aio_protocol,
caplog: pytest.LogCaptureFixture):
"""Test we can set levels."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\x33#\x25\x01\x10\x64\x00\x00\x00\x04\x00\xf0\x65")
await task
assert light.model_num == 0x33
assert light.dimmable_effects is False
transport.reset_mock()
with pytest.raises(ValueError):
# ValueError: RGBW command sent to non-RGBW devic
await light.async_set_levels(255, 255, 255, 255, 255)
await light.async_set_levels(255, 0, 0)
assert transport.mock_calls[0][0] == "write"
assert transport.mock_calls[0][1][0] == b"1\xff\x00\x00\x00\x00\x0f?"
async def test_async_set_brightness_rgbw(mock_aio_protocol):
"""Test we can set brightness with a rgbw only device."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\x25\x23\x61\x04\x10\xb6\x00\x98\x19\x04\x25\x0f\xdd")
await task
await light.async_stop()
await asyncio.sleep(0) # make sure nothing throws
transport.reset_mock()
await light.async_set_brightness(255)
assert transport.mock_calls[0][0] == "write"
assert transport.mock_calls[0][1][
0] == b"1\xff\x00\xd5\xff\xff\x00\x0f\x12"
assert light.brightness == 255
transport.reset_mock()
await light.async_set_brightness(128)
assert transport.mock_calls[0][0] == "write"
assert transport.mock_calls[0][1][0] == b"1\x80\x00k\x80\x80\x00\x0f+"
assert light.brightness == 128
async def test_handling_connection_lost(mock_aio_protocol):
"""Test we can reconnect."""
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
light._aio_protocol.connection_lost(None)
await asyncio.sleep(0) # make sure nothing throws
# Test we reconnect and can turn off
task = asyncio.create_task(light.async_turn_off())
# Wait for the future to get added
await asyncio.sleep(0.1) # wait for reconnect
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
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_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_reassemble(mock_aio_protocol):
"""Test we can reassemble."""
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
assert light.color_modes == {COLOR_MODE_RGBWW, COLOR_MODE_CCT}
assert light.protocol == PROTOCOL_LEDENET_9BYTE
assert light.model_num == 0x25
assert light.model == "RGB/WW/CW Controller (0x25)"
assert light.is_on is True
assert len(light.effect_list) == 21
light._aio_protocol.data_received(
b"\x81\x25\x23\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f\xde"
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
light._aio_protocol.data_received(b"\x81")
light._aio_protocol.data_received(
b"\x25\x23\x61\x05\x10\xb6\x00\x98\x19\x04\x25\x0f")
light._aio_protocol.data_received(b"\xde")
await asyncio.sleep(0)
assert light.is_on is True
async def test_cct_protocol_device(mock_aio_protocol):
"""Test a cct protocol device."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = await mock_aio_protocol()
light._aio_protocol.data_received(
b"\x81\x1C\x23\x61\x00\x05\x00\x64\x64\x64\x03\x64\x0F\xC8")
await task
assert light.getCCT() == (0, 255)
assert light.color_temp == 6500
assert light.brightness == 255
light._aio_protocol.data_received(
b"\x81\x1C\x23\x61\x00\x05\x00\x00\x00\x00\x03\x64\x00\x8D")
assert light.getCCT() == (255, 0)
assert light.color_temp == 2700
assert light.brightness == 255
transport.reset_mock()
await light.async_set_brightness(32)
assert transport.mock_calls[0][0] == "write"
assert (
transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x01\x00\t5\xb1\x00\r\x00\x00\x00\x03\xf6\xbe"
)
assert light.brightness == 33
transport.reset_mock()
await light.async_set_brightness(128)
assert transport.mock_calls[0][0] == "write"
assert (
transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x02\x00\t5\xb1\x002\x00\x00\x00\x03\x1b\t"
)
assert light.brightness == 128
transport.reset_mock()
await light.async_set_brightness(1)
assert transport.mock_calls[0][0] == "write"
assert (
transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x03\x00\t5\xb1\x00\x02\x00\x00\x00\x03\xeb\xaa"
)
assert light.brightness == 0
transport.reset_mock()
await light.async_set_levels(w=0, w2=255)
assert transport.mock_calls[0][0] == "write"
assert (
transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x04\x00\t5\xb1dd\x00\x00\x00\x03\xb17")
assert light.getCCT() == (0, 255)
assert light.color_temp == 6500
assert light.brightness == 255
async def test_async_set_custom_effect(mock_aio_protocol,
caplog: pytest.LogCaptureFixture):
"""Test we can set a custom effect."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = 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
assert light.model_num == 0x25
transport.reset_mock()
# no values
with pytest.raises(ValueError):
await light.async_set_custom_pattern([], 50, "jump")
await light.async_set_custom_pattern(
[
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 0),
(255, 0, 255),
(255, 0, 0),
(255, 0, 0),
],
50,
"jump",
)
assert transport.mock_calls[0][0] == "write"
assert (
transport.mock_calls[0][1][0] ==
b"Q\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\x00\x00\xff\x00\xff\x00\xff\x00\x00\x00\x10;\xff\x0f\x99"
)
async def test_shutdown(mock_aio_protocol):
"""Test we can shutdown."""
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
await light.async_stop()
await asyncio.sleep(0) # make sure nothing throws
async def test_invalid_initial_response(mock_aio_protocol):
"""Test we try switching protocol if we an unexpected response."""
light = AIOWifiLedBulb("192.168.1.166", timeout=0.1)
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = await mock_aio_protocol()
light._aio_protocol.data_received(b"\x31\x25")
with pytest.raises(RuntimeError):
await task
assert transport.mock_calls == [
call.get_extra_info("peername"),
call.write(bytearray(b"\x81\x8a\x8b\x96")),
call.write_eof(),
call.close(),
]
assert not light.available
async def test_handling_unavailable_after_no_response(mock_aio_protocol):
"""Test we handle the bulb not responding."""
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
await light.async_update()
await light.async_update()
await light.async_update()
await light.async_update()
with pytest.raises(RuntimeError):
await light.async_update()
assert light.available is False
async def test_no_initial_response(mock_aio_protocol):
"""Test we try switching protocol if we get no initial response."""
light = AIOWifiLedBulb("192.168.1.166", timeout=0.1)
assert light.protocol is None
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = await mock_aio_protocol()
with pytest.raises(RuntimeError):
await task
assert transport.mock_calls == [
call.get_extra_info("peername"),
call.write(bytearray(b"\x81\x8a\x8b\x96")),
call.write_eof(),
call.close(),
]
assert not light.available
assert light.protocol is PROTOCOL_LEDENET_ORIGINAL
async def test_async_set_effect(mock_aio_protocol,
caplog: pytest.LogCaptureFixture):
"""Test we can set an effect."""
light = AIOWifiLedBulb("192.168.1.166")
def _updated_callback(*args, **kwargs):
pass
task = asyncio.create_task(light.async_setup(_updated_callback))
transport, protocol = 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
assert light.model_num == 0xA3
assert light.dimmable_effects is True
transport.reset_mock()
await light.async_set_effect("random", 50)
assert transport.mock_calls[0][0] == "write"
assert transport.mock_calls[0][1][0].startswith(b"\xb0\xb1\xb2\xb3")
transport.reset_mock()
await light.async_set_effect("RBM 1", 50)
assert transport.mock_calls[0][0] == "write"
assert (transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x02\x00\x05B\x012d\x00\xa8")
assert light.effect == "RBM 1"
transport.reset_mock()
await light.async_set_brightness(255)
assert transport.mock_calls[0][0] == "write"
assert (transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x03\x00\x05B\x01\x10d\x00\x87")
transport.reset_mock()
await light.async_set_brightness(128)
assert transport.mock_calls[0][0] == "write"
assert (transport.mock_calls[0][1][0] ==
b"\xb0\xb1\xb2\xb3\x00\x01\x01\x04\x00\x05B\x01\x102\x00V")
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