mirror of
https://github.com/home-assistant/core.git
synced 2025-12-27 14:31:13 +00:00
f195ecca4b
* Moved climate components with tests into platform dirs. * Updated tests from climate component. * Moved binary_sensor components with tests into platform dirs. * Updated tests from binary_sensor component. * Moved calendar components with tests into platform dirs. * Updated tests from calendar component. * Moved camera components with tests into platform dirs. * Updated tests from camera component. * Moved cover components with tests into platform dirs. * Updated tests from cover component. * Moved device_tracker components with tests into platform dirs. * Updated tests from device_tracker component. * Moved fan components with tests into platform dirs. * Updated tests from fan component. * Moved geo_location components with tests into platform dirs. * Updated tests from geo_location component. * Moved image_processing components with tests into platform dirs. * Updated tests from image_processing component. * Moved light components with tests into platform dirs. * Updated tests from light component. * Moved lock components with tests into platform dirs. * Moved media_player components with tests into platform dirs. * Updated tests from media_player component. * Moved scene components with tests into platform dirs. * Moved sensor components with tests into platform dirs. * Updated tests from sensor component. * Moved switch components with tests into platform dirs. * Updated tests from sensor component. * Moved vacuum components with tests into platform dirs. * Updated tests from vacuum component. * Moved weather components with tests into platform dirs. * Fixed __init__.py files * Fixes for stuff moved as part of this branch. * Fix stuff needed to merge with balloob's branch. * Formatting issues. * Missing __init__.py files. * Fix-ups * Fixup * Regenerated requirements. * Linting errors fixed. * Fixed more broken tests. * Missing init files. * Fix broken tests. * More broken tests * There seems to be a thread race condition. I suspect the logger stuff is running in another thread, which means waiting until the aio loop is done is missing the log messages. Used sleep instead because that allows the logger thread to run. I think the api_streams sensor might not be thread safe. * Disabled tests, will remove sensor in #22147 * Updated coverage and codeowners.
848 lines
34 KiB
Python
848 lines
34 KiB
Python
"""The tests for the Flux switch platform."""
|
||
import unittest
|
||
from unittest.mock import patch
|
||
|
||
from homeassistant.setup import setup_component
|
||
from homeassistant.components import switch, light
|
||
from homeassistant.const import (
|
||
CONF_PLATFORM, STATE_ON, SERVICE_TURN_ON, SUN_EVENT_SUNRISE)
|
||
import homeassistant.loader as loader
|
||
import homeassistant.util.dt as dt_util
|
||
|
||
from tests.common import (
|
||
assert_setup_component, get_test_home_assistant, fire_time_changed,
|
||
mock_service)
|
||
from tests.components.light import common as common_light
|
||
from tests.components.switch import common
|
||
|
||
|
||
class TestSwitchFlux(unittest.TestCase):
|
||
"""Test the Flux switch platform."""
|
||
|
||
def setUp(self):
|
||
"""Set up things to be run when tests are started."""
|
||
self.hass = get_test_home_assistant()
|
||
|
||
def tearDown(self):
|
||
"""Stop everything that was started."""
|
||
self.hass.stop()
|
||
|
||
def test_valid_config(self):
|
||
"""Test configuration."""
|
||
assert setup_component(self.hass, 'switch', {
|
||
'switch': {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': ['light.desk', 'light.lamp'],
|
||
}
|
||
})
|
||
|
||
def test_valid_config_with_info(self):
|
||
"""Test configuration."""
|
||
assert setup_component(self.hass, 'switch', {
|
||
'switch': {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': ['light.desk', 'light.lamp'],
|
||
'stop_time': '22:59',
|
||
'start_time': '7:22',
|
||
'start_colortemp': '1000',
|
||
'sunset_colortemp': '2000',
|
||
'stop_colortemp': '4000'
|
||
}
|
||
})
|
||
|
||
def test_valid_config_no_name(self):
|
||
"""Test configuration."""
|
||
with assert_setup_component(1, 'switch'):
|
||
assert setup_component(self.hass, 'switch', {
|
||
'switch': {
|
||
'platform': 'flux',
|
||
'lights': ['light.desk', 'light.lamp']
|
||
}
|
||
})
|
||
|
||
def test_invalid_config_no_lights(self):
|
||
"""Test configuration."""
|
||
with assert_setup_component(0, 'switch'):
|
||
assert setup_component(self.hass, 'switch', {
|
||
'switch': {
|
||
'platform': 'flux',
|
||
'name': 'flux'
|
||
}
|
||
})
|
||
|
||
def test_flux_when_switch_is_off(self):
|
||
"""Test the flux switch when it is off."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=10, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||
with patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id]
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
assert 0 == len(turn_on_calls)
|
||
|
||
def test_flux_before_sunrise(self):
|
||
"""Test the flux switch before sunrise."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=2, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||
with patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id]
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 112
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.606, 0.379]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_sunrise_before_sunset(self):
|
||
"""Test the flux switch after sunrise and before sunset."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=8, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id]
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 173
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.439, 0.37]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_sunset_before_stop(self):
|
||
"""Test the flux switch after sunset and before stop."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=17, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '22:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 146
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.506, 0.385]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_stop_before_sunrise(self):
|
||
"""Test the flux switch after stop and before sunrise."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=23, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||
with patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id]
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 112
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.606, 0.379]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_with_custom_start_stop_times(self):
|
||
"""Test the flux with custom start and stop times."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=17, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'start_time': '6:00',
|
||
'stop_time': '23:30'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 147
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.504, 0.385]
|
||
|
||
def test_flux_before_sunrise_stop_next_day(self):
|
||
"""Test the flux switch before sunrise.
|
||
|
||
This test has the stop_time on the next day (after midnight).
|
||
"""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=2, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '01:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 112
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.606, 0.379]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_sunrise_before_sunset_stop_next_day(self):
|
||
"""
|
||
Test the flux switch after sunrise and before sunset.
|
||
|
||
This test has the stop_time on the next day (after midnight).
|
||
"""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=8, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '01:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 173
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.439, 0.37]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_sunset_before_midnight_stop_next_day(self):
|
||
"""Test the flux switch after sunset and before stop.
|
||
|
||
This test has the stop_time on the next day (after midnight).
|
||
"""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=23, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.util.dt.utcnow', return_value=test_time):
|
||
with patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '01:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 119
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.588, 0.386]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_sunset_after_midnight_stop_next_day(self):
|
||
"""Test the flux switch after sunset and before stop.
|
||
|
||
This test has the stop_time on the next day (after midnight).
|
||
"""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=00, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '01:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 114
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.601, 0.382]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_after_stop_before_sunrise_stop_next_day(self):
|
||
"""Test the flux switch after stop and before sunrise.
|
||
|
||
This test has the stop_time on the next day (after midnight).
|
||
"""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=2, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'stop_time': '01:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 112
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.606, 0.379]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_with_custom_colortemps(self):
|
||
"""Test the flux with custom start and stop colortemps."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=17, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'start_colortemp': '1000',
|
||
'stop_colortemp': '6000',
|
||
'stop_time': '22:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 159
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.469, 0.378]
|
||
|
||
# pylint: disable=invalid-name
|
||
def test_flux_with_custom_brightness(self):
|
||
"""Test the flux with custom start and stop colortemps."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=17, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'brightness': 255,
|
||
'stop_time': '22:00'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 255
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.506, 0.385]
|
||
|
||
def test_flux_with_multiple_lights(self):
|
||
"""Test the flux switch with multiple light entities."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1, dev2, dev3 = platform.DEVICES
|
||
common_light.turn_on(self.hass, entity_id=dev2.entity_id)
|
||
self.hass.block_till_done()
|
||
common_light.turn_on(self.hass, entity_id=dev3.entity_id)
|
||
self.hass.block_till_done()
|
||
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
state = self.hass.states.get(dev2.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
state = self.hass.states.get(dev3.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('xy_color') is None
|
||
assert state.attributes.get('brightness') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=12, minute=0, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
print('sunrise {}'.format(sunrise_time))
|
||
return sunrise_time
|
||
print('sunset {}'.format(sunset_time))
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id,
|
||
dev2.entity_id,
|
||
dev3.entity_id]
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 163
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.46, 0.376]
|
||
call = turn_on_calls[-2]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 163
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.46, 0.376]
|
||
call = turn_on_calls[-3]
|
||
assert call.data[light.ATTR_BRIGHTNESS] == 163
|
||
assert call.data[light.ATTR_XY_COLOR] == [0.46, 0.376]
|
||
|
||
def test_flux_with_mired(self):
|
||
"""Test the flux switch´s mode mired."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('color_temp') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=8, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'mode': 'mired'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
assert call.data[light.ATTR_COLOR_TEMP] == 269
|
||
|
||
def test_flux_with_rgb(self):
|
||
"""Test the flux switch´s mode rgb."""
|
||
platform = loader.get_component(self.hass, 'light.test')
|
||
platform.init()
|
||
assert setup_component(self.hass, light.DOMAIN,
|
||
{light.DOMAIN: {CONF_PLATFORM: 'test'}})
|
||
|
||
dev1 = platform.DEVICES[0]
|
||
|
||
# Verify initial state of light
|
||
state = self.hass.states.get(dev1.entity_id)
|
||
assert STATE_ON == state.state
|
||
assert state.attributes.get('color_temp') is None
|
||
|
||
test_time = dt_util.utcnow().replace(hour=8, minute=30, second=0)
|
||
sunset_time = test_time.replace(hour=17, minute=0, second=0)
|
||
sunrise_time = test_time.replace(hour=5, minute=0, second=0)
|
||
|
||
def event_date(hass, event, now=None):
|
||
if event == SUN_EVENT_SUNRISE:
|
||
return sunrise_time
|
||
return sunset_time
|
||
|
||
with patch('homeassistant.components.flux.switch.dt_utcnow',
|
||
return_value=test_time), \
|
||
patch('homeassistant.helpers.sun.get_astral_event_date',
|
||
side_effect=event_date):
|
||
assert setup_component(self.hass, switch.DOMAIN, {
|
||
switch.DOMAIN: {
|
||
'platform': 'flux',
|
||
'name': 'flux',
|
||
'lights': [dev1.entity_id],
|
||
'mode': 'rgb'
|
||
}
|
||
})
|
||
turn_on_calls = mock_service(
|
||
self.hass, light.DOMAIN, SERVICE_TURN_ON)
|
||
common.turn_on(self.hass, 'switch.flux')
|
||
self.hass.block_till_done()
|
||
fire_time_changed(self.hass, test_time)
|
||
self.hass.block_till_done()
|
||
call = turn_on_calls[-1]
|
||
rgb = (255, 198, 152)
|
||
rounded_call = tuple(map(round, call.data[light.ATTR_RGB_COLOR]))
|
||
assert rounded_call == rgb
|