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