From 7d74c89606da070a998d6abe2f9be1c90f3a55fb Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke <jan-philipp@bnck.me> Date: Tue, 19 Dec 2023 12:26:40 +0100 Subject: [PATCH] Use freezegun in image_upload test (#106045) --- tests/components/image_upload/test_init.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/components/image_upload/test_init.py b/tests/components/image_upload/test_init.py index 486f98e92c2..9f842d25b64 100644 --- a/tests/components/image_upload/test_init.py +++ b/tests/components/image_upload/test_init.py @@ -4,6 +4,7 @@ import tempfile from unittest.mock import patch from aiohttp import ClientSession, ClientWebSocketResponse +from freezegun.api import FrozenDateTimeFactory from homeassistant.components.websocket_api import const as ws_const from homeassistant.core import HomeAssistant @@ -17,15 +18,17 @@ from tests.typing import ClientSessionGenerator, WebSocketGenerator async def test_upload_image( hass: HomeAssistant, + freezer: FrozenDateTimeFactory, hass_client: ClientSessionGenerator, hass_ws_client: WebSocketGenerator, ) -> None: """Test we can upload an image.""" now = dt_util.utcnow() + freezer.move_to(now) with tempfile.TemporaryDirectory() as tempdir, patch.object( hass.config, "path", return_value=tempdir - ), patch("homeassistant.util.dt.utcnow", return_value=now): + ): assert await async_setup_component(hass, "image_upload", {}) ws_client: ClientWebSocketResponse = await hass_ws_client() client: ClientSession = await hass_client() -- GitLab