39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""
|
|
tests.test_component_demo
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests demo component.
|
|
"""
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
import homeassistant.core as ha
|
|
import homeassistant.components.demo as demo
|
|
|
|
from tests.common import mock_http_component
|
|
|
|
|
|
class TestDemo(unittest.TestCase):
|
|
""" Test the demo module. """
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
self.hass = ha.HomeAssistant()
|
|
mock_http_component(self.hass)
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
@patch('homeassistant.components.sun.setup')
|
|
def test_if_demo_state_shows_by_default(self, mock_sun_setup):
|
|
""" Test if demo state shows if we give no configuration. """
|
|
demo.setup(self.hass, {demo.DOMAIN: {}})
|
|
|
|
self.assertIsNotNone(self.hass.states.get('a.Demo_Mode'))
|
|
|
|
@patch('homeassistant.components.sun.setup')
|
|
def test_hiding_demo_state(self, mock_sun_setup):
|
|
""" Test if you can hide the demo card. """
|
|
demo.setup(self.hass, {demo.DOMAIN: {'hide_demo_state': 1}})
|
|
|
|
self.assertIsNone(self.hass.states.get('a.Demo_Mode'))
|