From e07768412a0343ba73bfc27190777927a27eb73a Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Fri, 16 Aug 2024 17:16:56 +0200
Subject: [PATCH] Update ffmpeg tests to async (#124058)
---
tests/components/ffmpeg/test_init.py | 30 ++++++++++++----------------
1 file changed, 13 insertions(+), 17 deletions(-)
diff --git a/tests/components/ffmpeg/test_init.py b/tests/components/ffmpeg/test_init.py
index ceefed8d62b..e9a781327e0 100644
--- a/tests/components/ffmpeg/test_init.py
+++ b/tests/components/ffmpeg/test_init.py
@@ -16,9 +16,9 @@ from homeassistant.const import (
EVENT_HOMEASSISTANT_STOP,
)
from homeassistant.core import HomeAssistant, callback
-from homeassistant.setup import async_setup_component, setup_component
+from homeassistant.setup import async_setup_component
-from tests.common import assert_setup_component, get_test_home_assistant
+from tests.common import assert_setup_component
@callback
@@ -82,26 +82,22 @@ class MockFFmpegDev(ffmpeg.FFmpegBase):
self.called_entities = entity_ids
-def test_setup_component() -> None:
+async def test_setup_component(hass: HomeAssistant) -> None:
"""Set up ffmpeg component."""
- with get_test_home_assistant() as hass:
- with assert_setup_component(1):
- setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
+ with assert_setup_component(1):
+ await async_setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
- assert hass.data[ffmpeg.DATA_FFMPEG].binary == "ffmpeg"
- hass.stop()
+ assert hass.data[ffmpeg.DATA_FFMPEG].binary == "ffmpeg"
-def test_setup_component_test_service() -> None:
+async def test_setup_component_test_service(hass: HomeAssistant) -> None:
"""Set up ffmpeg component test services."""
- with get_test_home_assistant() as hass:
- with assert_setup_component(1):
- setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
-
- assert hass.services.has_service(ffmpeg.DOMAIN, "start")
- assert hass.services.has_service(ffmpeg.DOMAIN, "stop")
- assert hass.services.has_service(ffmpeg.DOMAIN, "restart")
- hass.stop()
+ with assert_setup_component(1):
+ await async_setup_component(hass, ffmpeg.DOMAIN, {ffmpeg.DOMAIN: {}})
+
+ assert hass.services.has_service(ffmpeg.DOMAIN, "start")
+ assert hass.services.has_service(ffmpeg.DOMAIN, "stop")
+ assert hass.services.has_service(ffmpeg.DOMAIN, "restart")
async def test_setup_component_test_register(hass: HomeAssistant) -> None:
--
GitLab