From 6b899ddc1db3b59bb2c26ae22c172b75a352d0bf Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Thu, 14 Jan 2016 23:25:25 -0800 Subject: [PATCH] 100% test coverage for input_boolean --- homeassistant/components/input_boolean.py | 4 ++-- tests/components/test_input_boolean.py | 6 +++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/input_boolean.py b/homeassistant/components/input_boolean.py index 27216782ba1..f86fffcaeb6 100644 --- a/homeassistant/components/input_boolean.py +++ b/homeassistant/components/input_boolean.py @@ -65,8 +65,6 @@ def setup(hass, config): if not entities: return False - component.add_entities(entities) - def toggle_service(service): """Handle a calls to the input boolean services.""" target_inputs = component.extract_from_service(service) @@ -80,6 +78,8 @@ def setup(hass, config): hass.services.register(DOMAIN, SERVICE_TURN_OFF, toggle_service) hass.services.register(DOMAIN, SERVICE_TURN_ON, toggle_service) + component.add_entities(entities) + return True diff --git a/tests/components/test_input_boolean.py b/tests/components/test_input_boolean.py index a7366d91fd3..ace91018ccd 100644 --- a/tests/components/test_input_boolean.py +++ b/tests/components/test_input_boolean.py @@ -26,14 +26,18 @@ class TestInputBoolean(unittest.TestCase): def test_config(self): """Test config.""" + self.assertFalse(input_boolean.setup(self.hass, { + 'input_boolean': None + })) + self.assertFalse(input_boolean.setup(self.hass, { 'input_boolean': { - 'test 1': None, } })) self.assertFalse(input_boolean.setup(self.hass, { 'input_boolean': { + 'name with space': None } }))