evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

MQTT Switch to expose assumed_state if optimistic

Browse source
This commit is contained in:
Paulus Schoutsen 2016-02-20 17:17:30 -08:00
parent 8f70630790
commit 443b39bccd
2 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
homeassistant/components/switch/mqtt.py
View file

@ -97,6 +97,11 @@ class MqttSwitch(SwitchDevice):
""" True if device is on. """ """ True if device is on. """
return self._state return self._state
@property
def assumed_state(self):
"""Return True if we do optimistic updates."""
return self._optimistic
def turn_on(self, **kwargs): def turn_on(self, **kwargs):
""" Turn the device on. """ """ Turn the device on. """
mqtt.publish(self.hass, self._command_topic, self._payload_on, mqtt.publish(self.hass, self._command_topic, self._payload_on,

4
tests/components/switch/test_mqtt.py
View file

@ -6,7 +6,7 @@ Tests MQTT switch.
""" """
import unittest import unittest
from homeassistant.const import STATE_ON, STATE_OFF from homeassistant.const import STATE_ON, STATE_OFF, ATTR_ASSUMED_STATE
import homeassistant.components.switch as switch import homeassistant.components.switch as switch
from tests.common import ( from tests.common import (
mock_mqtt_component, fire_mqtt_message, get_test_home_assistant) mock_mqtt_component, fire_mqtt_message, get_test_home_assistant)
@ -37,6 +37,7 @@ class TestSensorMQTT(unittest.TestCase):
state = self.hass.states.get('switch.test') state = self.hass.states.get('switch.test')
self.assertEqual(STATE_OFF, state.state) self.assertEqual(STATE_OFF, state.state)
self.assertIsNone(state.attributes.get(ATTR_ASSUMED_STATE))
fire_mqtt_message(self.hass, 'state-topic', 'beer on') fire_mqtt_message(self.hass, 'state-topic', 'beer on')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
@ -64,6 +65,7 @@ class TestSensorMQTT(unittest.TestCase):
state = self.hass.states.get('switch.test') state = self.hass.states.get('switch.test')
self.assertEqual(STATE_OFF, state.state) self.assertEqual(STATE_OFF, state.state)
self.assertTrue(state.attributes.get(ATTR_ASSUMED_STATE))
switch.turn_on(self.hass, 'switch.test') switch.turn_on(self.hass, 'switch.test')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()