jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 21:57:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use freezegun in trafikverket_camera tests (#99067)

Browse Source
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
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/components/trafikverket_camera/test_camera.py
View File

@ -4,6 +4,7 @@ from __future__ import annotations
from datetime import timedelta from datetime import timedelta
from unittest.mock import patch from unittest.mock import patch
from freezegun.api import FrozenDateTimeFactory
import pytest import pytest
from pytrafikverket.trafikverket_camera import CameraInfo 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.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.util import dt as dt_util
from tests.common import async_fire_time_changed from tests.common import async_fire_time_changed
from tests.test_util.aiohttp import AiohttpClientMocker from tests.test_util.aiohttp import AiohttpClientMocker
@ -20,6 +20,7 @@ from tests.test_util.aiohttp import AiohttpClientMocker
async def test_camera( async def test_camera(
hass: HomeAssistant, hass: HomeAssistant,
load_int: ConfigEntry, load_int: ConfigEntry,
freezer: FrozenDateTimeFactory,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
aioclient_mock: AiohttpClientMocker, aioclient_mock: AiohttpClientMocker,
get_camera: CameraInfo, get_camera: CameraInfo,
@ -39,10 +40,8 @@ async def test_camera(
"https://www.testurl.com/test_photo.jpg?type=fullsize", "https://www.testurl.com/test_photo.jpg?type=fullsize",
content=b"9876543210", content=b"9876543210",
) )
async_fire_time_changed( freezer.tick(timedelta(minutes=6))
hass, async_fire_time_changed(hass)
dt_util.utcnow() + timedelta(minutes=6),
)
await hass.async_block_till_done() await hass.async_block_till_done()
state1 = hass.states.get("camera.test_location") state1 = hass.states.get("camera.test_location")
@ -65,10 +64,8 @@ async def test_camera(
"homeassistant.components.trafikverket_camera.coordinator.TrafikverketCamera.async_get_camera", "homeassistant.components.trafikverket_camera.coordinator.TrafikverketCamera.async_get_camera",
return_value=get_camera, return_value=get_camera,
): ):
async_fire_time_changed( freezer.tick(timedelta(minutes=6))
hass, async_fire_time_changed(hass)
dt_util.utcnow() + timedelta(minutes=6),
)
await hass.async_block_till_done() await hass.async_block_till_done()
with pytest.raises(HomeAssistantError): with pytest.raises(HomeAssistantError):