"""deCONZ sensor platform tests.""" from copy import deepcopy from homeassistant.components.deconz.const import CONF_ALLOW_CLIP_SENSOR from homeassistant.components.deconz.gateway import get_gateway_from_config_entry from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN from homeassistant.const import ( DEVICE_CLASS_BATTERY, DEVICE_CLASS_ILLUMINANCE, DEVICE_CLASS_POWER, STATE_UNAVAILABLE, ) from .test_gateway import DECONZ_WEB_REQUEST, setup_deconz_integration 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", }, } async def test_no_sensors(hass, aioclient_mock): """Test that no sensors in deconz results in no sensor entities.""" await setup_deconz_integration(hass, aioclient_mock) assert len(hass.states.async_all()) == 0 async def test_sensors(hass, aioclient_mock): """Test successful creation of sensor entities.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = deepcopy(SENSORS) config_entry = await setup_deconz_integration( hass, aioclient_mock, get_state_response=data ) gateway = get_gateway_from_config_entry(hass, config_entry) assert len(hass.states.async_all()) == 5 light_level_sensor = hass.states.get("sensor.light_level_sensor") assert light_level_sensor.state == "999.8" assert light_level_sensor.attributes["device_class"] == DEVICE_CLASS_ILLUMINANCE assert hass.states.get("sensor.presence_sensor") is None assert hass.states.get("sensor.switch_1") is None assert hass.states.get("sensor.switch_1_battery_level") is None assert hass.states.get("sensor.switch_2") is None switch_2_battery_level = hass.states.get("sensor.switch_2_battery_level") assert switch_2_battery_level.state == "100" assert switch_2_battery_level.attributes["device_class"] == DEVICE_CLASS_BATTERY assert hass.states.get("sensor.daylight_sensor") is None power_sensor = hass.states.get("sensor.power_sensor") assert power_sensor.state == "6" assert power_sensor.attributes["device_class"] == DEVICE_CLASS_POWER consumption_sensor = hass.states.get("sensor.consumption_sensor") assert consumption_sensor.state == "0.002" assert "device_class" not in consumption_sensor.attributes assert hass.states.get("sensor.clip_light_level_sensor") is None # Event signals new light level state_changed_event = { "t": "event", "e": "changed", "r": "sensors", "id": "1", "state": {"lightlevel": 2000}, } gateway.api.event_handler(state_changed_event) assert hass.states.get("sensor.light_level_sensor").state == "1.6" # Event signals new battery level state_changed_event = { "t": "event", "e": "changed", "r": "sensors", "id": "4", "config": {"battery": 75}, } gateway.api.event_handler(state_changed_event) await hass.async_block_till_done() assert hass.states.get("sensor.switch_2_battery_level").state == "75" await hass.config_entries.async_unload(config_entry.entry_id) states = hass.states.async_all() assert len(hass.states.async_all()) == 5 for state in states: assert state.state == STATE_UNAVAILABLE await hass.config_entries.async_remove(config_entry.entry_id) await hass.async_block_till_done() assert len(hass.states.async_all()) == 0 async def test_allow_clip_sensors(hass, aioclient_mock): """Test that CLIP sensors can be allowed.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = deepcopy(SENSORS) config_entry = await setup_deconz_integration( hass, aioclient_mock, options={CONF_ALLOW_CLIP_SENSOR: True}, get_state_response=data, ) assert len(hass.states.async_all()) == 6 assert hass.states.get("sensor.clip_light_level_sensor").state == "999.8" # Disallow clip sensors hass.config_entries.async_update_entry( config_entry, options={CONF_ALLOW_CLIP_SENSOR: False} ) await hass.async_block_till_done() assert len(hass.states.async_all()) == 5 assert hass.states.get("sensor.clip_light_level_sensor") is None # Allow clip sensors hass.config_entries.async_update_entry( config_entry, options={CONF_ALLOW_CLIP_SENSOR: True} ) await hass.async_block_till_done() assert len(hass.states.async_all()) == 6 assert hass.states.get("sensor.clip_light_level_sensor") async def test_add_new_sensor(hass, aioclient_mock): """Test that adding a new sensor works.""" config_entry = await setup_deconz_integration(hass, aioclient_mock) gateway = get_gateway_from_config_entry(hass, config_entry) assert len(hass.states.async_all()) == 0 state_added_event = { "t": "event", "e": "added", "r": "sensors", "id": "1", "sensor": deepcopy(SENSORS["1"]), } gateway.api.event_handler(state_added_event) await hass.async_block_till_done() assert len(hass.states.async_all()) == 1 assert hass.states.get("sensor.light_level_sensor").state == "999.8" async def test_add_battery_later(hass, aioclient_mock): """Test that a sensor without an initial battery state creates a battery sensor once state exist.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = {"1": deepcopy(SENSORS["3"])} config_entry = await setup_deconz_integration( hass, aioclient_mock, get_state_response=data ) gateway = get_gateway_from_config_entry(hass, config_entry) remote = gateway.api.sensors["1"] assert len(hass.states.async_all()) == 0 assert len(gateway.events) == 1 assert len(remote._callbacks) == 2 # Event and battery tracker remote.update({"config": {"battery": 50}}) await hass.async_block_till_done() assert len(hass.states.async_all()) == 1 assert len(gateway.events) == 1 assert len(remote._callbacks) == 2 # Event and battery entity assert hass.states.get("sensor.switch_1_battery_level") async def test_special_danfoss_battery_creation(hass, aioclient_mock): """Test the special Danfoss battery creation works. Normally there should only be one battery sensor per device from deCONZ. With specific Danfoss devices each endpoint can report its own battery state. """ data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = { "1": { "config": { "battery": 70, "heatsetpoint": 2300, "offset": 0, "on": True, "reachable": True, "schedule": {}, "schedule_on": False, }, "ep": 1, "etag": "982d9acc38bee5b251e24a9be26558e4", "lastseen": "2021-02-15T12:23Z", "manufacturername": "Danfoss", "modelid": "0x8030", "name": "0x8030", "state": { "lastupdated": "2021-02-15T12:23:07.994", "on": False, "temperature": 2307, }, "swversion": "YYYYMMDD", "type": "ZHAThermostat", "uniqueid": "58:8e:81:ff:fe:00:11:22-01-0201", }, "2": { "config": { "battery": 86, "heatsetpoint": 2300, "offset": 0, "on": True, "reachable": True, "schedule": {}, "schedule_on": False, }, "ep": 2, "etag": "62f12749f9f51c950086aff37dd02b61", "lastseen": "2021-02-15T12:23Z", "manufacturername": "Danfoss", "modelid": "0x8030", "name": "0x8030", "state": { "lastupdated": "2021-02-15T12:23:22.399", "on": False, "temperature": 2316, }, "swversion": "YYYYMMDD", "type": "ZHAThermostat", "uniqueid": "58:8e:81:ff:fe:00:11:22-02-0201", }, "3": { "config": { "battery": 86, "heatsetpoint": 2350, "offset": 0, "on": True, "reachable": True, "schedule": {}, "schedule_on": False, }, "ep": 3, "etag": "f50061174bb7f18a3d95789bab8b646d", "lastseen": "2021-02-15T12:23Z", "manufacturername": "Danfoss", "modelid": "0x8030", "name": "0x8030", "state": { "lastupdated": "2021-02-15T12:23:25.466", "on": False, "temperature": 2337, }, "swversion": "YYYYMMDD", "type": "ZHAThermostat", "uniqueid": "58:8e:81:ff:fe:00:11:22-03-0201", }, "4": { "config": { "battery": 85, "heatsetpoint": 2300, "offset": 0, "on": True, "reachable": True, "schedule": {}, "schedule_on": False, }, "ep": 4, "etag": "eea97adf8ce1b971b8b6a3a31793f96b", "lastseen": "2021-02-15T12:23Z", "manufacturername": "Danfoss", "modelid": "0x8030", "name": "0x8030", "state": { "lastupdated": "2021-02-15T12:23:41.939", "on": False, "temperature": 2333, }, "swversion": "YYYYMMDD", "type": "ZHAThermostat", "uniqueid": "58:8e:81:ff:fe:00:11:22-04-0201", }, "5": { "config": { "battery": 83, "heatsetpoint": 2300, "offset": 0, "on": True, "reachable": True, "schedule": {}, "schedule_on": False, }, "ep": 5, "etag": "1f7cd1a5d66dc27ac5eb44b8c47362fb", "lastseen": "2021-02-15T12:23Z", "manufacturername": "Danfoss", "modelid": "0x8030", "name": "0x8030", "state": {"lastupdated": "none", "on": False, "temperature": 2325}, "swversion": "YYYYMMDD", "type": "ZHAThermostat", "uniqueid": "58:8e:81:ff:fe:00:11:22-05-0201", }, } await setup_deconz_integration(hass, aioclient_mock, get_state_response=data) assert len(hass.states.async_all()) == 10 assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 5 async def test_air_quality_sensor(hass, aioclient_mock): """Test successful creation of air quality sensor entities.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = { "0": { "config": {"on": True, "reachable": True}, "ep": 2, "etag": "c2d2e42396f7c78e11e46c66e2ec0200", "lastseen": "2020-11-20T22:48Z", "manufacturername": "BOSCH", "modelid": "AIR", "name": "Air quality", "state": { "airquality": "poor", "airqualityppb": 809, "lastupdated": "2020-11-20T22:48:00.209", }, "swversion": "20200402", "type": "ZHAAirQuality", "uniqueid": "00:12:4b:00:14:4d:00:07-02-fdef", } } await setup_deconz_integration(hass, aioclient_mock, get_state_response=data) assert len(hass.states.async_all()) == 1 air_quality = hass.states.get("sensor.air_quality") assert air_quality.state == "poor" async def test_time_sensor(hass, aioclient_mock): """Test successful creation of time sensor entities.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = { "0": { "config": {"battery": 40, "on": True, "reachable": True}, "ep": 1, "etag": "28e796678d9a24712feef59294343bb6", "lastseen": "2020-11-22T11:26Z", "manufacturername": "Danfoss", "modelid": "eTRV0100", "name": "Time", "state": { "lastset": "2020-11-19T08:07:08Z", "lastupdated": "2020-11-22T10:51:03.444", "localtime": "2020-11-22T10:51:01", "utc": "2020-11-22T10:51:01Z", }, "swversion": "20200429", "type": "ZHATime", "uniqueid": "cc:cc:cc:ff:fe:38:4d:b3-01-000a", } } await setup_deconz_integration(hass, aioclient_mock, get_state_response=data) assert len(hass.states.async_all()) == 2 time = hass.states.get("sensor.time") assert time.state == "2020-11-19T08:07:08Z" time_battery = hass.states.get("sensor.time_battery_level") assert time_battery.state == "40" async def test_unsupported_sensor(hass, aioclient_mock): """Test that unsupported sensors doesn't break anything.""" data = deepcopy(DECONZ_WEB_REQUEST) data["sensors"] = { "0": {"type": "not supported", "name": "name", "state": {}, "config": {}} } await setup_deconz_integration(hass, aioclient_mock, get_state_response=data) assert len(hass.states.async_all()) == 1 unsupported_sensor = hass.states.get("sensor.name") assert unsupported_sensor.state == "unknown"