100% test coverage for input_boolean

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
 
 

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
             }
         }))