886d8bd6e2
* Improve binary sensor tests * Fix sensor tests * Improve readability of binary sensor * Fix climate tests Fix sensor platform not loading climate devices as sensors * Add test to verify adding new sensor after start up
277 lines
9.1 KiB
Python
277 lines
9.1 KiB
Python
"""deCONZ sensor platform tests."""
|
|
from copy import deepcopy
|
|
|
|
from asynctest import patch
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components import deconz
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
import homeassistant.components.sensor as sensor
|
|
|
|
|
|
SENSORS = {
|
|
"1": {
|
|
"id": "Light sensor id",
|
|
"name": "Light level sensor",
|
|
"type": "ZHALightLevel",
|
|
"state": {"lightlevel": 30000, "dark": False},
|
|
"config": {"on": True, "reachable": True, "temperature": 10},
|
|
"uniqueid": "00:00:00:00:00:00:00:00-00",
|
|
},
|
|
"2": {
|
|
"id": "Presence sensor id",
|
|
"name": "Presence sensor",
|
|
"type": "ZHAPresence",
|
|
"state": {"presence": False},
|
|
"config": {},
|
|
"uniqueid": "00:00:00:00:00:00:00:01-00",
|
|
},
|
|
"3": {
|
|
"id": "Switch 1 id",
|
|
"name": "Switch 1",
|
|
"type": "ZHASwitch",
|
|
"state": {"buttonevent": 1000},
|
|
"config": {},
|
|
"uniqueid": "00:00:00:00:00:00:00:02-00",
|
|
},
|
|
"4": {
|
|
"id": "Switch 2 id",
|
|
"name": "Switch 2",
|
|
"type": "ZHASwitch",
|
|
"state": {"buttonevent": 1000},
|
|
"config": {"battery": 100},
|
|
"uniqueid": "00:00:00:00:00:00:00:03-00",
|
|
},
|
|
"5": {
|
|
"id": "Daylight sensor id",
|
|
"name": "Daylight sensor",
|
|
"type": "Daylight",
|
|
"state": {"daylight": True, "status": 130},
|
|
"config": {},
|
|
"uniqueid": "00:00:00:00:00:00:00:04-00",
|
|
},
|
|
"6": {
|
|
"id": "Power sensor id",
|
|
"name": "Power sensor",
|
|
"type": "ZHAPower",
|
|
"state": {"current": 2, "power": 6, "voltage": 3},
|
|
"config": {"reachable": True},
|
|
"uniqueid": "00:00:00:00:00:00:00:05-00",
|
|
},
|
|
"7": {
|
|
"id": "Consumption id",
|
|
"name": "Consumption sensor",
|
|
"type": "ZHAConsumption",
|
|
"state": {"consumption": 2, "power": 6},
|
|
"config": {"reachable": True},
|
|
"uniqueid": "00:00:00:00:00:00:00:06-00",
|
|
},
|
|
"8": {
|
|
"id": "CLIP light sensor id",
|
|
"name": "CLIP light level sensor",
|
|
"type": "CLIPLightLevel",
|
|
"state": {"lightlevel": 30000},
|
|
"config": {"reachable": True},
|
|
"uniqueid": "00:00:00:00:00:00:00:07-00",
|
|
},
|
|
}
|
|
|
|
BRIDGEID = "0123456789"
|
|
|
|
ENTRY_CONFIG = {
|
|
deconz.config_flow.CONF_API_KEY: "ABCDEF",
|
|
deconz.config_flow.CONF_BRIDGEID: BRIDGEID,
|
|
deconz.config_flow.CONF_HOST: "1.2.3.4",
|
|
deconz.config_flow.CONF_PORT: 80,
|
|
}
|
|
|
|
DECONZ_CONFIG = {
|
|
"bridgeid": BRIDGEID,
|
|
"mac": "00:11:22:33:44:55",
|
|
"name": "deCONZ mock gateway",
|
|
"sw_version": "2.05.69",
|
|
"websocketport": 1234,
|
|
}
|
|
|
|
DECONZ_WEB_REQUEST = {"config": DECONZ_CONFIG}
|
|
|
|
|
|
async def setup_deconz_integration(hass, config, options, get_state_response):
|
|
"""Create the deCONZ gateway."""
|
|
config_entry = config_entries.ConfigEntry(
|
|
version=1,
|
|
domain=deconz.DOMAIN,
|
|
title="Mock Title",
|
|
data=config,
|
|
source="test",
|
|
connection_class=config_entries.CONN_CLASS_LOCAL_PUSH,
|
|
system_options={},
|
|
options=options,
|
|
entry_id="1",
|
|
)
|
|
|
|
with patch(
|
|
"pydeconz.DeconzSession.async_get_state", return_value=get_state_response
|
|
), patch("pydeconz.DeconzSession.start", return_value=True):
|
|
await deconz.async_setup_entry(hass, config_entry)
|
|
await hass.async_block_till_done()
|
|
|
|
hass.config_entries._entries.append(config_entry)
|
|
|
|
return hass.data[deconz.DOMAIN][config[deconz.CONF_BRIDGEID]]
|
|
|
|
|
|
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, sensor.DOMAIN, {"sensor": {"platform": deconz.DOMAIN}}
|
|
)
|
|
is True
|
|
)
|
|
assert deconz.DOMAIN not in hass.data
|
|
|
|
|
|
async def test_no_sensors(hass):
|
|
"""Test that no sensors in deconz results in no sensor entities."""
|
|
data = deepcopy(DECONZ_WEB_REQUEST)
|
|
gateway = await setup_deconz_integration(
|
|
hass, ENTRY_CONFIG, options={}, get_state_response=data
|
|
)
|
|
assert len(gateway.deconz_ids) == 0
|
|
assert len(hass.states.async_all()) == 0
|
|
|
|
|
|
async def test_sensors(hass):
|
|
"""Test successful creation of sensor entities."""
|
|
data = deepcopy(DECONZ_WEB_REQUEST)
|
|
data["sensors"] = deepcopy(SENSORS)
|
|
gateway = await setup_deconz_integration(
|
|
hass, ENTRY_CONFIG, options={}, get_state_response=data
|
|
)
|
|
assert "sensor.light_level_sensor" in gateway.deconz_ids
|
|
assert "sensor.presence_sensor" not in gateway.deconz_ids
|
|
assert "sensor.switch_1" not in gateway.deconz_ids
|
|
assert "sensor.switch_1_battery_level" not in gateway.deconz_ids
|
|
assert "sensor.switch_2" not in gateway.deconz_ids
|
|
assert "sensor.switch_2_battery_level" in gateway.deconz_ids
|
|
assert "sensor.daylight_sensor" in gateway.deconz_ids
|
|
assert "sensor.power_sensor" in gateway.deconz_ids
|
|
assert "sensor.consumption_sensor" in gateway.deconz_ids
|
|
assert "sensor.clip_light_level_sensor" not in gateway.deconz_ids
|
|
assert len(hass.states.async_all()) == 6
|
|
|
|
light_level_sensor = hass.states.get("sensor.light_level_sensor")
|
|
assert light_level_sensor.state == "999.8"
|
|
|
|
presence_sensor = hass.states.get("sensor.presence_sensor")
|
|
assert presence_sensor is None
|
|
|
|
switch_1 = hass.states.get("sensor.switch_1")
|
|
assert switch_1 is None
|
|
|
|
switch_1_battery_level = hass.states.get("sensor.switch_1_battery_level")
|
|
assert switch_1_battery_level is None
|
|
|
|
switch_2 = hass.states.get("sensor.switch_2")
|
|
assert switch_2 is None
|
|
|
|
switch_2_battery_level = hass.states.get("sensor.switch_2_battery_level")
|
|
assert switch_2_battery_level.state == "100"
|
|
|
|
daylight_sensor = hass.states.get("sensor.daylight_sensor")
|
|
assert daylight_sensor.state == "dawn"
|
|
|
|
power_sensor = hass.states.get("sensor.power_sensor")
|
|
assert power_sensor.state == "6"
|
|
|
|
consumption_sensor = hass.states.get("sensor.consumption_sensor")
|
|
assert consumption_sensor.state == "0.002"
|
|
|
|
gateway.api.sensors["1"].async_update({"state": {"lightlevel": 2000}})
|
|
gateway.api.sensors["4"].async_update({"config": {"battery": 75}})
|
|
await hass.async_block_till_done()
|
|
|
|
light_level_sensor = hass.states.get("sensor.light_level_sensor")
|
|
assert light_level_sensor.state == "1.6"
|
|
|
|
switch_2_battery_level = hass.states.get("sensor.switch_2_battery_level")
|
|
assert switch_2_battery_level.state == "75"
|
|
|
|
|
|
async def test_allow_clip_sensors(hass):
|
|
"""Test that CLIP sensors can be allowed."""
|
|
data = deepcopy(DECONZ_WEB_REQUEST)
|
|
data["sensors"] = deepcopy(SENSORS)
|
|
gateway = await setup_deconz_integration(
|
|
hass,
|
|
ENTRY_CONFIG,
|
|
options={deconz.gateway.CONF_ALLOW_CLIP_SENSOR: True},
|
|
get_state_response=data,
|
|
)
|
|
assert "sensor.light_level_sensor" in gateway.deconz_ids
|
|
assert "sensor.presence_sensor" not in gateway.deconz_ids
|
|
assert "sensor.switch_1" not in gateway.deconz_ids
|
|
assert "sensor.switch_1_battery_level" not in gateway.deconz_ids
|
|
assert "sensor.switch_2" not in gateway.deconz_ids
|
|
assert "sensor.switch_2_battery_level" in gateway.deconz_ids
|
|
assert "sensor.daylight_sensor" in gateway.deconz_ids
|
|
assert "sensor.power_sensor" in gateway.deconz_ids
|
|
assert "sensor.consumption_sensor" in gateway.deconz_ids
|
|
assert "sensor.clip_light_level_sensor" in gateway.deconz_ids
|
|
assert len(hass.states.async_all()) == 7
|
|
|
|
light_level_sensor = hass.states.get("sensor.light_level_sensor")
|
|
assert light_level_sensor.state == "999.8"
|
|
|
|
presence_sensor = hass.states.get("sensor.presence_sensor")
|
|
assert presence_sensor is None
|
|
|
|
switch_1 = hass.states.get("sensor.switch_1")
|
|
assert switch_1 is None
|
|
|
|
switch_1_battery_level = hass.states.get("sensor.switch_1_battery_level")
|
|
assert switch_1_battery_level is None
|
|
|
|
switch_2 = hass.states.get("sensor.switch_2")
|
|
assert switch_2 is None
|
|
|
|
switch_2_battery_level = hass.states.get("sensor.switch_2_battery_level")
|
|
assert switch_2_battery_level.state == "100"
|
|
|
|
daylight_sensor = hass.states.get("sensor.daylight_sensor")
|
|
assert daylight_sensor.state == "dawn"
|
|
|
|
power_sensor = hass.states.get("sensor.power_sensor")
|
|
assert power_sensor.state == "6"
|
|
|
|
consumption_sensor = hass.states.get("sensor.consumption_sensor")
|
|
assert consumption_sensor.state == "0.002"
|
|
|
|
clip_light_level_sensor = hass.states.get("sensor.clip_light_level_sensor")
|
|
assert clip_light_level_sensor.state == "999.8"
|
|
|
|
|
|
async def test_add_new_sensor(hass):
|
|
"""Test that adding a new sensor works."""
|
|
data = deepcopy(DECONZ_WEB_REQUEST)
|
|
gateway = await setup_deconz_integration(
|
|
hass, ENTRY_CONFIG, options={}, get_state_response=data
|
|
)
|
|
assert len(gateway.deconz_ids) == 0
|
|
|
|
state_added = {
|
|
"t": "event",
|
|
"e": "added",
|
|
"r": "sensors",
|
|
"id": "1",
|
|
"sensor": deepcopy(SENSORS["1"]),
|
|
}
|
|
gateway.api.async_event_handler(state_added)
|
|
await hass.async_block_till_done()
|
|
|
|
assert "sensor.light_level_sensor" in gateway.deconz_ids
|
|
|
|
light_level_sensor = hass.states.get("sensor.light_level_sensor")
|
|
assert light_level_sensor.state == "999.8"
|