From da5ad4a17d18fc9f4a9d459c752148e35f70df0d Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Wed, 24 May 2023 13:17:11 -0500 Subject: [PATCH] Update binary_sensor tests to avoid patching utcnow (#93474) --- tests/components/binary_sensor/test_device_condition.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/components/binary_sensor/test_device_condition.py b/tests/components/binary_sensor/test_device_condition.py index d476d965144..ecc5f7bc8b0 100644 --- a/tests/components/binary_sensor/test_device_condition.py +++ b/tests/components/binary_sensor/test_device_condition.py @@ -1,7 +1,7 @@ """The test for binary_sensor device automation.""" from datetime import timedelta -from unittest.mock import patch +from freezegun import freeze_time import pytest import homeassistant.components.automation as automation @@ -285,8 +285,7 @@ async def test_if_fires_on_for_condition( sensor1 = platform.ENTITIES["battery"] - with patch("homeassistant.core.dt_util.utcnow") as mock_utcnow: - mock_utcnow.return_value = point1 + with freeze_time(point1) as time_freeze: assert await async_setup_component( hass, automation.DOMAIN, @@ -324,7 +323,7 @@ async def test_if_fires_on_for_condition( assert len(calls) == 0 # Time travel 10 secs into the future - mock_utcnow.return_value = point2 + time_freeze.move_to(point2) hass.bus.async_fire("test_event1") await hass.async_block_till_done() assert len(calls) == 0 @@ -335,7 +334,7 @@ async def test_if_fires_on_for_condition( assert len(calls) == 0 # Time travel 20 secs into the future - mock_utcnow.return_value = point3 + time_freeze.move_to(point3) hass.bus.async_fire("test_event1") await hass.async_block_till_done() assert len(calls) == 1