Fix configurator tests

This commit is contained in:
Paulus Schoutsen 2016-06-10 20:50:04 -07:00
parent 809e613148
commit 71eb09ee5e

View File

@ -3,7 +3,7 @@
import unittest import unittest
import homeassistant.components.configurator as configurator import homeassistant.components.configurator as configurator
from homeassistant.const import EVENT_TIME_CHANGED from homeassistant.const import EVENT_TIME_CHANGED, ATTR_FRIENDLY_NAME
from tests.common import get_test_home_assistant from tests.common import get_test_home_assistant
@ -40,26 +40,25 @@ class TestConfigurator(unittest.TestCase):
def test_request_all_info(self): def test_request_all_info(self):
"""Test request config with all possible info.""" """Test request config with all possible info."""
values = [ exp_attr = {
"config_description", "config image url", ATTR_FRIENDLY_NAME: "Test Request",
"config submit caption", []] configurator.ATTR_DESCRIPTION: "config description",
configurator.ATTR_DESCRIPTION_IMAGE: "config image url",
keys = [ configurator.ATTR_SUBMIT_CAPTION: "config submit caption",
configurator.ATTR_DESCRIPTION, configurator.ATTR_DESCRIPTION_IMAGE, configurator.ATTR_FIELDS: [],
configurator.ATTR_SUBMIT_CAPTION, configurator.ATTR_FIELDS] configurator.ATTR_CONFIGURE_ID: configurator.request_config(
exp_attr = dict(zip(keys, values))
exp_attr[configurator.ATTR_CONFIGURE_ID] = configurator.request_config(
self.hass, "Test Request", lambda _: None, self.hass, "Test Request", lambda _: None,
*values) "config description", "config image url",
"config submit caption"
)
}
states = self.hass.states.all() states = self.hass.states.all()
self.assertEqual(1, len(states)) self.assertEqual(1, len(states))
state = states[0] state = states[0]
self.assertEqual(configurator.STATE_CONFIGURE, state.state) self.assertEqual(configurator.STATE_CONFIGURE, state.state)
self.assertEqual(exp_attr, state.attributes) assert exp_attr == dict(state.attributes)
def test_callback_called_on_configure(self): def test_callback_called_on_configure(self):
"""Test if our callback gets called when configure service called.""" """Test if our callback gets called when configure service called."""