From e97e9ae55ac6eb5856796a56331819439c628237 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Mon, 28 Aug 2023 09:23:32 +0200 Subject: [PATCH] Use freezegun in trafikverket_camera tests (#99067) --- .../components/trafikverket_camera/test_camera.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/tests/components/trafikverket_camera/test_camera.py b/tests/components/trafikverket_camera/test_camera.py index 57451ae93a9..b3df7cfcdcb 100644 --- a/tests/components/trafikverket_camera/test_camera.py +++ b/tests/components/trafikverket_camera/test_camera.py @@ -4,6 +4,7 @@ from __future__ import annotations from datetime import timedelta from unittest.mock import patch +from freezegun.api import FrozenDateTimeFactory import pytest from pytrafikverket.trafikverket_camera import CameraInfo @@ -11,7 +12,6 @@ from homeassistant.components.camera import async_get_image from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.exceptions import HomeAssistantError -from homeassistant.util import dt as dt_util from tests.common import async_fire_time_changed from tests.test_util.aiohttp import AiohttpClientMocker @@ -20,6 +20,7 @@ from tests.test_util.aiohttp import AiohttpClientMocker async def test_camera( hass: HomeAssistant, load_int: ConfigEntry, + freezer: FrozenDateTimeFactory, monkeypatch: pytest.MonkeyPatch, aioclient_mock: AiohttpClientMocker, get_camera: CameraInfo, @@ -39,10 +40,8 @@ async def test_camera( "https://www.testurl.com/test_photo.jpg?type=fullsize", content=b"9876543210", ) - async_fire_time_changed( - hass, - dt_util.utcnow() + timedelta(minutes=6), - ) + freezer.tick(timedelta(minutes=6)) + async_fire_time_changed(hass) await hass.async_block_till_done() state1 = hass.states.get("camera.test_location") @@ -65,10 +64,8 @@ async def test_camera( "homeassistant.components.trafikverket_camera.coordinator.TrafikverketCamera.async_get_camera", return_value=get_camera, ): - async_fire_time_changed( - hass, - dt_util.utcnow() + timedelta(minutes=6), - ) + freezer.tick(timedelta(minutes=6)) + async_fire_time_changed(hass) await hass.async_block_till_done() with pytest.raises(HomeAssistantError):