evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use freezegun in image_upload test (#106045)

Browse source
This commit is contained in:
Jan-Philipp Benecke 2023-12-19 12:26:40 +01:00 • committed by GitHub
parent f84e43bc0a
commit 7d74c89606
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/components/image_upload/test_init.py
View file

@ -4,6 +4,7 @@ import tempfile
from unittest.mock import patch from unittest.mock import patch
from aiohttp import ClientSession, ClientWebSocketResponse from aiohttp import ClientSession, ClientWebSocketResponse
from freezegun.api import FrozenDateTimeFactory
from homeassistant.components.websocket_api import const as ws_const from homeassistant.components.websocket_api import const as ws_const
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -17,15 +18,17 @@ from tests.typing import ClientSessionGenerator, WebSocketGenerator
async def test_upload_image( async def test_upload_image(
hass: HomeAssistant, hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
hass_client: ClientSessionGenerator, hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator, hass_ws_client: WebSocketGenerator,
) -> None: ) -> None:
"""Test we can upload an image.""" """Test we can upload an image."""
now = dt_util.utcnow() now = dt_util.utcnow()
freezer.move_to(now)
with tempfile.TemporaryDirectory() as tempdir, patch.object( with tempfile.TemporaryDirectory() as tempdir, patch.object(
hass.config, "path", return_value=tempdir hass.config, "path", return_value=tempdir
), patch("homeassistant.util.dt.utcnow", return_value=now): ):
assert await async_setup_component(hass, "image_upload", {}) assert await async_setup_component(hass, "image_upload", {})
ws_client: ClientWebSocketResponse = await hass_ws_client() ws_client: ClientWebSocketResponse = await hass_ws_client()
client: ClientSession = await hass_client() client: ClientSession = await hass_client()