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