jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix test_wait_template_with_utcnow (#45356)

Browse Source
This commit is contained in:
J. Nick Koston 2021-01-20 16:56:00 -06:00 committed by GitHub
parent 96448c6778
commit 243014bff1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/helpers/test_script.py
View File

@ -824,6 +824,8 @@ async def test_wait_template_with_utcnow(hass):
start_time = dt_util.utcnow().replace(minute=1) + timedelta(hours=48) start_time = dt_util.utcnow().replace(minute=1) + timedelta(hours=48)
try: try:
non_maching_time = start_time.replace(hour=3)
with patch("homeassistant.util.dt.utcnow", return_value=non_maching_time):
hass.async_create_task(script_obj.async_run(context=Context())) hass.async_create_task(script_obj.async_run(context=Context()))
await asyncio.wait_for(wait_started_flag.wait(), 1) await asyncio.wait_for(wait_started_flag.wait(), 1)
assert script_obj.is_running assert script_obj.is_running
@ -848,13 +850,17 @@ async def test_wait_template_with_utcnow_no_match(hass):
timed_out = False timed_out = False
try: try:
non_maching_time = start_time.replace(hour=3)
with patch("homeassistant.util.dt.utcnow", return_value=non_maching_time):
hass.async_create_task(script_obj.async_run(context=Context())) hass.async_create_task(script_obj.async_run(context=Context()))
await asyncio.wait_for(wait_started_flag.wait(), 1) await asyncio.wait_for(wait_started_flag.wait(), 1)
assert script_obj.is_running assert script_obj.is_running
non_maching_time = start_time.replace(hour=3) second_non_maching_time = start_time.replace(hour=4)
with patch("homeassistant.util.dt.utcnow", return_value=non_maching_time): with patch(
async_fire_time_changed(hass, non_maching_time) "homeassistant.util.dt.utcnow", return_value=second_non_maching_time
):
async_fire_time_changed(hass, second_non_maching_time)
with timeout(0.1): with timeout(0.1):
await hass.async_block_till_done() await hass.async_block_till_done()