jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup SmartTub filtration cycles (#47192)

Browse Source
This commit is contained in:
Matt Zimmerman 2021-02-28 15:53:57 -08:00 committed by GitHub
parent 1d7660f071
commit 277c3cb661
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/smarttub/sensor.py
View File

@ -7,9 +7,11 @@ from .entity import SmartTubSensorBase
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
# the desired duration, in hours, of the cycle
ATTR_DURATION = "duration" ATTR_DURATION = "duration"
ATTR_LAST_UPDATED = "last_updated" ATTR_CYCLE_LAST_UPDATED = "cycle_last_updated"
ATTR_MODE = "mode" ATTR_MODE = "mode"
# the hour of the day at which to start the cycle (0-23)
ATTR_START_HOUR = "start_hour" ATTR_START_HOUR = "start_hour"
@ -73,7 +75,7 @@ class SmartTubPrimaryFiltrationCycle(SmartTubSensor):
state = self._state state = self._state
return { return {
ATTR_DURATION: state.duration, ATTR_DURATION: state.duration,
ATTR_LAST_UPDATED: state.last_updated.isoformat(), ATTR_CYCLE_LAST_UPDATED: state.last_updated.isoformat(),
ATTR_MODE: state.mode.name.lower(), ATTR_MODE: state.mode.name.lower(),
ATTR_START_HOUR: state.start_hour, ATTR_START_HOUR: state.start_hour,
} }
@ -98,6 +100,6 @@ class SmartTubSecondaryFiltrationCycle(SmartTubSensor):
"""Return the state attributes.""" """Return the state attributes."""
state = self._state state = self._state
return { return {
ATTR_LAST_UPDATED: state.last_updated.isoformat(), ATTR_CYCLE_LAST_UPDATED: state.last_updated.isoformat(),
ATTR_MODE: state.mode.name.lower(), ATTR_MODE: state.mode.name.lower(),
} }

4
tests/components/smarttub/test_sensor.py
View File

@ -47,7 +47,7 @@ async def test_sensors(spa, setup_entry, hass):
assert state is not None assert state is not None
assert state.state == "inactive" assert state.state == "inactive"
assert state.attributes["duration"] == 4 assert state.attributes["duration"] == 4
assert state.attributes["last_updated"] is not None assert state.attributes["cycle_last_updated"] is not None
assert state.attributes["mode"] == "normal" assert state.attributes["mode"] == "normal"
assert state.attributes["start_hour"] == 2 assert state.attributes["start_hour"] == 2
@ -55,5 +55,5 @@ async def test_sensors(spa, setup_entry, hass):
state = hass.states.get(entity_id) state = hass.states.get(entity_id)
assert state is not None assert state is not None
assert state.state == "inactive" assert state.state == "inactive"
assert state.attributes["last_updated"] is not None assert state.attributes["cycle_last_updated"] is not None
assert state.attributes["mode"] == "away" assert state.attributes["mode"] == "away"