31b2f331db
* Improved sensors * Lib update signalling * Replace reason with changed * Move imports to top of file * Add support for secondary temperature reported by some Xiaomi devices * Bump dependency to v59
136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
"""deCONZ binary sensor platform tests."""
|
|
from unittest.mock import Mock, patch
|
|
|
|
from tests.common import mock_coro
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components import deconz
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
import homeassistant.components.binary_sensor as binary_sensor
|
|
|
|
|
|
SENSOR = {
|
|
"1": {
|
|
"id": "Sensor 1 id",
|
|
"name": "Sensor 1 name",
|
|
"type": "ZHAPresence",
|
|
"state": {"presence": False},
|
|
"config": {},
|
|
"uniqueid": "00:00:00:00:00:00:00:00-00"
|
|
},
|
|
"2": {
|
|
"id": "Sensor 2 id",
|
|
"name": "Sensor 2 name",
|
|
"type": "ZHATemperature",
|
|
"state": {"temperature": False},
|
|
"config": {}
|
|
}
|
|
}
|
|
|
|
|
|
ENTRY_CONFIG = {
|
|
deconz.const.CONF_ALLOW_CLIP_SENSOR: True,
|
|
deconz.const.CONF_ALLOW_DECONZ_GROUPS: True,
|
|
deconz.config_flow.CONF_API_KEY: "ABCDEF",
|
|
deconz.config_flow.CONF_BRIDGEID: "0123456789",
|
|
deconz.config_flow.CONF_HOST: "1.2.3.4",
|
|
deconz.config_flow.CONF_PORT: 80
|
|
}
|
|
|
|
|
|
async def setup_gateway(hass, data, allow_clip_sensor=True):
|
|
"""Load the deCONZ binary sensor platform."""
|
|
from pydeconz import DeconzSession
|
|
loop = Mock()
|
|
session = Mock()
|
|
|
|
ENTRY_CONFIG[deconz.const.CONF_ALLOW_CLIP_SENSOR] = allow_clip_sensor
|
|
|
|
config_entry = config_entries.ConfigEntry(
|
|
1, deconz.DOMAIN, 'Mock Title', ENTRY_CONFIG, 'test',
|
|
config_entries.CONN_CLASS_LOCAL_PUSH)
|
|
gateway = deconz.DeconzGateway(hass, config_entry)
|
|
gateway.api = DeconzSession(loop, session, **config_entry.data)
|
|
gateway.api.config = Mock()
|
|
hass.data[deconz.DOMAIN] = {gateway.bridgeid: gateway}
|
|
|
|
with patch('pydeconz.DeconzSession.async_get_state',
|
|
return_value=mock_coro(data)):
|
|
await gateway.api.async_load_parameters()
|
|
|
|
await hass.config_entries.async_forward_entry_setup(
|
|
config_entry, 'binary_sensor')
|
|
# To flush out the service call to update the group
|
|
await hass.async_block_till_done()
|
|
return gateway
|
|
|
|
|
|
async def test_platform_manually_configured(hass):
|
|
"""Test that we do not discover anything or try to set up a gateway."""
|
|
assert await async_setup_component(hass, binary_sensor.DOMAIN, {
|
|
'binary_sensor': {
|
|
'platform': deconz.DOMAIN
|
|
}
|
|
}) is True
|
|
assert deconz.DOMAIN not in hass.data
|
|
|
|
|
|
async def test_no_binary_sensors(hass):
|
|
"""Test that no sensors in deconz results in no sensor entities."""
|
|
data = {}
|
|
gateway = await setup_gateway(hass, data)
|
|
assert len(hass.data[deconz.DOMAIN][gateway.bridgeid].deconz_ids) == 0
|
|
assert len(hass.states.async_all()) == 0
|
|
|
|
|
|
async def test_binary_sensors(hass):
|
|
"""Test successful creation of binary sensor entities."""
|
|
data = {"sensors": SENSOR}
|
|
gateway = await setup_gateway(hass, data)
|
|
assert "binary_sensor.sensor_1_name" in gateway.deconz_ids
|
|
assert "binary_sensor.sensor_2_name" not in gateway.deconz_ids
|
|
assert len(hass.states.async_all()) == 1
|
|
|
|
hass.data[deconz.DOMAIN][gateway.bridgeid].api.sensors['1'].async_update(
|
|
{'state': {'on': False}})
|
|
|
|
|
|
async def test_add_new_sensor(hass):
|
|
"""Test successful creation of sensor entities."""
|
|
data = {}
|
|
gateway = await setup_gateway(hass, data)
|
|
sensor = Mock()
|
|
sensor.name = 'name'
|
|
sensor.type = 'ZHAPresence'
|
|
sensor.BINARY = True
|
|
sensor.register_async_callback = Mock()
|
|
async_dispatcher_send(
|
|
hass, gateway.async_event_new_device('sensor'), [sensor])
|
|
await hass.async_block_till_done()
|
|
assert "binary_sensor.name" in gateway.deconz_ids
|
|
|
|
|
|
async def test_do_not_allow_clip_sensor(hass):
|
|
"""Test that clip sensors can be ignored."""
|
|
data = {}
|
|
gateway = await setup_gateway(hass, data, allow_clip_sensor=False)
|
|
sensor = Mock()
|
|
sensor.name = 'name'
|
|
sensor.type = 'CLIPPresence'
|
|
sensor.register_async_callback = Mock()
|
|
async_dispatcher_send(
|
|
hass, gateway.async_event_new_device('sensor'), [sensor])
|
|
await hass.async_block_till_done()
|
|
assert len(gateway.deconz_ids) == 0
|
|
|
|
|
|
async def test_unload_switch(hass):
|
|
"""Test that it works to unload switch entities."""
|
|
data = {"sensors": SENSOR}
|
|
gateway = await setup_gateway(hass, data)
|
|
|
|
await gateway.async_reset()
|
|
|
|
assert len(hass.states.async_all()) == 0
|