diff --git a/tests/components/version/common.py b/tests/components/version/common.py index 3e3ae6c3970..c4759604a44 100644 --- a/tests/components/version/common.py +++ b/tests/components/version/common.py @@ -4,6 +4,8 @@ from __future__ import annotations from typing import Any, Final from unittest.mock import patch +from freezegun.api import FrozenDateTimeFactory + from homeassistant import config_entries from homeassistant.components.version.const import ( DEFAULT_CONFIGURATION, @@ -14,7 +16,6 @@ from homeassistant.components.version.const import ( ) from homeassistant.const import CONF_NAME from homeassistant.core import HomeAssistant -from homeassistant.util import dt as dt_util from tests.common import MockConfigEntry, async_fire_time_changed @@ -37,6 +38,7 @@ TEST_DEFAULT_IMPORT_CONFIG: Final = { async def mock_get_version_update( hass: HomeAssistant, + freezer: FrozenDateTimeFactory, version: str = MOCK_VERSION, data: dict[str, Any] = MOCK_VERSION_DATA, side_effect: Exception = None, @@ -47,9 +49,8 @@ async def mock_get_version_update( return_value=(version, data), side_effect=side_effect, ): - async_fire_time_changed( - hass, dt_util.utcnow() + UPDATE_COORDINATOR_UPDATE_INTERVAL - ) + freezer.tick(UPDATE_COORDINATOR_UPDATE_INTERVAL) + async_fire_time_changed(hass) await hass.async_block_till_done() diff --git a/tests/components/version/test_sensor.py b/tests/components/version/test_sensor.py index 1c7f9040b22..0a3e89494f1 100644 --- a/tests/components/version/test_sensor.py +++ b/tests/components/version/test_sensor.py @@ -1,6 +1,7 @@ """The test for the version sensor platform.""" from __future__ import annotations +from freezegun.api import FrozenDateTimeFactory from pyhaversion.exceptions import HaVersionException import pytest @@ -19,15 +20,19 @@ async def test_version_sensor(hass: HomeAssistant) -> None: assert "channel" not in state.attributes -async def test_update(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -> None: +async def test_update( + hass: HomeAssistant, + caplog: pytest.LogCaptureFixture, + freezer: FrozenDateTimeFactory, +) -> None: """Test updates.""" await setup_version_integration(hass) assert hass.states.get("sensor.local_installation").state == MOCK_VERSION - await mock_get_version_update(hass, version="1970.1.1") + await mock_get_version_update(hass, freezer, version="1970.1.1") assert hass.states.get("sensor.local_installation").state == "1970.1.1" assert "Error fetching version data" not in caplog.text - await mock_get_version_update(hass, side_effect=HaVersionException) + await mock_get_version_update(hass, freezer, side_effect=HaVersionException) assert hass.states.get("sensor.local_installation").state == "unavailable" assert "Error fetching version data" in caplog.text