hass-core/tests/components/axis/test_switch.py

"""Axis switch platform tests."""

from unittest.mock import call as mock_call, Mock

from homeassistant import config_entries
from homeassistant.components import axis
from homeassistant.setup import async_setup_component

import homeassistant.components.switch as switch

EVENTS = [
    {
        "operation": "Initialized",
        "topic": "tns1:Device/Trigger/Relay",
        "source": "RelayToken",
        "source_idx": "0",
        "type": "LogicalState",
        "value": "inactive",
    },
    {
        "operation": "Initialized",
        "topic": "tns1:Device/Trigger/Relay",
        "source": "RelayToken",
        "source_idx": "1",
        "type": "LogicalState",
        "value": "active",
    },
]

ENTRY_CONFIG = {
    axis.CONF_DEVICE: {
        axis.config_flow.CONF_HOST: "1.2.3.4",
        axis.config_flow.CONF_USERNAME: "user",
        axis.config_flow.CONF_PASSWORD: "pass",
        axis.config_flow.CONF_PORT: 80,
    },
    axis.config_flow.CONF_MAC: "1234ABCD",
    axis.config_flow.CONF_MODEL: "model",
    axis.config_flow.CONF_NAME: "model 0",
}

ENTRY_OPTIONS = {
    axis.CONF_CAMERA: False,
    axis.CONF_EVENTS: True,
    axis.CONF_TRIGGER_TIME: 0,
}


async def setup_device(hass):
    """Load the Axis switch platform."""
    from axis import AxisDevice

    loop = Mock()

    config_entry = config_entries.ConfigEntry(
        1,
        axis.DOMAIN,
        "Mock Title",
        ENTRY_CONFIG,
        "test",
        config_entries.CONN_CLASS_LOCAL_PUSH,
        system_options={},
        options=ENTRY_OPTIONS,
    )
    device = axis.AxisNetworkDevice(hass, config_entry)
    device.api = AxisDevice(loop=loop, **config_entry.data[axis.CONF_DEVICE])
    hass.data[axis.DOMAIN] = {device.serial: device}
    device.api.enable_events(event_callback=device.async_event_callback)

    await hass.config_entries.async_forward_entry_setup(config_entry, "switch")
    # To flush out the service call to update the group
    await hass.async_block_till_done()

    return device


async def test_platform_manually_configured(hass):
    """Test that nothing happens when platform is manually configured."""
    assert await async_setup_component(
        hass, switch.DOMAIN, {"switch": {"platform": axis.DOMAIN}}
    )

    assert axis.DOMAIN not in hass.data


async def test_no_switches(hass):
    """Test that no output events in Axis results in no switch entities."""
    await setup_device(hass)

    assert not hass.states.async_entity_ids("switch")


async def test_switches(hass):
    """Test that switches are loaded properly."""
    device = await setup_device(hass)
    device.api.vapix.ports = {"0": Mock(), "1": Mock()}
    device.api.vapix.ports["0"].name = "Doorbell"
    device.api.vapix.ports["1"].name = ""

    for event in EVENTS:
        device.api.stream.event.manage_event(event)
    await hass.async_block_till_done()

    assert len(hass.states.async_all()) == 3

    relay_0 = hass.states.get("switch.model_0_doorbell")
    assert relay_0.state == "off"
    assert relay_0.name == "model 0 Doorbell"

    relay_1 = hass.states.get("switch.model_0_relay_1")
    assert relay_1.state == "on"
    assert relay_1.name == "model 0 Relay 1"

    device.api.vapix.ports["0"].action = Mock()

    await hass.services.async_call(
        "switch", "turn_on", {"entity_id": "switch.model_0_doorbell"}, blocking=True
    )

    await hass.services.async_call(
        "switch", "turn_off", {"entity_id": "switch.model_0_doorbell"}, blocking=True
    )

    assert device.api.vapix.ports["0"].action.call_args_list == [
        mock_call("/"),
        mock_call("\\"),
    ]