Light control support to Axis devices (#36611)

* IR light support to Axis devices

* Change how to read light state

* Add tests

* Bump dependency to v32

* Assert variables passed to set_intensity

tests/components/axis/test_light.py

@@ -0,0 +1,150 @@
+"""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()