151 lines
4.6 KiB
Python
151 lines
4.6 KiB
Python
"""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()
|