39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""Test the SmartTub sensor platform."""
|
|
|
|
from . import trigger_update
|
|
|
|
|
|
async def test_sensors(spa, setup_entry, hass, smarttub_api):
|
|
"""Test the sensors."""
|
|
|
|
entity_id = f"sensor.{spa.brand}_{spa.model}_state"
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "normal"
|
|
|
|
spa.get_status.return_value["state"] = "BAD"
|
|
await trigger_update(hass)
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "bad"
|
|
|
|
entity_id = f"sensor.{spa.brand}_{spa.model}_flow_switch"
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "open"
|
|
|
|
entity_id = f"sensor.{spa.brand}_{spa.model}_ozone"
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "off"
|
|
|
|
entity_id = f"sensor.{spa.brand}_{spa.model}_blowout_cycle"
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "inactive"
|
|
|
|
entity_id = f"sensor.{spa.brand}_{spa.model}_cleanup_cycle"
|
|
state = hass.states.get(entity_id)
|
|
assert state is not None
|
|
assert state.state == "inactive"
|