79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
"""The tests for the demo platform."""
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components import geo_location
|
|
from homeassistant.components.demo.geo_location import (
|
|
DEFAULT_UNIT_OF_MEASUREMENT,
|
|
DEFAULT_UPDATE_INTERVAL,
|
|
NUMBER_OF_DEMO_DEVICES,
|
|
)
|
|
from homeassistant.setup import setup_component
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from tests.common import (
|
|
assert_setup_component,
|
|
fire_time_changed,
|
|
get_test_home_assistant,
|
|
)
|
|
|
|
CONFIG = {geo_location.DOMAIN: [{"platform": "demo"}]}
|
|
|
|
|
|
class TestDemoPlatform(unittest.TestCase):
|
|
"""Test the demo platform."""
|
|
|
|
def setUp(self):
|
|
"""Initialize values for this testcase class."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
def tearDown(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_setup_platform(self):
|
|
"""Test setup of demo platform via configuration."""
|
|
utcnow = dt_util.utcnow()
|
|
# Patching 'utcnow' to gain more control over the timed update.
|
|
with patch("homeassistant.util.dt.utcnow", return_value=utcnow):
|
|
with assert_setup_component(1, geo_location.DOMAIN):
|
|
assert setup_component(self.hass, geo_location.DOMAIN, CONFIG)
|
|
self.hass.block_till_done()
|
|
|
|
# In this test, one zone and geolocation entities have been
|
|
# generated.
|
|
all_states = [
|
|
self.hass.states.get(entity_id)
|
|
for entity_id in self.hass.states.entity_ids(geo_location.DOMAIN)
|
|
]
|
|
assert len(all_states) == NUMBER_OF_DEMO_DEVICES
|
|
|
|
for state in all_states:
|
|
# Check a single device's attributes.
|
|
if state.domain != geo_location.DOMAIN:
|
|
# ignore home zone state
|
|
continue
|
|
assert (
|
|
abs(state.attributes["latitude"] - self.hass.config.latitude) < 1.0
|
|
)
|
|
assert (
|
|
abs(state.attributes["longitude"] - self.hass.config.longitude)
|
|
< 1.0
|
|
)
|
|
assert (
|
|
state.attributes["unit_of_measurement"]
|
|
== DEFAULT_UNIT_OF_MEASUREMENT
|
|
)
|
|
|
|
# Update (replaces 1 device).
|
|
fire_time_changed(self.hass, utcnow + DEFAULT_UPDATE_INTERVAL)
|
|
self.hass.block_till_done()
|
|
# Get all states again, ensure that the number of states is still
|
|
# the same, but the lists are different.
|
|
all_states_updated = [
|
|
self.hass.states.get(entity_id)
|
|
for entity_id in self.hass.states.entity_ids(geo_location.DOMAIN)
|
|
]
|
|
assert len(all_states_updated) == NUMBER_OF_DEMO_DEVICES
|
|
assert all_states != all_states_updated
|