diff --git a/tests/components/automation/test_sun.py b/tests/components/automation/test_sun.py index 030662da05c..9f0698a5ec7 100644 --- a/tests/components/automation/test_sun.py +++ b/tests/components/automation/test_sun.py @@ -14,10 +14,12 @@ from tests.common import ( async_fire_time_changed, mock_component, async_mock_service) from tests.components.automation import common +ORIG_TIME_ZONE = dt_util.DEFAULT_TIME_ZONE + @pytest.fixture def calls(hass): - """Track calls to a mock serivce.""" + """Track calls to a mock service.""" return async_mock_service(hass, 'test', 'automation') @@ -25,10 +27,16 @@ def calls(hass): def setup_comp(hass): """Initialize components.""" mock_component(hass, 'group') + dt_util.set_default_time_zone(hass.config.time_zone) hass.loop.run_until_complete(async_setup_component(hass, sun.DOMAIN, { sun.DOMAIN: {sun.CONF_ELEVATION: 0}})) +def teardown(): + """Restore.""" + dt_util.set_default_time_zone(ORIG_TIME_ZONE) + + async def test_sunset_trigger(hass, calls): """Test the sunset trigger.""" now = datetime(2015, 9, 15, 23, tzinfo=dt_util.UTC) @@ -145,8 +153,12 @@ async def test_sunrise_trigger_with_offset(hass, calls): assert 1 == len(calls) -async def test_if_action_before(hass, calls): - """Test if action was before.""" +async def test_if_action_before_sunrise_no_offset(hass, calls): + """ + Test if action was before sunrise. + + Before sunrise is true from midnight until sunset, local time. + """ await async_setup_component(hass, automation.DOMAIN, { automation.DOMAIN: { 'trigger': { @@ -163,23 +175,47 @@ async def test_if_action_before(hass, calls): } }) - now = datetime(2015, 9, 16, 15, tzinfo=dt_util.UTC) + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = sunrise + 1s -> 'before sunrise' not true + now = datetime(2015, 9, 16, 13, 32, 44, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 0 == len(calls) - now = datetime(2015, 9, 16, 10, tzinfo=dt_util.UTC) + # now = sunrise -> 'before sunrise' true + now = datetime(2015, 9, 16, 13, 32, 43, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 1 == len(calls) + # now = local midnight -> 'before sunrise' true + now = datetime(2015, 9, 16, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) -async def test_if_action_after(hass, calls): - """Test if action was after.""" + # now = local midnight - 1s -> 'before sunrise' not true + now = datetime(2015, 9, 17, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + +async def test_if_action_after_sunrise_no_offset(hass, calls): + """ + Test if action was after sunrise. + + After sunrise is true from sunrise until midnight, local time. + """ await async_setup_component(hass, automation.DOMAIN, { automation.DOMAIN: { 'trigger': { @@ -196,23 +232,47 @@ async def test_if_action_after(hass, calls): } }) - now = datetime(2015, 9, 16, 13, tzinfo=dt_util.UTC) + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = sunrise - 1s -> 'after sunrise' not true + now = datetime(2015, 9, 16, 13, 32, 42, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 0 == len(calls) - now = datetime(2015, 9, 16, 15, tzinfo=dt_util.UTC) + # now = sunrise + 1s -> 'after sunrise' true + now = datetime(2015, 9, 16, 13, 32, 43, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 1 == len(calls) + # now = local midnight -> 'after sunrise' not true + now = datetime(2015, 9, 16, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) -async def test_if_action_before_with_offset(hass, calls): - """Test if action was before offset.""" + # now = local midnight - 1s -> 'after sunrise' true + now = datetime(2015, 9, 17, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + +async def test_if_action_before_sunrise_with_offset(hass, calls): + """ + Test if action was before sunrise with offset. + + Before sunrise is true from midnight until sunset, local time. + """ await async_setup_component(hass, automation.DOMAIN, { automation.DOMAIN: { 'trigger': { @@ -230,6 +290,9 @@ async def test_if_action_before_with_offset(hass, calls): } }) + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = sunrise + 1s + 1h -> 'before sunrise' with offset +1h not true now = datetime(2015, 9, 16, 14, 32, 44, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): @@ -237,6 +300,7 @@ async def test_if_action_before_with_offset(hass, calls): await hass.async_block_till_done() assert 0 == len(calls) + # now = sunrise + 1h -> 'before sunrise' with offset +1h true now = datetime(2015, 9, 16, 14, 32, 43, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): @@ -244,9 +308,151 @@ async def test_if_action_before_with_offset(hass, calls): await hass.async_block_till_done() assert 1 == len(calls) + # now = UTC midnight -> 'before sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 0, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) -async def test_if_action_after_with_offset(hass, calls): - """Test if action was after offset.""" + # now = UTC midnight - 1s -> 'before sunrise' with offset +1h not true + now = datetime(2015, 9, 16, 23, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = local midnight -> 'before sunrise' with offset +1h true + now = datetime(2015, 9, 16, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = local midnight - 1s -> 'before sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = sunset -> 'before sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 1, 56, 48, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = sunset -1s -> 'before sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 1, 56, 45, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + +async def test_if_action_before_sunset_with_offset(hass, calls): + """ + Test if action was before sunset with offset. + + Before sunset is true from midnight until sunset, local time. + """ + await async_setup_component(hass, automation.DOMAIN, { + automation.DOMAIN: { + 'trigger': { + 'platform': 'event', + 'event_type': 'test_event', + }, + 'condition': { + 'condition': 'sun', + 'before': 'sunset', + 'before_offset': '+1:00:00' + }, + 'action': { + 'service': 'test.automation' + } + } + }) + + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = local midnight -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 16, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = sunset + 1s + 1h -> 'before sunset' with offset +1h not true + now = datetime(2015, 9, 17, 2, 55, 25, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = sunset + 1h -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 17, 2, 55, 24, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = UTC midnight -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 17, 0, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 3 == len(calls) + + # now = UTC midnight - 1s -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 16, 23, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 4 == len(calls) + + # now = sunrise -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 16, 13, 32, 43, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 5 == len(calls) + + # now = sunrise -1s -> 'before sunset' with offset +1h true + now = datetime(2015, 9, 16, 13, 32, 42, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 6 == len(calls) + + # now = local midnight-1s -> 'after sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 6 == len(calls) + + +async def test_if_action_after_sunrise_with_offset(hass, calls): + """ + Test if action was after sunrise with offset. + + After sunrise is true from sunrise until midnight, local time. + """ await async_setup_component(hass, automation.DOMAIN, { automation.DOMAIN: { 'trigger': { @@ -264,6 +470,9 @@ async def test_if_action_after_with_offset(hass, calls): } }) + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = sunrise - 1s + 1h -> 'after sunrise' with offset +1h not true now = datetime(2015, 9, 16, 14, 32, 42, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): @@ -271,6 +480,7 @@ async def test_if_action_after_with_offset(hass, calls): await hass.async_block_till_done() assert 0 == len(calls) + # now = sunrise + 1h -> 'after sunrise' with offset +1h true now = datetime(2015, 9, 16, 14, 32, 43, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): @@ -278,9 +488,135 @@ async def test_if_action_after_with_offset(hass, calls): await hass.async_block_till_done() assert 1 == len(calls) + # now = UTC noon -> 'after sunrise' with offset +1h not true + now = datetime(2015, 9, 16, 12, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = UTC noon - 1s -> 'after sunrise' with offset +1h not true + now = datetime(2015, 9, 16, 11, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = local noon -> 'after sunrise' with offset +1h true + now = datetime(2015, 9, 16, 19, 1, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = local noon - 1s -> 'after sunrise' with offset +1h true + now = datetime(2015, 9, 16, 18, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 3 == len(calls) + + # now = sunset -> 'after sunrise' with offset +1h true + now = datetime(2015, 9, 17, 1, 55, 24, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 4 == len(calls) + + # now = sunset + 1s -> 'after sunrise' with offset +1h true + now = datetime(2015, 9, 17, 1, 55, 25, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 5 == len(calls) + + # now = local midnight-1s -> 'after sunrise' with offset +1h true + now = datetime(2015, 9, 17, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 6 == len(calls) + + # now = local midnight -> 'after sunrise' with offset +1h not true + now = datetime(2015, 9, 17, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 6 == len(calls) + + +async def test_if_action_after_sunset_with_offset(hass, calls): + """ + Test if action was after sunset with offset. + + After sunset is true from sunset until midnight, local time. + """ + await async_setup_component(hass, automation.DOMAIN, { + automation.DOMAIN: { + 'trigger': { + 'platform': 'event', + 'event_type': 'test_event', + }, + 'condition': { + 'condition': 'sun', + 'after': 'sunset', + 'after_offset': '+1:00:00' + }, + 'action': { + 'service': 'test.automation' + } + } + }) + + # sunrise: 2015-09-15 06:32:05 local, sunset: 2015-09-15 18:56:46 local + # sunrise: 2015-09-15 13:32:05 UTC, sunset: 2015-09-16 01:56:46 UTC + # now = sunset - 1s + 1h -> 'after sunset' with offset +1h not true + now = datetime(2015, 9, 16, 2, 56, 45, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 0 == len(calls) + + # now = sunset + 1h -> 'after sunset' with offset +1h true + now = datetime(2015, 9, 16, 2, 56, 46, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 1 == len(calls) + + # now = midnight-1s -> 'after sunset' with offset +1h true + now = datetime(2015, 9, 16, 6, 59, 59, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = midnight -> 'after sunset' with offset +1h not true + now = datetime(2015, 9, 16, 7, 0, 0, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + async def test_if_action_before_and_after_during(hass, calls): - """Test if action was before and after during.""" + """ + Test if action was after sunset and before sunrise. + + This is true from sunrise until sunset. + """ await async_setup_component(hass, automation.DOMAIN, { automation.DOMAIN: { 'trigger': { @@ -298,23 +634,44 @@ async def test_if_action_before_and_after_during(hass, calls): } }) - now = datetime(2015, 9, 16, 13, 8, 51, tzinfo=dt_util.UTC) + # sunrise: 2015-09-16 06:32:43 local, sunset: 2015-09-16 18:55:24 local + # sunrise: 2015-09-16 13:32:43 UTC, sunset: 2015-09-17 01:55:24 UTC + # now = sunrise - 1s -> 'after sunrise' + 'before sunset' not true + now = datetime(2015, 9, 16, 13, 32, 42, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 0 == len(calls) - now = datetime(2015, 9, 17, 2, 25, 18, tzinfo=dt_util.UTC) + # now = sunset + 1s -> 'after sunrise' + 'before sunset' not true + now = datetime(2015, 9, 17, 1, 55, 25, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 0 == len(calls) - now = datetime(2015, 9, 16, 16, tzinfo=dt_util.UTC) + # now = sunrise -> 'after sunrise' + 'before sunset' true + now = datetime(2015, 9, 16, 13, 32, 43, tzinfo=dt_util.UTC) with patch('homeassistant.util.dt.utcnow', return_value=now): hass.bus.async_fire('test_event') await hass.async_block_till_done() assert 1 == len(calls) + + # now = sunset -> 'after sunrise' + 'before sunset' true + now = datetime(2015, 9, 17, 1, 55, 24, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 2 == len(calls) + + # now = 9AM local -> 'after sunrise' + 'before sunset' true + now = datetime(2015, 9, 16, 16, tzinfo=dt_util.UTC) + with patch('homeassistant.util.dt.utcnow', + return_value=now): + hass.bus.async_fire('test_event') + await hass.async_block_till_done() + assert 3 == len(calls)