github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-07-27 18:58:57 +02:00
Code Releases Activity

Update state helper test to avoid patching utcnow (#93484)

Browse Source
This commit is contained in:
J. Nick Koston 2023-05-24 17:38:09 -05:00 committed by GitHub
parent ee62584570
commit f5eb872eaf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/helpers/test_state.py
View File

@ -3,6 +3,7 @@ import asyncio
from datetime import timedelta
from unittest.mock import Mock, patch
from freezegun import freeze_time
import pytest
from homeassistant.components.sun import STATE_ABOVE_HORIZON, STATE_BELOW_HORIZON
@ -33,20 +34,17 @@ async def test_async_track_states(
point2 = point1 + timedelta(seconds=5)
point3 = point2 + timedelta(seconds=5)
with patch("homeassistant.core.dt_util.utcnow") as mock_utcnow:
mock_utcnow.return_value = point2
with freeze_time(point2) as freezer, state.AsyncTrackStates(hass) as states:
freezer.move_to(point1)
hass.states.async_set("light.test", "on")
with state.AsyncTrackStates(hass) as states:
mock_utcnow.return_value = point1
hass.states.async_set("light.test", "on")
freezer.move_to(point2)
hass.states.async_set("light.test2", "on")
state2 = hass.states.get("light.test2")
mock_utcnow.return_value = point2
hass.states.async_set("light.test2", "on")
state2 = hass.states.get("light.test2")
mock_utcnow.return_value = point3
hass.states.async_set("light.test3", "on")
state3 = hass.states.get("light.test3")
freezer.move_to(point3)
hass.states.async_set("light.test3", "on")
state3 = hass.states.get("light.test3")
assert [state2, state3] == sorted(states, key=lambda state: state.entity_id)
@ -92,15 +90,15 @@ async def test_get_changed_since(
point2 = point1 + timedelta(seconds=5)
point3 = point2 + timedelta(seconds=5)
with patch("homeassistant.core.dt_util.utcnow", return_value=point1):
with freeze_time(point1) as freezer:
hass.states.async_set("light.test", "on")
state1 = hass.states.get("light.test")
with patch("homeassistant.core.dt_util.utcnow", return_value=point2):
freezer.move_to(point2)
hass.states.async_set("light.test2", "on")
state2 = hass.states.get("light.test2")
with patch("homeassistant.core.dt_util.utcnow", return_value=point3):
freezer.move_to(point3)
hass.states.async_set("light.test3", "on")
state3 = hass.states.get("light.test3")