diff --git a/tests/components/cert_expiry/test_init.py b/tests/components/cert_expiry/test_init.py index 5ba63ad1af1..9b188cf41d7 100644 --- a/tests/components/cert_expiry/test_init.py +++ b/tests/components/cert_expiry/test_init.py @@ -2,7 +2,7 @@ from unittest.mock import patch -from freezegun import freeze_time +import pytest from homeassistant.components.cert_expiry.const import DOMAIN from homeassistant.config_entries import ConfigEntryState @@ -44,7 +44,7 @@ async def test_update_unique_id(hass: HomeAssistant) -> None: assert entry.unique_id == f"{HOST}:{PORT}" -@freeze_time(static_datetime()) +@pytest.mark.freeze_time(static_datetime()) async def test_unload_config_entry(hass: HomeAssistant) -> None: """Test unloading a config entry.""" assert hass.state is CoreState.running diff --git a/tests/components/cert_expiry/test_sensor.py b/tests/components/cert_expiry/test_sensor.py index d68e973e1fa..68e2539f613 100644 --- a/tests/components/cert_expiry/test_sensor.py +++ b/tests/components/cert_expiry/test_sensor.py @@ -6,6 +6,7 @@ import ssl from unittest.mock import patch from freezegun import freeze_time +import pytest from homeassistant.components.cert_expiry.const import DOMAIN from homeassistant.const import CONF_HOST, CONF_PORT, STATE_UNAVAILABLE, STATE_UNKNOWN @@ -18,7 +19,7 @@ from .helpers import future_timestamp, static_datetime from tests.common import MockConfigEntry, async_fire_time_changed -@freeze_time(static_datetime()) +@pytest.mark.freeze_time(static_datetime()) async def test_async_setup_entry(hass: HomeAssistant) -> None: """Test async_setup_entry.""" assert hass.state is CoreState.running