51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
"""Test the Moat sensors."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.bluetooth import BluetoothChange
|
|
from homeassistant.components.moat.const import DOMAIN
|
|
from homeassistant.components.sensor import ATTR_STATE_CLASS
|
|
from homeassistant.const import ATTR_FRIENDLY_NAME, ATTR_UNIT_OF_MEASUREMENT
|
|
|
|
from . import MOAT_S2_SERVICE_INFO
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_sensors(hass):
|
|
"""Test setting up creates the sensors."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="61DE521B-F0BF-9F44-64D4-75BBE1738105",
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
saved_callback = None
|
|
|
|
def _async_register_callback(_hass, _callback, _matcher, _mode):
|
|
nonlocal saved_callback
|
|
saved_callback = _callback
|
|
return lambda: None
|
|
|
|
with patch(
|
|
"homeassistant.components.bluetooth.update_coordinator.async_register_callback",
|
|
_async_register_callback,
|
|
):
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all()) == 0
|
|
saved_callback(MOAT_S2_SERVICE_INFO, BluetoothChange.ADVERTISEMENT)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_all()) == 4
|
|
|
|
temp_sensor = hass.states.get("sensor.moat_s2_eeff_voltage")
|
|
temp_sensor_attribtes = temp_sensor.attributes
|
|
assert temp_sensor.state == "3.061"
|
|
assert temp_sensor_attribtes[ATTR_FRIENDLY_NAME] == "Moat S2 EEFF Voltage"
|
|
assert temp_sensor_attribtes[ATTR_UNIT_OF_MEASUREMENT] == "V"
|
|
assert temp_sensor_attribtes[ATTR_STATE_CLASS] == "measurement"
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|