128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
"""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("\\"),
|
|
]
|