jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 05:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Mqtt vacuum unknown state (#65311)

Browse Source 
* Add Mqtt vacuum `unknown` status

* Update tests/components/mqtt/test_state_vacuum.py

Co-authored-by: Erik Montnemery <erik@montnemery.com>

Co-authored-by: Erik Montnemery <erik@montnemery.com>
This commit is contained in:
Jan Bouwhuis 2022-02-03 16:50:39 +01:00 committed by GitHub
parent cf52357229
commit 2f0d0998a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/mqtt/vacuum/schema_state.py
View File

@ -206,8 +206,12 @@ class MqttStateVacuum(MqttEntity, StateVacuumEntity):
def state_message_received(msg): def state_message_received(msg):
"""Handle state MQTT message.""" """Handle state MQTT message."""
payload = json.loads(msg.payload) payload = json.loads(msg.payload)
if STATE in payload and payload[STATE] in POSSIBLE_STATES: if STATE in payload and (
self._state = POSSIBLE_STATES[payload[STATE]] payload[STATE] in POSSIBLE_STATES or payload[STATE] is None
):
self._state = (
POSSIBLE_STATES[payload[STATE]] if payload[STATE] else None
)
del payload[STATE] del payload[STATE]
self._state_attrs.update(payload) self._state_attrs.update(payload)
self.async_write_ha_state() self.async_write_ha_state()

7
tests/components/mqtt/test_state_vacuum.py
View File

@ -235,6 +235,8 @@ async def test_status(hass, mqtt_mock):
assert await async_setup_component(hass, vacuum.DOMAIN, {vacuum.DOMAIN: config}) assert await async_setup_component(hass, vacuum.DOMAIN, {vacuum.DOMAIN: config})
await hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("vacuum.mqtttest")
assert state.state == STATE_UNKNOWN
message = """{ message = """{
"battery_level": 54, "battery_level": 54,
@ -262,6 +264,11 @@ async def test_status(hass, mqtt_mock):
assert state.attributes.get(ATTR_FAN_SPEED) == "min" assert state.attributes.get(ATTR_FAN_SPEED) == "min"
assert state.attributes.get(ATTR_FAN_SPEED_LIST) == ["min", "medium", "high", "max"] assert state.attributes.get(ATTR_FAN_SPEED_LIST) == ["min", "medium", "high", "max"]
message = '{"state":null}'
async_fire_mqtt_message(hass, "vacuum/state", message)
state = hass.states.get("vacuum.mqtttest")
assert state.state == STATE_UNKNOWN
async def test_no_fan_vacuum(hass, mqtt_mock): async def test_no_fan_vacuum(hass, mqtt_mock):
"""Test status updates from the vacuum when fan is not supported.""" """Test status updates from the vacuum when fan is not supported."""