Fixes #10773: Demo Alarm Broken

homeassistant/components/alarm_control_panel/demo.py

@@ -7,7 +7,7 @@
 import homeassistant.components.alarm_control_panel.manual as manual
 from homeassistant.const import (
     STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_NIGHT,
-    STATE_ALARM_TRIGGERED, CONF_PENDING_TIME)
+    STATE_ALARM_ARMED_CUSTOM_BYPASS, STATE_ALARM_TRIGGERED, CONF_PENDING_TIME)
 
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
@@ -23,6 +23,9 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
             STATE_ALARM_ARMED_NIGHT: {
                 CONF_PENDING_TIME: 5
             },
+            STATE_ALARM_ARMED_CUSTOM_BYPASS: {
+                CONF_PENDING_TIME: 5
+            },
             STATE_ALARM_TRIGGERED: {
                 CONF_PENDING_TIME: 5
             },

tests/components/alarm_control_panel/test_manual.py

@@ -1,7 +1,9 @@
 """The tests for the manual Alarm Control Panel component."""
 from datetime import timedelta
 import unittest
-from unittest.mock import patch
+from unittest.mock import patch, MagicMock
+from homeassistant.components.alarm_control_panel import demo
+
 
 from homeassistant.setup import setup_component
 from homeassistant.const import (
@@ -11,6 +13,8 @@
 from homeassistant.components import alarm_control_panel
 import homeassistant.util.dt as dt_util
 
+from tests.common import assert_setup_component
+
 from tests.common import fire_time_changed, get_test_home_assistant
 
 CODE = 'HELLO_CODE'
@@ -27,6 +31,13 @@ def tearDown(self):  # pylint: disable=invalid-name
         """Stop down everything that was started."""
         self.hass.stop()
 
+    def test_setup_demo_platform(self):
+        """Test setup."""
+        mock = MagicMock();
+        add_devices = mock.MagicMock()
+        demo.setup_platform(self.hass, {}, add_devices)
+        add_devices.assert_called_once()
+
     def test_arm_home_no_pending(self):
         """Test arm home method."""
         self.assertTrue(setup_component(