314 lines
10 KiB
Python
314 lines
10 KiB
Python
"""Test the UniFi Protect sensor platform."""
|
|
# pylint: disable=protected-access
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
import pytest
|
|
from pyunifiprotect.data.base import WifiConnectionState, WiredConnectionState
|
|
from pyunifiprotect.data.devices import Camera, Sensor
|
|
from pyunifiprotect.data.nvr import NVR
|
|
|
|
from homeassistant.components.unifiprotect.const import DEFAULT_ATTRIBUTION
|
|
from homeassistant.components.unifiprotect.sensor import (
|
|
ALL_DEVICES_SENSORS,
|
|
CAMERA_DISABLED_SENSORS,
|
|
CAMERA_SENSORS,
|
|
NVR_DISABLED_SENSORS,
|
|
NVR_SENSORS,
|
|
SENSE_SENSORS,
|
|
)
|
|
from homeassistant.const import ATTR_ATTRIBUTION, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from .conftest import (
|
|
MockEntityFixture,
|
|
assert_entity_counts,
|
|
enable_entity,
|
|
ids_from_device_description,
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="sensor")
|
|
async def sensor_fixture(
|
|
hass: HomeAssistant,
|
|
mock_entry: MockEntityFixture,
|
|
mock_sensor: Sensor,
|
|
now: datetime,
|
|
):
|
|
"""Fixture for a single sensor for testing the sensor platform."""
|
|
|
|
# disable pydantic validation so mocking can happen
|
|
Sensor.__config__.validate_assignment = False
|
|
|
|
sensor_obj = mock_sensor.copy(deep=True)
|
|
sensor_obj._api = mock_entry.api
|
|
sensor_obj.name = "Test Sensor"
|
|
sensor_obj.battery_status.percentage = 10.0
|
|
sensor_obj.stats.light.value = 10.0
|
|
sensor_obj.stats.humidity.value = 10.0
|
|
sensor_obj.stats.temperature.value = 10.0
|
|
sensor_obj.up_since = now
|
|
sensor_obj.bluetooth_connection_state.signal_strength = -50.0
|
|
|
|
mock_entry.api.bootstrap.reset_objects()
|
|
mock_entry.api.bootstrap.sensors = {
|
|
sensor_obj.id: sensor_obj,
|
|
}
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# 2 from all, 4 from sense, 12 NVR
|
|
assert_entity_counts(hass, Platform.SENSOR, 18, 13)
|
|
|
|
yield sensor_obj
|
|
|
|
Sensor.__config__.validate_assignment = True
|
|
|
|
|
|
@pytest.fixture(name="camera")
|
|
async def camera_fixture(
|
|
hass: HomeAssistant,
|
|
mock_entry: MockEntityFixture,
|
|
mock_camera: Camera,
|
|
now: datetime,
|
|
):
|
|
"""Fixture for a single camera for testing the sensor platform."""
|
|
|
|
# disable pydantic validation so mocking can happen
|
|
Camera.__config__.validate_assignment = False
|
|
|
|
camera_obj = mock_camera.copy(deep=True)
|
|
camera_obj._api = mock_entry.api
|
|
camera_obj.channels[0]._api = mock_entry.api
|
|
camera_obj.channels[1]._api = mock_entry.api
|
|
camera_obj.channels[2]._api = mock_entry.api
|
|
camera_obj.name = "Test Camera"
|
|
camera_obj.wired_connection_state = WiredConnectionState(phy_rate=1000)
|
|
camera_obj.wifi_connection_state = WifiConnectionState(
|
|
signal_quality=100, signal_strength=-50
|
|
)
|
|
camera_obj.stats.rx_bytes = 100.0
|
|
camera_obj.stats.tx_bytes = 100.0
|
|
camera_obj.stats.video.recording_start = now
|
|
camera_obj.stats.storage.used = 100.0
|
|
camera_obj.stats.storage.used = 100.0
|
|
camera_obj.stats.storage.rate = 100.0
|
|
camera_obj.voltage = 20.0
|
|
|
|
mock_entry.api.bootstrap.reset_objects()
|
|
mock_entry.api.bootstrap.nvr.system_info.storage.devices = []
|
|
mock_entry.api.bootstrap.cameras = {
|
|
camera_obj.id: camera_obj,
|
|
}
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# 3 from all, 6 from camera, 12 NVR
|
|
assert_entity_counts(hass, Platform.SENSOR, 21, 13)
|
|
|
|
yield camera_obj
|
|
|
|
Camera.__config__.validate_assignment = True
|
|
|
|
|
|
async def test_sensor_setup_sensor(
|
|
hass: HomeAssistant, mock_entry: MockEntityFixture, sensor: Sensor
|
|
):
|
|
"""Test sensor entity setup for sensor devices."""
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
expected_values = ("10", "10.0", "10.0", "10.0")
|
|
for index, description in enumerate(SENSE_SENSORS):
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, sensor, description
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.unique_id == unique_id
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == expected_values[index]
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
# BLE signal
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, sensor, ALL_DEVICES_SENSORS[1]
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is True
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "-50"
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
|
|
async def test_sensor_setup_nvr(
|
|
hass: HomeAssistant, mock_entry: MockEntityFixture, now: datetime
|
|
):
|
|
"""Test sensor entity setup for NVR device."""
|
|
|
|
mock_entry.api.bootstrap.reset_objects()
|
|
nvr: NVR = mock_entry.api.bootstrap.nvr
|
|
nvr.up_since = now
|
|
nvr.system_info.cpu.average_load = 50.0
|
|
nvr.system_info.cpu.temperature = 50.0
|
|
nvr.storage_stats.utilization = 50.0
|
|
nvr.system_info.memory.available = 50.0
|
|
nvr.system_info.memory.total = 100.0
|
|
nvr.storage_stats.storage_distribution.timelapse_recordings.percentage = 50.0
|
|
nvr.storage_stats.storage_distribution.continuous_recordings.percentage = 50.0
|
|
nvr.storage_stats.storage_distribution.detections_recordings.percentage = 50.0
|
|
nvr.storage_stats.storage_distribution.hd_usage.percentage = 50.0
|
|
nvr.storage_stats.storage_distribution.uhd_usage.percentage = 50.0
|
|
nvr.storage_stats.storage_distribution.free.percentage = 50.0
|
|
nvr.storage_stats.capacity = 50.0
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# 2 from all, 4 from sense, 12 NVR
|
|
assert_entity_counts(hass, Platform.SENSOR, 12, 9)
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
expected_values = (
|
|
now.replace(second=0, microsecond=0).isoformat(),
|
|
"50.0",
|
|
"50.0",
|
|
"50.0",
|
|
"50.0",
|
|
"50.0",
|
|
"50.0",
|
|
"50.0",
|
|
"50",
|
|
)
|
|
for index, description in enumerate(NVR_SENSORS):
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, nvr, description
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is not description.entity_registry_enabled_default
|
|
assert entity.unique_id == unique_id
|
|
|
|
if not description.entity_registry_enabled_default:
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == expected_values[index]
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
expected_values = ("50.0", "50.0", "50.0")
|
|
for index, description in enumerate(NVR_DISABLED_SENSORS):
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, nvr, description
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is not description.entity_registry_enabled_default
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == expected_values[index]
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
|
|
async def test_sensor_setup_camera(
|
|
hass: HomeAssistant, mock_entry: MockEntityFixture, camera: Camera, now: datetime
|
|
):
|
|
"""Test sensor entity setup for camera devices."""
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
expected_values = (
|
|
now.replace(second=0, microsecond=0).isoformat(),
|
|
"100",
|
|
"100.0",
|
|
"20.0",
|
|
)
|
|
for index, description in enumerate(CAMERA_SENSORS):
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, camera, description
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is not description.entity_registry_enabled_default
|
|
assert entity.unique_id == unique_id
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == expected_values[index]
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
expected_values = ("100", "100")
|
|
for index, description in enumerate(CAMERA_DISABLED_SENSORS):
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, camera, description
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is not description.entity_registry_enabled_default
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == expected_values[index]
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
# Wired signal
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, camera, ALL_DEVICES_SENSORS[2]
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is True
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "1000"
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
# WiFi signal
|
|
unique_id, entity_id = ids_from_device_description(
|
|
Platform.SENSOR, camera, ALL_DEVICES_SENSORS[3]
|
|
)
|
|
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled is True
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "-50"
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|