jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ensure MockEntityPlatform shuts down after tests (#89849)

Browse Source
This commit is contained in:
epenet 2023-03-17 10:22:43 +01:00 committed by GitHub
parent aa72b48725
commit abd91dd934
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/common.py
View File

@ -37,6 +37,7 @@ from homeassistant.config import async_process_component_config
from homeassistant.const import ( from homeassistant.const import (
DEVICE_DEFAULT_NAME, DEVICE_DEFAULT_NAME,
EVENT_HOMEASSISTANT_CLOSE, EVENT_HOMEASSISTANT_CLOSE,
EVENT_HOMEASSISTANT_STOP,
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED,
STATE_OFF, STATE_OFF,
STATE_ON, STATE_ON,
@ -757,7 +758,7 @@ class MockEntityPlatform(entity_platform.EntityPlatform):
def __init__( def __init__(
self, self,
hass, hass: HomeAssistant,
logger=None, logger=None,
domain="test_domain", domain="test_domain",
platform_name="test_platform", platform_name="test_platform",
@ -783,6 +784,11 @@ class MockEntityPlatform(entity_platform.EntityPlatform):
entity_namespace=entity_namespace, 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): class MockToggleEntity(entity.ToggleEntity):
"""Provide a mock toggle device.""" """Provide a mock toggle device."""