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
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,7 +6,7 @@ Tests MQTT switch.
"""
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
from tests.common import (
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')
self.assertEqual(STATE_OFF, state.state)
self.assertIsNone(state.attributes.get(ATTR_ASSUMED_STATE))
fire_mqtt_message(self.hass, 'state-topic', 'beer on')
self.hass.pool.block_till_done()
@ -64,6 +65,7 @@ class TestSensorMQTT(unittest.TestCase):
state = self.hass.states.get('switch.test')
self.assertEqual(STATE_OFF, state.state)
self.assertTrue(state.attributes.get(ATTR_ASSUMED_STATE))
switch.turn_on(self.hass, 'switch.test')
self.hass.pool.block_till_done()