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

Type checks for MQTT config

Browse Source
This commit is contained in:
Paulus Schoutsen 2016-03-10 21:42:17 -08:00
parent ce82bd7515
commit 025713ee41
4 changed files with 28 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/light/mqtt.py
View File

@ -11,6 +11,7 @@ import homeassistant.components.mqtt as mqtt
from homeassistant.components.light import ( from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_RGB_COLOR, Light) ATTR_BRIGHTNESS, ATTR_RGB_COLOR, Light)
from homeassistant.helpers.template import render_with_possible_json_value from homeassistant.helpers.template import render_with_possible_json_value
from homeassistant.util import convert
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -32,20 +33,20 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
add_devices_callback([MqttLight( add_devices_callback([MqttLight(
hass, hass,
config.get('name', DEFAULT_NAME), convert(config.get('name'), str, DEFAULT_NAME),
{key: config.get(key) for key in {key: convert(config.get(key), str) for key in
(typ + topic (typ + topic
for typ in ('', 'brightness_', 'rgb_') for typ in ('', 'brightness_', 'rgb_')
for topic in ('state_topic', 'command_topic'))}, for topic in ('state_topic', 'command_topic'))},
{key: config.get(key + '_value_template') {key: convert(config.get(key + '_value_template'), str)
for key in ('state', 'brightness', 'rgb')}, for key in ('state', 'brightness', 'rgb')},
config.get('qos', DEFAULT_QOS), convert(config.get('qos'), int, DEFAULT_QOS),
{ {
'on': config.get('payload_on', DEFAULT_PAYLOAD_ON), 'on': convert(config.get('payload_on'), str, DEFAULT_PAYLOAD_ON),
'off': config.get('payload_off', DEFAULT_PAYLOAD_OFF) 'off': convert(config.get('payload_off'), str, DEFAULT_PAYLOAD_OFF)
}, },
config.get('optimistic', DEFAULT_OPTIMISTIC), convert(config.get('optimistic'), bool, DEFAULT_OPTIMISTIC),
config.get('brightness_scale', DEFAULT_BRIGHTNESS_SCALE) convert(config.get('brightness_scale'), int, DEFAULT_BRIGHTNESS_SCALE)
)]) )])

13
homeassistant/components/switch/mqtt.py
View File

@ -10,6 +10,7 @@ import homeassistant.components.mqtt as mqtt
from homeassistant.components.switch import SwitchDevice from homeassistant.components.switch import SwitchDevice
from homeassistant.const import CONF_VALUE_TEMPLATE from homeassistant.const import CONF_VALUE_TEMPLATE
from homeassistant.helpers import template from homeassistant.helpers import template
from homeassistant.util import convert
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -32,14 +33,14 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
add_devices_callback([MqttSwitch( add_devices_callback([MqttSwitch(
hass, hass,
config.get('name', DEFAULT_NAME), convert(config.get('name'), str, DEFAULT_NAME),
config.get('state_topic'), config.get('state_topic'),
config.get('command_topic'), config.get('command_topic'),
config.get('qos', DEFAULT_QOS), convert(config.get('qos'), int, DEFAULT_QOS),
config.get('retain', DEFAULT_RETAIN), convert(config.get('retain'), bool, DEFAULT_RETAIN),
config.get('payload_on', DEFAULT_PAYLOAD_ON), convert(config.get('payload_on'), str, DEFAULT_PAYLOAD_ON),
config.get('payload_off', DEFAULT_PAYLOAD_OFF), convert(config.get('payload_off'), str, DEFAULT_PAYLOAD_OFF),
config.get('optimistic', DEFAULT_OPTIMISTIC), convert(config.get('optimistic'), bool, DEFAULT_OPTIMISTIC),
config.get(CONF_VALUE_TEMPLATE))]) config.get(CONF_VALUE_TEMPLATE))])

14
tests/components/light/test_mqtt.py
View File

@ -122,9 +122,9 @@ class TestLightMQTT(unittest.TestCase):
'brightness_command_topic': 'test_light_rgb/brightness/set', 'brightness_command_topic': 'test_light_rgb/brightness/set',
'rgb_state_topic': 'test_light_rgb/rgb/status', 'rgb_state_topic': 'test_light_rgb/rgb/status',
'rgb_command_topic': 'test_light_rgb/rgb/set', 'rgb_command_topic': 'test_light_rgb/rgb/set',
'qos': 0, 'qos': '0',
'payload_on': 'on', 'payload_on': 1,
'payload_off': 'off' 'payload_off': 0
} }
})) }))
@ -134,7 +134,7 @@ class TestLightMQTT(unittest.TestCase):
self.assertIsNone(state.attributes.get('brightness')) self.assertIsNone(state.attributes.get('brightness'))
self.assertIsNone(state.attributes.get(ATTR_ASSUMED_STATE)) self.assertIsNone(state.attributes.get(ATTR_ASSUMED_STATE))
fire_mqtt_message(self.hass, 'test_light_rgb/status', 'on') fire_mqtt_message(self.hass, 'test_light_rgb/status', '1')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
state = self.hass.states.get('light.test') state = self.hass.states.get('light.test')
@ -142,13 +142,13 @@ class TestLightMQTT(unittest.TestCase):
self.assertEqual([255, 255, 255], state.attributes.get('rgb_color')) self.assertEqual([255, 255, 255], state.attributes.get('rgb_color'))
self.assertEqual(255, state.attributes.get('brightness')) self.assertEqual(255, state.attributes.get('brightness'))
fire_mqtt_message(self.hass, 'test_light_rgb/status', 'off') fire_mqtt_message(self.hass, 'test_light_rgb/status', '0')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
state = self.hass.states.get('light.test') state = self.hass.states.get('light.test')
self.assertEqual(STATE_OFF, state.state) self.assertEqual(STATE_OFF, state.state)
fire_mqtt_message(self.hass, 'test_light_rgb/status', 'on') fire_mqtt_message(self.hass, 'test_light_rgb/status', '1')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
fire_mqtt_message(self.hass, 'test_light_rgb/brightness/status', '100') fire_mqtt_message(self.hass, 'test_light_rgb/brightness/status', '100')
@ -159,7 +159,7 @@ class TestLightMQTT(unittest.TestCase):
self.assertEqual(100, self.assertEqual(100,
light_state.attributes['brightness']) light_state.attributes['brightness'])
fire_mqtt_message(self.hass, 'test_light_rgb/status', 'on') fire_mqtt_message(self.hass, 'test_light_rgb/status', '1')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
fire_mqtt_message(self.hass, 'test_light_rgb/rgb/status', fire_mqtt_message(self.hass, 'test_light_rgb/rgb/status',

10
tests/components/switch/test_mqtt.py
View File

@ -27,8 +27,8 @@ class TestSensorMQTT(unittest.TestCase):
'name': 'test', 'name': 'test',
'state_topic': 'state-topic', 'state_topic': 'state-topic',
'command_topic': 'command-topic', 'command_topic': 'command-topic',
'payload_on': 'beer on', 'payload_on': 1,
'payload_off': 'beer off' 'payload_off': 0
} }
})) }))
@ -36,13 +36,13 @@ class TestSensorMQTT(unittest.TestCase):
self.assertEqual(STATE_OFF, state.state) self.assertEqual(STATE_OFF, state.state)
self.assertIsNone(state.attributes.get(ATTR_ASSUMED_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', '1')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
state = self.hass.states.get('switch.test') state = self.hass.states.get('switch.test')
self.assertEqual(STATE_ON, state.state) self.assertEqual(STATE_ON, state.state)
fire_mqtt_message(self.hass, 'state-topic', 'beer off') fire_mqtt_message(self.hass, 'state-topic', '0')
self.hass.pool.block_till_done() self.hass.pool.block_till_done()
state = self.hass.states.get('switch.test') state = self.hass.states.get('switch.test')
@ -57,7 +57,7 @@ class TestSensorMQTT(unittest.TestCase):
'command_topic': 'command-topic', 'command_topic': 'command-topic',
'payload_on': 'beer on', 'payload_on': 'beer on',
'payload_off': 'beer off', 'payload_off': 'beer off',
'qos': 2 'qos': '2'
} }
})) }))