121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
"""
|
|
tests.test_component_configurator
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests Configurator component.
|
|
"""
|
|
# pylint: disable=too-many-public-methods,protected-access
|
|
import unittest
|
|
|
|
import homeassistant.core as ha
|
|
import homeassistant.components.configurator as configurator
|
|
from homeassistant.const import EVENT_TIME_CHANGED
|
|
|
|
|
|
class TestConfigurator(unittest.TestCase):
|
|
""" Test the chromecast module. """
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
self.hass = ha.HomeAssistant()
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
def test_request_least_info(self):
|
|
""" Test request config with least amount of data. """
|
|
|
|
request_id = configurator.request_config(
|
|
self.hass, "Test Request", lambda _: None)
|
|
|
|
self.assertEqual(
|
|
1, len(self.hass.services.services.get(configurator.DOMAIN, [])),
|
|
"No new service registered")
|
|
|
|
states = self.hass.states.all()
|
|
|
|
self.assertEqual(1, len(states), "Expected a new state registered")
|
|
|
|
state = states[0]
|
|
|
|
self.assertEqual(configurator.STATE_CONFIGURE, state.state)
|
|
self.assertEqual(
|
|
request_id, state.attributes.get(configurator.ATTR_CONFIGURE_ID))
|
|
|
|
def test_request_all_info(self):
|
|
""" Test request config with all possible info. """
|
|
|
|
values = [
|
|
"config_description", "config image url",
|
|
"config submit caption", []]
|
|
|
|
keys = [
|
|
configurator.ATTR_DESCRIPTION, configurator.ATTR_DESCRIPTION_IMAGE,
|
|
configurator.ATTR_SUBMIT_CAPTION, configurator.ATTR_FIELDS]
|
|
|
|
exp_attr = dict(zip(keys, values))
|
|
|
|
exp_attr[configurator.ATTR_CONFIGURE_ID] = configurator.request_config(
|
|
self.hass, "Test Request", lambda _: None,
|
|
*values)
|
|
|
|
states = self.hass.states.all()
|
|
|
|
self.assertEqual(1, len(states))
|
|
|
|
state = states[0]
|
|
|
|
self.assertEqual(configurator.STATE_CONFIGURE, state.state)
|
|
self.assertEqual(exp_attr, state.attributes)
|
|
|
|
def test_callback_called_on_configure(self):
|
|
""" Test if our callback gets called when configure service called. """
|
|
calls = []
|
|
|
|
request_id = configurator.request_config(
|
|
self.hass, "Test Request", lambda _: calls.append(1))
|
|
|
|
self.hass.services.call(
|
|
configurator.DOMAIN, configurator.SERVICE_CONFIGURE,
|
|
{configurator.ATTR_CONFIGURE_ID: request_id})
|
|
|
|
self.hass.pool.block_till_done()
|
|
|
|
self.assertEqual(1, len(calls), "Callback not called")
|
|
|
|
def test_state_change_on_notify_errors(self):
|
|
""" Test state change on notify errors. """
|
|
request_id = configurator.request_config(
|
|
self.hass, "Test Request", lambda _: None)
|
|
|
|
error = "Oh no bad bad bad"
|
|
|
|
configurator.notify_errors(request_id, error)
|
|
|
|
state = self.hass.states.all()[0]
|
|
|
|
self.assertEqual(error, state.attributes.get(configurator.ATTR_ERRORS))
|
|
|
|
def test_notify_errors_fail_silently_on_bad_request_id(self):
|
|
""" Test if notify errors fails silently with a bad request id. """
|
|
configurator.notify_errors(2015, "Try this error")
|
|
|
|
def test_request_done_works(self):
|
|
""" Test if calling request done works. """
|
|
request_id = configurator.request_config(
|
|
self.hass, "Test Request", lambda _: None)
|
|
|
|
configurator.request_done(request_id)
|
|
|
|
self.assertEqual(1, len(self.hass.states.all()))
|
|
|
|
self.hass.bus.fire(EVENT_TIME_CHANGED)
|
|
|
|
self.hass.pool.block_till_done()
|
|
|
|
self.assertEqual(0, len(self.hass.states.all()))
|
|
|
|
def test_request_done_fail_silently_on_bad_request_id(self):
|
|
""" Test that request_done fails silently with a bad request id. """
|
|
configurator.request_done(2016)
|