From bdaf269ba36d9fa348709c49474e26f8f14d8077 Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke Date: Tue, 2 Jan 2024 12:53:36 +0100 Subject: [PATCH] Migrate geo_rss_events test to use freezegun (#105895) --- tests/components/geo_rss_events/test_sensor.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/components/geo_rss_events/test_sensor.py b/tests/components/geo_rss_events/test_sensor.py index 02225df3755..c86ef393875 100644 --- a/tests/components/geo_rss_events/test_sensor.py +++ b/tests/components/geo_rss_events/test_sensor.py @@ -1,6 +1,7 @@ """The test for the geo rss events sensor platform.""" from unittest.mock import MagicMock, patch +from freezegun.api import FrozenDateTimeFactory import pytest from homeassistant.components import sensor @@ -56,7 +57,9 @@ def _generate_mock_feed_entry( return feed_entry -async def test_setup(hass: HomeAssistant, mock_feed) -> None: +async def test_setup( + hass: HomeAssistant, freezer: FrozenDateTimeFactory, mock_feed +) -> None: """Test the general setup of the platform.""" # Set up some mock feed entries for this test. mock_entry_1 = _generate_mock_feed_entry( @@ -68,10 +71,8 @@ async def test_setup(hass: HomeAssistant, mock_feed) -> None: mock_feed.return_value.update.return_value = "OK", [mock_entry_1, mock_entry_2] utcnow = dt_util.utcnow() - # Patching 'utcnow' to gain more control over the timed update. - with patch( - "homeassistant.util.dt.utcnow", return_value=utcnow - ), assert_setup_component(1, sensor.DOMAIN): + freezer.move_to(utcnow) + with assert_setup_component(1, sensor.DOMAIN): assert await async_setup_component(hass, sensor.DOMAIN, VALID_CONFIG) # Artificially trigger update. hass.bus.fire(EVENT_HOMEASSISTANT_START)