Async device sun light trigger tests (#18689)
Adam Mills
GitHub
parent
f3ce463862
commit
91c526d9fe
|
|
@ -1,115 +1,114 @@
|
|||
"""The tests device sun light trigger component."""
|
||||
# pylint: disable=protected-access
|
||||
from datetime import datetime
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
from asynctest import patch
|
||||
import pytest
|
||||
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.setup import async_setup_component
|
||||
import homeassistant.loader as loader
|
||||
from homeassistant.const import CONF_PLATFORM, STATE_HOME, STATE_NOT_HOME
|
||||
from homeassistant.components import (
|
||||
device_tracker, light, device_sun_light_trigger)
|
||||
from homeassistant.util import dt as dt_util
|
||||
|
||||
from tests.common import get_test_home_assistant, fire_time_changed
|
||||
from tests.common import async_fire_time_changed
|
||||
from tests.components.light import common as common_light
|
||||
|
||||
|
||||
class TestDeviceSunLightTrigger(unittest.TestCase):
|
||||
"""Test the device sun light trigger module."""
|
||||
@pytest.fixture
|
||||
def scanner(hass):
|
||||
"""Initialize components."""
|
||||
scanner = loader.get_component(
|
||||
hass, 'device_tracker.test').get_scanner(None, None)
|
||||
|
||||
def setUp(self): # pylint: disable=invalid-name
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
scanner.reset()
|
||||
scanner.come_home('DEV1')
|
||||
|
||||
self.scanner = loader.get_component(
|
||||
self.hass, 'device_tracker.test').get_scanner(None, None)
|
||||
loader.get_component(hass, 'light.test').init()
|
||||
|
||||
self.scanner.reset()
|
||||
self.scanner.come_home('DEV1')
|
||||
|
||||
loader.get_component(self.hass, 'light.test').init()
|
||||
|
||||
with patch(
|
||||
'homeassistant.components.device_tracker.load_yaml_config_file',
|
||||
return_value={
|
||||
'device_1': {
|
||||
'hide_if_away': False,
|
||||
'mac': 'DEV1',
|
||||
'name': 'Unnamed Device',
|
||||
'picture': 'http://example.com/dev1.jpg',
|
||||
'track': True,
|
||||
'vendor': None
|
||||
},
|
||||
'device_2': {
|
||||
'hide_if_away': False,
|
||||
'mac': 'DEV2',
|
||||
'name': 'Unnamed Device',
|
||||
'picture': 'http://example.com/dev2.jpg',
|
||||
'track': True,
|
||||
'vendor': None}
|
||||
}):
|
||||
assert setup_component(self.hass, device_tracker.DOMAIN, {
|
||||
with patch(
|
||||
'homeassistant.components.device_tracker.load_yaml_config_file',
|
||||
return_value={
|
||||
'device_1': {
|
||||
'hide_if_away': False,
|
||||
'mac': 'DEV1',
|
||||
'name': 'Unnamed Device',
|
||||
'picture': 'http://example.com/dev1.jpg',
|
||||
'track': True,
|
||||
'vendor': None
|
||||
},
|
||||
'device_2': {
|
||||
'hide_if_away': False,
|
||||
'mac': 'DEV2',
|
||||
'name': 'Unnamed Device',
|
||||
'picture': 'http://example.com/dev2.jpg',
|
||||
'track': True,
|
||||
'vendor': None}
|
||||
}):
|
||||
assert hass.loop.run_until_complete(async_setup_component(
|
||||
hass, device_tracker.DOMAIN, {
|
||||
device_tracker.DOMAIN: {CONF_PLATFORM: 'test'}
|
||||
})
|
||||
}))
|
||||
|
||||
assert setup_component(self.hass, light.DOMAIN, {
|
||||
assert hass.loop.run_until_complete(async_setup_component(
|
||||
hass, light.DOMAIN, {
|
||||
light.DOMAIN: {CONF_PLATFORM: 'test'}
|
||||
})
|
||||
}))
|
||||
|
||||
def tearDown(self): # pylint: disable=invalid-name
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
return scanner
|
||||
|
||||
def test_lights_on_when_sun_sets(self):
|
||||
"""Test lights go on when there is someone home and the sun sets."""
|
||||
test_time = datetime(2017, 4, 5, 1, 2, 3, tzinfo=dt_util.UTC)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
assert setup_component(
|
||||
self.hass, device_sun_light_trigger.DOMAIN, {
|
||||
device_sun_light_trigger.DOMAIN: {}})
|
||||
|
||||
common_light.turn_off(self.hass)
|
||||
|
||||
self.hass.block_till_done()
|
||||
|
||||
test_time = test_time.replace(hour=3)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
fire_time_changed(self.hass, test_time)
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert light.is_on(self.hass)
|
||||
|
||||
def test_lights_turn_off_when_everyone_leaves(self):
|
||||
"""Test lights turn off when everyone leaves the house."""
|
||||
common_light.turn_on(self.hass)
|
||||
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert setup_component(
|
||||
self.hass, device_sun_light_trigger.DOMAIN, {
|
||||
async def test_lights_on_when_sun_sets(hass, scanner):
|
||||
"""Test lights go on when there is someone home and the sun sets."""
|
||||
test_time = datetime(2017, 4, 5, 1, 2, 3, tzinfo=dt_util.UTC)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
assert await async_setup_component(
|
||||
hass, device_sun_light_trigger.DOMAIN, {
|
||||
device_sun_light_trigger.DOMAIN: {}})
|
||||
|
||||
self.hass.states.set(device_tracker.ENTITY_ID_ALL_DEVICES,
|
||||
STATE_NOT_HOME)
|
||||
common_light.async_turn_off(hass)
|
||||
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not light.is_on(self.hass)
|
||||
test_time = test_time.replace(hour=3)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
async_fire_time_changed(hass, test_time)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
def test_lights_turn_on_when_coming_home_after_sun_set(self):
|
||||
"""Test lights turn on when coming home after sun set."""
|
||||
test_time = datetime(2017, 4, 5, 3, 2, 3, tzinfo=dt_util.UTC)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
common_light.turn_off(self.hass)
|
||||
self.hass.block_till_done()
|
||||
assert light.is_on(hass)
|
||||
|
||||
assert setup_component(
|
||||
self.hass, device_sun_light_trigger.DOMAIN, {
|
||||
device_sun_light_trigger.DOMAIN: {}})
|
||||
|
||||
self.hass.states.set(
|
||||
device_tracker.ENTITY_ID_FORMAT.format('device_2'), STATE_HOME)
|
||||
async def test_lights_turn_off_when_everyone_leaves(hass, scanner):
|
||||
"""Test lights turn off when everyone leaves the house."""
|
||||
common_light.async_turn_on(hass)
|
||||
|
||||
self.hass.block_till_done()
|
||||
assert light.is_on(self.hass)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert await async_setup_component(
|
||||
hass, device_sun_light_trigger.DOMAIN, {
|
||||
device_sun_light_trigger.DOMAIN: {}})
|
||||
|
||||
hass.states.async_set(device_tracker.ENTITY_ID_ALL_DEVICES,
|
||||
STATE_NOT_HOME)
|
||||
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert not light.is_on(hass)
|
||||
|
||||
|
||||
async def test_lights_turn_on_when_coming_home_after_sun_set(hass, scanner):
|
||||
"""Test lights turn on when coming home after sun set."""
|
||||
test_time = datetime(2017, 4, 5, 3, 2, 3, tzinfo=dt_util.UTC)
|
||||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||||
common_light.async_turn_off(hass)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert await async_setup_component(
|
||||
hass, device_sun_light_trigger.DOMAIN, {
|
||||
device_sun_light_trigger.DOMAIN: {}})
|
||||
|
||||
hass.states.async_set(
|
||||
device_tracker.ENTITY_ID_FORMAT.format('device_2'), STATE_HOME)
|
||||
|
||||
await hass.async_block_till_done()
|
||||
assert light.is_on(hass)
|
||||
|
|
|
|||
Loading…
Reference in New Issue