53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""The tests for the Demo component."""
|
|
import json
|
|
import os
|
|
import unittest
|
|
|
|
from homeassistant.setup import setup_component
|
|
from homeassistant.components import demo, device_tracker
|
|
from homeassistant.remote import JSONEncoder
|
|
|
|
from tests.common import mock_http_component, get_test_home_assistant
|
|
|
|
|
|
class TestDemo(unittest.TestCase):
|
|
"""Test the Demo component."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
mock_http_component(self.hass)
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
try:
|
|
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
def test_if_demo_state_shows_by_default(self):
|
|
"""Test if demo state shows if we give no configuration."""
|
|
setup_component(self.hass, demo.DOMAIN, {demo.DOMAIN: {}})
|
|
|
|
self.assertIsNotNone(self.hass.states.get('a.Demo_Mode'))
|
|
|
|
def test_hiding_demo_state(self):
|
|
"""Test if you can hide the demo card."""
|
|
setup_component(self.hass, demo.DOMAIN, {
|
|
demo.DOMAIN: {'hide_demo_state': 1}})
|
|
|
|
self.assertIsNone(self.hass.states.get('a.Demo_Mode'))
|
|
|
|
def test_all_entities_can_be_loaded_over_json(self):
|
|
"""Test if you can hide the demo card."""
|
|
setup_component(self.hass, demo.DOMAIN, {
|
|
demo.DOMAIN: {'hide_demo_state': 1}})
|
|
|
|
try:
|
|
json.dumps(self.hass.states.all(), cls=JSONEncoder)
|
|
except Exception:
|
|
self.fail('Unable to convert all demo entities to JSON. '
|
|
'Wrong data in state machine!')
|