core/tests/components/axis/test_light.py

"""Axis light platform tests."""

from copy import deepcopy

from homeassistant.components.axis.const import DOMAIN as AXIS_DOMAIN
from homeassistant.components.light import ATTR_BRIGHTNESS, DOMAIN as LIGHT_DOMAIN
from homeassistant.setup import async_setup_component

from .test_device import API_DISCOVERY_RESPONSE, NAME, setup_axis_integration

from tests.async_mock import patch

API_DISCOVERY_LIGHT_CONTROL = {
    "id": "light-control",
    "version": "1.1",
    "name": "Light Control",
}

EVENT_ON = {
    "operation": "Initialized",
    "topic": "tns1:Device/tnsaxis:Light/Status",
    "source": "id",
    "source_idx": "0",
    "type": "state",
    "value": "ON",
}

EVENT_OFF = {
    "operation": "Initialized",
    "topic": "tns1:Device/tnsaxis:Light/Status",
    "source": "id",
    "source_idx": "0",
    "type": "state",
    "value": "OFF",
}


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

    assert AXIS_DOMAIN not in hass.data


async def test_no_lights(hass):
    """Test that no light events in Axis results in no light entities."""
    await setup_axis_integration(hass)

    assert not hass.states.async_entity_ids(LIGHT_DOMAIN)


async def test_lights(hass):
    """Test that lights are loaded properly."""
    api_discovery = deepcopy(API_DISCOVERY_RESPONSE)
    api_discovery["data"]["apiList"].append(API_DISCOVERY_LIGHT_CONTROL)

    with patch.dict(API_DISCOVERY_RESPONSE, api_discovery):
        device = await setup_axis_integration(hass)

    # Add light
    with patch(
        "axis.light_control.LightControl.get_current_intensity",
        return_value={"data": {"intensity": 100}},
    ), patch(
        "axis.light_control.LightControl.get_valid_intensity",
        return_value={"data": {"ranges": [{"high": 150}]}},
    ):
        device.api.event.process_event(EVENT_ON)
        await hass.async_block_till_done()

    assert len(hass.states.async_entity_ids(LIGHT_DOMAIN)) == 1

    light_0 = hass.states.get(f"light.{NAME}_ir_light_0")
    assert light_0.state == "on"
    assert light_0.name == f"{NAME} IR Light 0"

    # Turn on, set brightness, light already on
    with patch(
        "axis.light_control.LightControl.activate_light"
    ) as mock_activate, patch(
        "axis.light_control.LightControl.set_manual_intensity"
    ) as mock_set_intensity, patch(
        "axis.light_control.LightControl.get_current_intensity",
        return_value={"data": {"intensity": 100}},
    ):
        await hass.services.async_call(
            LIGHT_DOMAIN,
            "turn_on",
            {"entity_id": f"light.{NAME}_ir_light_0", ATTR_BRIGHTNESS: 50},
            blocking=True,
        )
        mock_activate.not_called()
        mock_set_intensity.assert_called_once_with("led0", 29)

    # Turn off
    with patch(
        "axis.light_control.LightControl.deactivate_light"
    ) as mock_deactivate, patch(
        "axis.light_control.LightControl.get_current_intensity",
        return_value={"data": {"intensity": 100}},
    ):
        await hass.services.async_call(
            LIGHT_DOMAIN,
            "turn_off",
            {"entity_id": f"light.{NAME}_ir_light_0"},
            blocking=True,
        )
        mock_deactivate.assert_called_once()

    # Event turn off light
    device.api.event.process_event(EVENT_OFF)
    await hass.async_block_till_done()

    light_0 = hass.states.get(f"light.{NAME}_ir_light_0")
    assert light_0.state == "off"

    # Turn on, set brightness
    with patch(
        "axis.light_control.LightControl.activate_light"
    ) as mock_activate, patch(
        "axis.light_control.LightControl.set_manual_intensity"
    ) as mock_set_intensity, patch(
        "axis.light_control.LightControl.get_current_intensity",
        return_value={"data": {"intensity": 100}},
    ):
        await hass.services.async_call(
            LIGHT_DOMAIN,
            "turn_on",
            {"entity_id": f"light.{NAME}_ir_light_0"},
            blocking=True,
        )
        mock_activate.assert_called_once()
        mock_set_intensity.assert_not_called()

    # Turn off, light already off
    with patch(
        "axis.light_control.LightControl.deactivate_light"
    ) as mock_deactivate, patch(
        "axis.light_control.LightControl.get_current_intensity",
        return_value={"data": {"intensity": 100}},
    ):
        await hass.services.async_call(
            LIGHT_DOMAIN,
            "turn_off",
            {"entity_id": f"light.{NAME}_ir_light_0"},
            blocking=True,
        )
        mock_deactivate.assert_not_called()