mirror of
https://github.com/home-assistant/core
synced 2024-10-05 11:17:53 +00:00
Use freezegun in ign_sismologia test (#106051)
This commit is contained in:
parent
f20e4b9df1
commit
2d31f9e984
|
@ -2,6 +2,8 @@
|
||||||
import datetime
|
import datetime
|
||||||
from unittest.mock import MagicMock, call, patch
|
from unittest.mock import MagicMock, call, patch
|
||||||
|
|
||||||
|
from freezegun.api import FrozenDateTimeFactory
|
||||||
|
|
||||||
from homeassistant.components import geo_location
|
from homeassistant.components import geo_location
|
||||||
from homeassistant.components.geo_location import ATTR_SOURCE
|
from homeassistant.components.geo_location import ATTR_SOURCE
|
||||||
from homeassistant.components.ign_sismologia.geo_location import (
|
from homeassistant.components.ign_sismologia.geo_location import (
|
||||||
|
@ -71,7 +73,7 @@ def _generate_mock_feed_entry(
|
||||||
return 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."""
|
"""Test the general setup of the platform."""
|
||||||
# Set up some mock feed entries for this test.
|
# Set up some mock feed entries for this test.
|
||||||
mock_entry_1 = _generate_mock_feed_entry(
|
mock_entry_1 = _generate_mock_feed_entry(
|
||||||
|
@ -93,11 +95,10 @@ async def test_setup(hass: HomeAssistant) -> None:
|
||||||
)
|
)
|
||||||
mock_entry_4 = _generate_mock_feed_entry("4567", "Title 4", 12.5, (38.3, -3.3))
|
mock_entry_4 = _generate_mock_feed_entry("4567", "Title 4", 12.5, (38.3, -3.3))
|
||||||
|
|
||||||
# Patching 'utcnow' to gain more control over the timed update.
|
|
||||||
utcnow = dt_util.utcnow()
|
utcnow = dt_util.utcnow()
|
||||||
with patch("homeassistant.util.dt.utcnow", return_value=utcnow), patch(
|
freezer.move_to(utcnow)
|
||||||
"georss_ign_sismologia_client.IgnSismologiaFeed"
|
|
||||||
) as mock_feed:
|
with patch("georss_ign_sismologia_client.IgnSismologiaFeed") as mock_feed:
|
||||||
mock_feed.return_value.update.return_value = (
|
mock_feed.return_value.update.return_value = (
|
||||||
"OK",
|
"OK",
|
||||||
[mock_entry_1, mock_entry_2, mock_entry_3],
|
[mock_entry_1, mock_entry_2, mock_entry_3],
|
||||||
|
|
Loading…
Reference in a new issue