hass-core/tests/components/rfxtrx/test_light.py

"""The tests for the Rfxtrx light platform."""
from homeassistant.components import rfxtrx as rfxtrx_core
from homeassistant.setup import async_setup_component

from . import _signal_event

from tests.common import assert_setup_component


async def test_valid_config(hass, rfxtrx):
    """Test configuration."""
    with assert_setup_component(1):
        assert await async_setup_component(
            hass,
            "light",
            {
                "light": {
                    "platform": "rfxtrx",
                    "automatic_add": True,
                    "devices": {
                        "0b1100cd0213c7f210010f51": {
                            "name": "Test",
                            rfxtrx_core.ATTR_FIRE_EVENT: True,
                        }
                    },
                }
            },
        )

        assert await async_setup_component(
            hass,
            "light",
            {
                "light": {
                    "platform": "rfxtrx",
                    "automatic_add": True,
                    "devices": {
                        "0b1100cd0213c7f210010f51": {
                            "name": "Test",
                            "signal_repetitions": 3,
                        }
                    },
                }
            },
        )


async def test_default_config(hass, rfxtrx):
    """Test with 0 switches."""
    with assert_setup_component(1):
        await async_setup_component(
            hass, "light", {"light": {"platform": "rfxtrx", "devices": {}}}
        )
        await hass.async_block_till_done()

    assert 0 == len(rfxtrx_core.RFX_DEVICES)


async def test_one_light(hass, rfxtrx):
    """Test with 1 light."""
    await async_setup_component(
        hass,
        "light",
        {
            "light": {
                "platform": "rfxtrx",
                "devices": {"0b1100cd0213c7f210010f51": {"name": "Test"}},
            }
        },
    )
    await hass.async_block_till_done()

    import RFXtrx as rfxtrxmod

    hass.data[rfxtrx_core.DATA_RFXOBJECT] = rfxtrxmod.Core(
        "", transport_protocol=rfxtrxmod.DummyTransport
    )

    assert 1 == len(rfxtrx_core.RFX_DEVICES)
    entity = rfxtrx_core.RFX_DEVICES["213c7f2_16"]
    entity.hass = hass
    assert "Test" == entity.name
    assert "off" == entity.state
    assert entity.assumed_state
    assert entity.signal_repetitions == 1
    assert not entity.should_fire_event
    assert not entity.should_poll

    assert not entity.is_on

    entity.turn_on()
    assert entity.is_on
    assert entity.brightness == 255

    entity.turn_off()
    assert not entity.is_on
    assert entity.brightness == 0

    entity.turn_on(brightness=100)
    assert entity.is_on
    assert entity.brightness == 100

    entity.turn_on(brightness=10)
    assert entity.is_on
    assert entity.brightness == 10

    entity.turn_on(brightness=255)
    assert entity.is_on
    assert entity.brightness == 255

    entity.turn_off()
    assert "Test" == entity.name
    assert "off" == entity.state

    entity.turn_on()
    assert "on" == entity.state

    entity.turn_off()
    assert "off" == entity.state

    entity.turn_on(brightness=100)
    assert "on" == entity.state

    entity.turn_on(brightness=10)
    assert "on" == entity.state

    entity.turn_on(brightness=255)
    assert "on" == entity.state


async def test_several_lights(hass, rfxtrx):
    """Test with 3 lights."""
    await async_setup_component(
        hass,
        "light",
        {
            "light": {
                "platform": "rfxtrx",
                "signal_repetitions": 3,
                "devices": {
                    "0b1100cd0213c7f230010f71": {"name": "Test"},
                    "0b1100100118cdea02010f70": {"name": "Bath"},
                    "0b1100101118cdea02010f70": {"name": "Living"},
                },
            }
        },
    )
    await hass.async_block_till_done()

    assert 3 == len(rfxtrx_core.RFX_DEVICES)
    device_num = 0
    for id in rfxtrx_core.RFX_DEVICES:
        entity = rfxtrx_core.RFX_DEVICES[id]
        assert entity.signal_repetitions == 3
        if entity.name == "Living":
            device_num = device_num + 1
            assert "off" == entity.state
            assert "<Entity Living: off>" == entity.__str__()
        elif entity.name == "Bath":
            device_num = device_num + 1
            assert "off" == entity.state
            assert "<Entity Bath: off>" == entity.__str__()
        elif entity.name == "Test":
            device_num = device_num + 1
            assert "off" == entity.state
            assert "<Entity Test: off>" == entity.__str__()

    assert 3 == device_num


async def test_discover_light(hass, rfxtrx):
    """Test with discovery of lights."""
    await async_setup_component(
        hass,
        "light",
        {"light": {"platform": "rfxtrx", "automatic_add": True, "devices": {}}},
    )
    await hass.async_block_till_done()

    event = rfxtrx_core.get_rfx_object("0b11009e00e6116202020070")
    event.data = bytearray(b"\x0b\x11\x00\x9e\x00\xe6\x11b\x02\x02\x00p")

    await _signal_event(hass, event)
    entity = rfxtrx_core.RFX_DEVICES["0e61162_2"]
    assert 1 == len(rfxtrx_core.RFX_DEVICES)
    assert "<Entity 0b11009e00e6116202020070: on>" == entity.__str__()

    event = rfxtrx_core.get_rfx_object("0b11009e00e6116201010070")
    event.data = bytearray(b"\x0b\x11\x00\x9e\x00\xe6\x11b\x01\x01\x00p")

    await _signal_event(hass, event)
    assert 1 == len(rfxtrx_core.RFX_DEVICES)

    event = rfxtrx_core.get_rfx_object("0b1100120118cdea02020070")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x12, 0x01, 0x18, 0xCD, 0xEA, 0x02, 0x02, 0x00, 0x70]
    )

    await _signal_event(hass, event)
    entity = rfxtrx_core.RFX_DEVICES["118cdea_2"]
    assert 2 == len(rfxtrx_core.RFX_DEVICES)
    assert "<Entity 0b1100120118cdea02020070: on>" == entity.__str__()

    # trying to add a sensor
    event = rfxtrx_core.get_rfx_object("0a52085e070100b31b0279")
    event.data = bytearray(b"\nR\x08^\x07\x01\x00\xb3\x1b\x02y")
    await _signal_event(hass, event)
    assert 2 == len(rfxtrx_core.RFX_DEVICES)

    # trying to add a switch
    event = rfxtrx_core.get_rfx_object("0b1100100118cdea02010f70")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x10, 0x01, 0x18, 0xCD, 0xEA, 0x01, 0x01, 0x0F, 0x70]
    )

    await _signal_event(hass, event)
    assert 2 == len(rfxtrx_core.RFX_DEVICES)

    # Trying to add a rollershutter
    event = rfxtrx_core.get_rfx_object("0a1400adf394ab020e0060")
    event.data = bytearray(
        [0x0A, 0x14, 0x00, 0xAD, 0xF3, 0x94, 0xAB, 0x02, 0x0E, 0x00, 0x60]
    )
    await _signal_event(hass, event)
    assert 2 == len(rfxtrx_core.RFX_DEVICES)


async def test_discover_light_noautoadd(hass, rfxtrx):
    """Test with discover of light when auto add is False."""
    await async_setup_component(
        hass,
        "light",
        {"light": {"platform": "rfxtrx", "automatic_add": False, "devices": {}}},
    )
    await hass.async_block_till_done()

    event = rfxtrx_core.get_rfx_object("0b1100120118cdea02020070")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x12, 0x01, 0x18, 0xCD, 0xEA, 0x02, 0x02, 0x00, 0x70]
    )

    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)

    event = rfxtrx_core.get_rfx_object("0b1100120118cdea02010070")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x12, 0x01, 0x18, 0xCD, 0xEA, 0x02, 0x01, 0x00, 0x70]
    )

    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)

    event = rfxtrx_core.get_rfx_object("0b1100120118cdea02020070")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x12, 0x01, 0x18, 0xCD, 0xEA, 0x02, 0x02, 0x00, 0x70]
    )

    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)

    # Trying to add a sensor
    event = rfxtrx_core.get_rfx_object("0a52085e070100b31b0279")
    event.data = bytearray(b"\nR\x08^\x07\x01\x00\xb3\x1b\x02y")
    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)

    # Trying to add a switch
    event = rfxtrx_core.get_rfx_object("0b1100100118cdea02010f70")
    event.data = bytearray(
        [0x0B, 0x11, 0x00, 0x10, 0x01, 0x18, 0xCD, 0xEA, 0x01, 0x01, 0x0F, 0x70]
    )
    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)

    # Trying to add a rollershutter
    event = rfxtrx_core.get_rfx_object("0a1400adf394ab020e0060")
    event.data = bytearray(
        [0x0A, 0x14, 0x00, 0xAD, 0xF3, 0x94, 0xAB, 0x02, 0x0E, 0x00, 0x60]
    )
    await _signal_event(hass, event)
    assert 0 == len(rfxtrx_core.RFX_DEVICES)