jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 06:07:17 +00:00
Code Issues Packages Projects Releases Wiki 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

18
tests/helpers/test_state.py
View File

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