Use freezegun in trafikverket_camera tests (#99067)

This commit is contained in:
Erik Montnemery 2023-08-28 09:23:32 +02:00 committed by GitHub
parent a3b526eef6
commit e97e9ae55a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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):