mirror of https://github.com/home-assistant/core
86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
"""Test the Mopeka sensors."""
|
|
|
|
|
|
from homeassistant.components.mopeka.const import DOMAIN
|
|
from homeassistant.components.sensor import ATTR_STATE_CLASS
|
|
from homeassistant.const import (
|
|
ATTR_FRIENDLY_NAME,
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
STATE_UNKNOWN,
|
|
UnitOfLength,
|
|
UnitOfTemperature,
|
|
)
|
|
|
|
from . import PRO_GOOD_SIGNAL_SERVICE_INFO, PRO_SERVICE_INFO
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.bluetooth import inject_bluetooth_service_info
|
|
|
|
|
|
async def test_sensors_bad_signal(hass):
|
|
"""Test setting up creates the sensors when there is bad signal."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="aa:bb:cc:dd:ee:ff",
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all("sensor")) == 0
|
|
inject_bluetooth_service_info(hass, PRO_SERVICE_INFO)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_all("sensor")) == 6
|
|
|
|
temp_sensor = hass.states.get("sensor.pro_plus_eeff_temperature")
|
|
temp_sensor_attrs = temp_sensor.attributes
|
|
assert temp_sensor.state == "30"
|
|
assert temp_sensor_attrs[ATTR_FRIENDLY_NAME] == "Pro Plus EEFF Temperature"
|
|
assert temp_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == UnitOfTemperature.CELSIUS
|
|
assert temp_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
|
|
|
|
tank_sensor = hass.states.get("sensor.pro_plus_eeff_tank_level")
|
|
tank_sensor_attrs = tank_sensor.attributes
|
|
assert tank_sensor.state == STATE_UNKNOWN
|
|
assert tank_sensor_attrs[ATTR_FRIENDLY_NAME] == "Pro Plus EEFF Tank Level"
|
|
assert tank_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == UnitOfLength.MILLIMETERS
|
|
assert tank_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def test_sensors_good_signal(hass):
|
|
"""Test setting up creates the sensors when there is good signal."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="aa:bb:cc:dd:ee:ff",
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all("sensor")) == 0
|
|
inject_bluetooth_service_info(hass, PRO_GOOD_SIGNAL_SERVICE_INFO)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_all("sensor")) == 6
|
|
|
|
temp_sensor = hass.states.get("sensor.pro_plus_eeff_temperature")
|
|
temp_sensor_attrs = temp_sensor.attributes
|
|
assert temp_sensor.state == "27"
|
|
assert temp_sensor_attrs[ATTR_FRIENDLY_NAME] == "Pro Plus EEFF Temperature"
|
|
assert temp_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == UnitOfTemperature.CELSIUS
|
|
assert temp_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
|
|
|
|
tank_sensor = hass.states.get("sensor.pro_plus_eeff_tank_level")
|
|
tank_sensor_attrs = tank_sensor.attributes
|
|
assert tank_sensor.state == "341"
|
|
assert tank_sensor_attrs[ATTR_FRIENDLY_NAME] == "Pro Plus EEFF Tank Level"
|
|
assert tank_sensor_attrs[ATTR_UNIT_OF_MEASUREMENT] == UnitOfLength.MILLIMETERS
|
|
assert tank_sensor_attrs[ATTR_STATE_CLASS] == "measurement"
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|