From abd91dd9345cb3e99d857ced485444901deb09cb Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 17 Mar 2023 10:22:43 +0100 Subject: [PATCH] Ensure MockEntityPlatform shuts down after tests (#89849) --- tests/common.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/common.py b/tests/common.py index 8b9a9c24017..569813d221c 100644 --- a/tests/common.py +++ b/tests/common.py @@ -37,6 +37,7 @@ from homeassistant.config import async_process_component_config from homeassistant.const import ( DEVICE_DEFAULT_NAME, EVENT_HOMEASSISTANT_CLOSE, + EVENT_HOMEASSISTANT_STOP, EVENT_STATE_CHANGED, STATE_OFF, STATE_ON, @@ -757,7 +758,7 @@ class MockEntityPlatform(entity_platform.EntityPlatform): def __init__( self, - hass, + hass: HomeAssistant, logger=None, domain="test_domain", platform_name="test_platform", @@ -783,6 +784,11 @@ class MockEntityPlatform(entity_platform.EntityPlatform): entity_namespace=entity_namespace, ) + async def _async_on_stop(_: Event) -> None: + await self.async_shutdown() + + hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, _async_on_stop) + class MockToggleEntity(entity.ToggleEntity): """Provide a mock toggle device."""