mirror of
https://github.com/home-assistant/core.git
synced 2025-11-07 18:09:31 +00:00
b626204f63
* Add default device class display precision for Sensor * Renaming, docstrings, cleanup * Simplify units list * Fix tests * Fix missing precision when suggested is specified * Update snapshots * Fix when unit of measurement is not valid * Fix tests * Fix deprecated unit usage * Fix goalzero tests The sensor native_value method was accessing the data dict and trowing, since the mock did not have any data for the sensors. Since now the precision is always specified (it was missing for those sensors), the throw was hitting async_update_entity_options in _update_suggested_precision. Previously, async_update_entity_options was not called since it had no precision. * Fix metoffice * Fix smartthings * Add default sensor data for Tesla Wall Connector tests * Update snapshots * Revert spaces * Update smartthings snapshots * Add missing sensor mock for tesla wall connector * Address review comments * Add doc comment * Add cap to doc comment * Update comment * Update snapshots * Update comment
204 lines
6.4 KiB
Python
204 lines
6.4 KiB
Python
"""Vera tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from typing import Any
|
|
from unittest.mock import MagicMock
|
|
|
|
import pyvera as pv
|
|
|
|
from homeassistant.components.sensor import async_rounded_state
|
|
from homeassistant.const import ATTR_UNIT_OF_MEASUREMENT, LIGHT_LUX, PERCENTAGE
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import ComponentFactory, new_simple_controller_config
|
|
|
|
|
|
async def run_sensor_test(
|
|
hass: HomeAssistant,
|
|
vera_component_factory: ComponentFactory,
|
|
category: int,
|
|
class_property: str,
|
|
assert_states: tuple[tuple[Any, Any]],
|
|
assert_unit_of_measurement: str | None = None,
|
|
setup_callback: Callable[[pv.VeraController], None] | None = None,
|
|
) -> None:
|
|
"""Test generic sensor."""
|
|
vera_device: pv.VeraSensor = MagicMock(spec=pv.VeraSensor)
|
|
vera_device.device_id = 1
|
|
vera_device.vera_device_id = vera_device.device_id
|
|
vera_device.comm_failure = False
|
|
vera_device.name = "dev1"
|
|
vera_device.category = category
|
|
setattr(vera_device, class_property, "33")
|
|
entity_id = "sensor.dev1_1"
|
|
|
|
component_data = await vera_component_factory.configure_component(
|
|
hass=hass,
|
|
controller_config=new_simple_controller_config(
|
|
devices=(vera_device,), setup_callback=setup_callback
|
|
),
|
|
)
|
|
update_callback = component_data.controller_data[0].update_callback
|
|
|
|
for initial_value, state_value in assert_states:
|
|
setattr(vera_device, class_property, initial_value)
|
|
update_callback(vera_device)
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get(entity_id)
|
|
assert async_rounded_state(hass, entity_id, state) == state_value
|
|
if assert_unit_of_measurement:
|
|
assert (
|
|
state.attributes[ATTR_UNIT_OF_MEASUREMENT] == assert_unit_of_measurement
|
|
)
|
|
|
|
|
|
async def test_temperature_sensor_f(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
|
|
def setup_callback(controller: pv.VeraController) -> None:
|
|
controller.temperature_units = "F"
|
|
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_TEMPERATURE_SENSOR,
|
|
class_property="temperature",
|
|
assert_states=(("33", "0.6"), ("44", "6.7")),
|
|
setup_callback=setup_callback,
|
|
)
|
|
|
|
|
|
async def test_temperature_sensor_c(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_TEMPERATURE_SENSOR,
|
|
class_property="temperature",
|
|
assert_states=(("33", "33.0"), ("44", "44.0")),
|
|
)
|
|
|
|
|
|
async def test_light_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_LIGHT_SENSOR,
|
|
class_property="light",
|
|
assert_states=(("12", "12"), ("13", "13")),
|
|
assert_unit_of_measurement=LIGHT_LUX,
|
|
)
|
|
|
|
|
|
async def test_uv_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_UV_SENSOR,
|
|
class_property="light",
|
|
assert_states=(("12", "12"), ("13", "13")),
|
|
assert_unit_of_measurement="level",
|
|
)
|
|
|
|
|
|
async def test_humidity_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_HUMIDITY_SENSOR,
|
|
class_property="humidity",
|
|
assert_states=(("12", "12"), ("13", "13")),
|
|
assert_unit_of_measurement=PERCENTAGE,
|
|
)
|
|
|
|
|
|
async def test_power_meter_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=pv.CATEGORY_POWER_METER,
|
|
class_property="power",
|
|
assert_states=(("12", "12"), ("13", "13")),
|
|
assert_unit_of_measurement="W",
|
|
)
|
|
|
|
|
|
async def test_trippable_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
|
|
def setup_callback(controller: pv.VeraController) -> None:
|
|
controller.get_devices()[0].is_trippable = True
|
|
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=999,
|
|
class_property="is_tripped",
|
|
assert_states=((True, "Tripped"), (False, "Not Tripped"), (True, "Tripped")),
|
|
setup_callback=setup_callback,
|
|
)
|
|
|
|
|
|
async def test_unknown_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
|
|
def setup_callback(controller: pv.VeraController) -> None:
|
|
controller.get_devices()[0].is_trippable = False
|
|
|
|
await run_sensor_test(
|
|
hass=hass,
|
|
vera_component_factory=vera_component_factory,
|
|
category=999,
|
|
class_property="is_tripped",
|
|
assert_states=((True, "Unknown"), (False, "Unknown"), (True, "Unknown")),
|
|
setup_callback=setup_callback,
|
|
)
|
|
|
|
|
|
async def test_scene_controller_sensor(
|
|
hass: HomeAssistant, vera_component_factory: ComponentFactory
|
|
) -> None:
|
|
"""Test function."""
|
|
vera_device: pv.VeraSensor = MagicMock(spec=pv.VeraSensor)
|
|
vera_device.device_id = 1
|
|
vera_device.vera_device_id = vera_device.device_id
|
|
vera_device.comm_failure = False
|
|
vera_device.name = "dev1"
|
|
vera_device.category = pv.CATEGORY_SCENE_CONTROLLER
|
|
vera_device.get_last_scene_id = MagicMock(return_value="id0")
|
|
vera_device.get_last_scene_time = MagicMock(return_value="0000")
|
|
entity_id = "sensor.dev1_1"
|
|
|
|
component_data = await vera_component_factory.configure_component(
|
|
hass=hass,
|
|
controller_config=new_simple_controller_config(devices=(vera_device,)),
|
|
)
|
|
update_callback = component_data.controller_data[0].update_callback
|
|
|
|
vera_device.get_last_scene_time.return_value = "1111"
|
|
update_callback(vera_device)
|
|
await hass.async_block_till_done()
|
|
assert hass.states.get(entity_id).state == "id0"
|