mirror of
https://github.com/home-assistant/core.git
synced 2025-04-29 11:47:50 +00:00
2fb3be50ab
* Make async_get_device connections Optional, default None * Remove unnecessary async_get_device connections arg usages Some of these were using an incorrect collection type, which didn't cause issues mostly just due to luck.
214 lines
7.0 KiB
Python
214 lines
7.0 KiB
Python
"""The tests for the Canary sensor platform."""
|
|
from datetime import timedelta
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.canary.const import DOMAIN, MANUFACTURER
|
|
from homeassistant.components.canary.sensor import (
|
|
ATTR_AIR_QUALITY,
|
|
STATE_AIR_QUALITY_ABNORMAL,
|
|
STATE_AIR_QUALITY_NORMAL,
|
|
STATE_AIR_QUALITY_VERY_ABNORMAL,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
DEVICE_CLASS_BATTERY,
|
|
DEVICE_CLASS_HUMIDITY,
|
|
DEVICE_CLASS_SIGNAL_STRENGTH,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
PERCENTAGE,
|
|
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
|
|
TEMP_CELSIUS,
|
|
)
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from . import mock_device, mock_location, mock_reading
|
|
|
|
from tests.common import async_fire_time_changed, mock_device_registry, mock_registry
|
|
|
|
|
|
async def test_sensors_pro(hass, canary) -> None:
|
|
"""Test the creation and values of the sensors for Canary Pro."""
|
|
await async_setup_component(hass, "persistent_notification", {})
|
|
|
|
registry = mock_registry(hass)
|
|
device_registry = mock_device_registry(hass)
|
|
|
|
online_device_at_home = mock_device(20, "Dining Room", True, "Canary Pro")
|
|
|
|
instance = canary.return_value
|
|
instance.get_locations.return_value = [
|
|
mock_location(100, "Home", True, devices=[online_device_at_home]),
|
|
]
|
|
|
|
instance.get_latest_readings.return_value = [
|
|
mock_reading("temperature", "21.12"),
|
|
mock_reading("humidity", "50.46"),
|
|
mock_reading("air_quality", "0.59"),
|
|
]
|
|
|
|
config = {DOMAIN: {"username": "test-username", "password": "test-password"}}
|
|
with patch("homeassistant.components.canary.PLATFORMS", ["sensor"]):
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
await hass.async_block_till_done()
|
|
|
|
sensors = {
|
|
"home_dining_room_temperature": (
|
|
"20_temperature",
|
|
"21.12",
|
|
TEMP_CELSIUS,
|
|
DEVICE_CLASS_TEMPERATURE,
|
|
None,
|
|
),
|
|
"home_dining_room_humidity": (
|
|
"20_humidity",
|
|
"50.46",
|
|
PERCENTAGE,
|
|
DEVICE_CLASS_HUMIDITY,
|
|
None,
|
|
),
|
|
"home_dining_room_air_quality": (
|
|
"20_air_quality",
|
|
"0.59",
|
|
None,
|
|
None,
|
|
"mdi:weather-windy",
|
|
),
|
|
}
|
|
|
|
for (sensor_id, data) in sensors.items():
|
|
entity_entry = registry.async_get(f"sensor.{sensor_id}")
|
|
assert entity_entry
|
|
assert entity_entry.device_class == data[3]
|
|
assert entity_entry.unique_id == data[0]
|
|
assert entity_entry.original_icon == data[4]
|
|
|
|
state = hass.states.get(f"sensor.{sensor_id}")
|
|
assert state
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == data[2]
|
|
assert state.state == data[1]
|
|
|
|
device = device_registry.async_get_device({(DOMAIN, "20")})
|
|
assert device
|
|
assert device.manufacturer == MANUFACTURER
|
|
assert device.name == "Dining Room"
|
|
assert device.model == "Canary Pro"
|
|
|
|
|
|
async def test_sensors_attributes_pro(hass, canary) -> None:
|
|
"""Test the creation and values of the sensors attributes for Canary Pro."""
|
|
await async_setup_component(hass, "persistent_notification", {})
|
|
|
|
online_device_at_home = mock_device(20, "Dining Room", True, "Canary Pro")
|
|
|
|
instance = canary.return_value
|
|
instance.get_locations.return_value = [
|
|
mock_location(100, "Home", True, devices=[online_device_at_home]),
|
|
]
|
|
|
|
instance.get_latest_readings.return_value = [
|
|
mock_reading("temperature", "21.12"),
|
|
mock_reading("humidity", "50.46"),
|
|
mock_reading("air_quality", "0.59"),
|
|
]
|
|
|
|
config = {DOMAIN: {"username": "test-username", "password": "test-password"}}
|
|
with patch("homeassistant.components.canary.PLATFORMS", ["sensor"]):
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_id = "sensor.home_dining_room_air_quality"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes[ATTR_AIR_QUALITY] == STATE_AIR_QUALITY_ABNORMAL
|
|
|
|
instance.get_latest_readings.return_value = [
|
|
mock_reading("temperature", "21.12"),
|
|
mock_reading("humidity", "50.46"),
|
|
mock_reading("air_quality", "0.4"),
|
|
]
|
|
|
|
future = utcnow() + timedelta(seconds=30)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.helpers.entity_component.async_update_entity(entity_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes[ATTR_AIR_QUALITY] == STATE_AIR_QUALITY_VERY_ABNORMAL
|
|
|
|
instance.get_latest_readings.return_value = [
|
|
mock_reading("temperature", "21.12"),
|
|
mock_reading("humidity", "50.46"),
|
|
mock_reading("air_quality", "1.0"),
|
|
]
|
|
|
|
future += timedelta(seconds=30)
|
|
async_fire_time_changed(hass, future)
|
|
await hass.helpers.entity_component.async_update_entity(entity_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes[ATTR_AIR_QUALITY] == STATE_AIR_QUALITY_NORMAL
|
|
|
|
|
|
async def test_sensors_flex(hass, canary) -> None:
|
|
"""Test the creation and values of the sensors for Canary Flex."""
|
|
await async_setup_component(hass, "persistent_notification", {})
|
|
|
|
registry = mock_registry(hass)
|
|
device_registry = mock_device_registry(hass)
|
|
|
|
online_device_at_home = mock_device(20, "Dining Room", True, "Canary Flex")
|
|
|
|
instance = canary.return_value
|
|
instance.get_locations.return_value = [
|
|
mock_location(100, "Home", True, devices=[online_device_at_home]),
|
|
]
|
|
|
|
instance.get_latest_readings.return_value = [
|
|
mock_reading("battery", "70.4567"),
|
|
mock_reading("wifi", "-57"),
|
|
]
|
|
|
|
config = {DOMAIN: {"username": "test-username", "password": "test-password"}}
|
|
with patch("homeassistant.components.canary.PLATFORMS", ["sensor"]):
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
await hass.async_block_till_done()
|
|
|
|
sensors = {
|
|
"home_dining_room_battery": (
|
|
"20_battery",
|
|
"70.46",
|
|
PERCENTAGE,
|
|
DEVICE_CLASS_BATTERY,
|
|
None,
|
|
),
|
|
"home_dining_room_wifi": (
|
|
"20_wifi",
|
|
"-57.0",
|
|
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
|
|
DEVICE_CLASS_SIGNAL_STRENGTH,
|
|
None,
|
|
),
|
|
}
|
|
|
|
for (sensor_id, data) in sensors.items():
|
|
entity_entry = registry.async_get(f"sensor.{sensor_id}")
|
|
assert entity_entry
|
|
assert entity_entry.device_class == data[3]
|
|
assert entity_entry.unique_id == data[0]
|
|
assert entity_entry.original_icon == data[4]
|
|
|
|
state = hass.states.get(f"sensor.{sensor_id}")
|
|
assert state
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == data[2]
|
|
assert state.state == data[1]
|
|
|
|
device = device_registry.async_get_device({(DOMAIN, "20")})
|
|
assert device
|
|
assert device.manufacturer == MANUFACTURER
|
|
assert device.name == "Dining Room"
|
|
assert device.model == "Canary Flex"
|