161 lines
4.8 KiB
Python
161 lines
4.8 KiB
Python
"""Axis switch platform tests."""
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.axis.const import DOMAIN as AXIS_DOMAIN
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_OFF,
|
|
STATE_ON,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .const import API_DISCOVERY_PORT_MANAGEMENT, NAME
|
|
|
|
|
|
async def test_platform_manually_configured(hass: HomeAssistant) -> None:
|
|
"""Test that nothing happens when platform is manually configured."""
|
|
assert await async_setup_component(
|
|
hass, SWITCH_DOMAIN, {SWITCH_DOMAIN: {"platform": AXIS_DOMAIN}}
|
|
)
|
|
|
|
assert AXIS_DOMAIN not in hass.data
|
|
|
|
|
|
async def test_no_switches(hass: HomeAssistant, setup_config_entry) -> None:
|
|
"""Test that no output events in Axis results in no switch entities."""
|
|
assert not hass.states.async_entity_ids(SWITCH_DOMAIN)
|
|
|
|
|
|
async def test_switches_with_port_cgi(
|
|
hass: HomeAssistant, setup_config_entry, mock_rtsp_event
|
|
) -> None:
|
|
"""Test that switches are loaded properly using port.cgi."""
|
|
device = hass.data[AXIS_DOMAIN][setup_config_entry.entry_id]
|
|
|
|
device.api.vapix.ports = {"0": AsyncMock(), "1": AsyncMock()}
|
|
device.api.vapix.ports["0"].name = "Doorbell"
|
|
device.api.vapix.ports["0"].open = AsyncMock()
|
|
device.api.vapix.ports["0"].close = AsyncMock()
|
|
device.api.vapix.ports["1"].name = ""
|
|
|
|
mock_rtsp_event(
|
|
topic="tns1:Device/Trigger/Relay",
|
|
data_type="LogicalState",
|
|
data_value="inactive",
|
|
source_name="RelayToken",
|
|
source_idx="0",
|
|
)
|
|
mock_rtsp_event(
|
|
topic="tns1:Device/Trigger/Relay",
|
|
data_type="LogicalState",
|
|
data_value="active",
|
|
source_name="RelayToken",
|
|
source_idx="1",
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
|
|
relay_1 = hass.states.get(f"{SWITCH_DOMAIN}.{NAME}_relay_1")
|
|
assert relay_1.state == STATE_ON
|
|
assert relay_1.name == f"{NAME} Relay 1"
|
|
|
|
entity_id = f"{SWITCH_DOMAIN}.{NAME}_doorbell"
|
|
|
|
relay_0 = hass.states.get(entity_id)
|
|
assert relay_0.state == STATE_OFF
|
|
assert relay_0.name == f"{NAME} Doorbell"
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
device.api.vapix.ports["0"].close.assert_called_once()
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
device.api.vapix.ports["0"].open.assert_called_once()
|
|
|
|
|
|
@pytest.mark.parametrize("api_discovery_items", [API_DISCOVERY_PORT_MANAGEMENT])
|
|
async def test_switches_with_port_management(
|
|
hass: HomeAssistant, setup_config_entry, mock_rtsp_event
|
|
) -> None:
|
|
"""Test that switches are loaded properly using port management."""
|
|
device = hass.data[AXIS_DOMAIN][setup_config_entry.entry_id]
|
|
|
|
device.api.vapix.ports = {"0": AsyncMock(), "1": AsyncMock()}
|
|
device.api.vapix.ports["0"].name = "Doorbell"
|
|
device.api.vapix.ports["0"].open = AsyncMock()
|
|
device.api.vapix.ports["0"].close = AsyncMock()
|
|
device.api.vapix.ports["1"].name = ""
|
|
|
|
mock_rtsp_event(
|
|
topic="tns1:Device/Trigger/Relay",
|
|
data_type="LogicalState",
|
|
data_value="inactive",
|
|
source_name="RelayToken",
|
|
source_idx="0",
|
|
)
|
|
mock_rtsp_event(
|
|
topic="tns1:Device/Trigger/Relay",
|
|
data_type="LogicalState",
|
|
data_value="active",
|
|
source_name="RelayToken",
|
|
source_idx="1",
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
|
|
relay_1 = hass.states.get(f"{SWITCH_DOMAIN}.{NAME}_relay_1")
|
|
assert relay_1.state == STATE_ON
|
|
assert relay_1.name == f"{NAME} Relay 1"
|
|
|
|
entity_id = f"{SWITCH_DOMAIN}.{NAME}_doorbell"
|
|
|
|
relay_0 = hass.states.get(entity_id)
|
|
assert relay_0.state == STATE_OFF
|
|
assert relay_0.name == f"{NAME} Doorbell"
|
|
|
|
# State update
|
|
|
|
mock_rtsp_event(
|
|
topic="tns1:Device/Trigger/Relay",
|
|
data_type="LogicalState",
|
|
data_value="active",
|
|
source_name="RelayToken",
|
|
source_idx="0",
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get(f"{SWITCH_DOMAIN}.{NAME}_relay_1").state == STATE_ON
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
SERVICE_TURN_ON,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
device.api.vapix.ports["0"].close.assert_called_once()
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN,
|
|
SERVICE_TURN_OFF,
|
|
{ATTR_ENTITY_ID: entity_id},
|
|
blocking=True,
|
|
)
|
|
device.api.vapix.ports["0"].open.assert_called_once()
|