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

Remove loop shutdown indicator when done with test hass (#122432)

Browse source
This commit is contained in:
Erik Montnemery 2024-07-23 00:56:06 +02:00 • committed by GitHub
parent 96de0a4c94
commit f4125eaf4c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/common.py
View file

@ -13,7 +13,7 @@ from collections.abc import (
Mapping, Mapping,
Sequence, Sequence,
) )
from contextlib import asynccontextmanager, contextmanager from contextlib import asynccontextmanager, contextmanager, suppress
from datetime import UTC, datetime, timedelta from datetime import UTC, datetime, timedelta
from enum import Enum from enum import Enum
import functools as ft import functools as ft
@ -91,7 +91,10 @@ from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.json import JSONEncoder, _orjson_default_encoder, json_dumps from homeassistant.helpers.json import JSONEncoder, _orjson_default_encoder, json_dumps
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.async_ import run_callback_threadsafe from homeassistant.util.async_ import (
_SHUTDOWN_RUN_CALLBACK_THREADSAFE,
run_callback_threadsafe,
)
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from homeassistant.util.event_type import EventType from homeassistant.util.event_type import EventType
from homeassistant.util.json import ( from homeassistant.util.json import (
@ -376,6 +379,9 @@ async def async_test_home_assistant(
finally: finally:
# Restore timezone, it is set when creating the hass object # Restore timezone, it is set when creating the hass object
dt_util.set_default_time_zone(orig_tz) dt_util.set_default_time_zone(orig_tz)
# Remove loop shutdown indicator to not interfere with additional hass objects
with suppress(AttributeError):
delattr(hass.loop, _SHUTDOWN_RUN_CALLBACK_THREADSAFE)
def async_mock_service( def async_mock_service(