jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 09:47:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use freezegun in nsw_rural_fire_service_feed test (#106050)

Browse Source
This commit is contained in:
Jan-Philipp Benecke 2023-12-19 12:20:56 +01:00 committed by GitHub
parent 2d31f9e984
commit c50098a845
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/nsw_rural_fire_service_feed/test_geo_location.py
View File

@ -3,6 +3,7 @@ import datetime
from unittest.mock import ANY, MagicMock, call, patch
from aio_geojson_nsw_rfs_incidents import NswRuralFireServiceIncidentsFeed
from freezegun.api import FrozenDateTimeFactory
from homeassistant.components import geo_location
from homeassistant.components.geo_location import ATTR_SOURCE
@ -90,7 +91,7 @@ def _generate_mock_feed_entry(
return feed_entry
async def test_setup(hass: HomeAssistant) -> None:
async def test_setup(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> None:
"""Test the general setup of the platform."""
# Set up some mock feed entries for this test.
mock_entry_1 = _generate_mock_feed_entry(
@ -114,11 +115,10 @@ async def test_setup(hass: HomeAssistant) -> None:
mock_entry_3 = _generate_mock_feed_entry("3456", "Title 3", 25.5, (-31.2, 150.2))
mock_entry_4 = _generate_mock_feed_entry("4567", "Title 4", 12.5, (-31.3, 150.3))
# Patching 'utcnow' to gain more control over the timed update.
utcnow = dt_util.utcnow()
with patch("homeassistant.util.dt.utcnow", return_value=utcnow), patch(
"aio_geojson_client.feed.GeoJsonFeed.update"
) as mock_feed_update:
freezer.move_to(utcnow)
with patch("aio_geojson_client.feed.GeoJsonFeed.update") as mock_feed_update:
mock_feed_update.return_value = (
"OK",
[mock_entry_1, mock_entry_2, mock_entry_3],