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

Refactor integration startup time to show wall clock time (#113707)

Browse source 
* Refactor setup time tracking to exclude time waiting on other operations

We now exclude the import time and th time waiting on
base platforms to setup from the setup times

* tweak

* tweak

* tweak

* tweak

* adjust

* fixes

* fixes

* preen

* preen

* tweak

* tweak

* adjust

* tweak

* reduce

* do not count integrtion platforms against their parent integration

* handle legacy tts platforms

* stt as well

* one more wait

* use the same pattern in all the legacy

* fix tts and stt legacy

* fix

* fix

* reduce

* preen

* entity comp does not wait for platforms

* scene blocks as well

* fix test

* test fixes

* coverage

* coverage

* coverage

* fix test

* Update tests/test_setup.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* Update tests/test_setup.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* Update homeassistant/setup.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* strip

* strip WAIT_PLATFORM_INTEGRATION

* strip WAIT_PLATFORM_INTEGRATION

* strip WAIT_PLATFORM_INTEGRATION

* strip WAIT_PLATFORM_INTEGRATION

* remove complexity

* Apply suggestions from code review

* no longer works that way

* fixes

* fixes

* fixes

---------

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
J. Nick Koston 2024-03-18 15:45:34 -10:00 • committed by GitHub
parent 9be5f3531f
commit c615b52840
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 598 additions and 176 deletions
Download patch file Download diff file Expand all files Collapse all files

27
homeassistant/bootstrap.py
View file

@ -3,8 +3,8 @@
from __future__ import annotations
import asyncio
from collections import defaultdict
import contextlib
from datetime import timedelta
from functools import partial
from itertools import chain
import logging
@ -82,7 +82,7 @@ from .helpers.typing import ConfigType
from .setup import (
BASE_PLATFORMS,
DATA_SETUP_STARTED,
DATA_SETUP_TIME,
async_get_setup_timings,
async_notify_setup_error,
async_set_domains_to_be_loaded,
async_setup_component,
@ -597,7 +597,9 @@ class _WatchPendingSetups:
"""Periodic log and dispatch of setups that are pending."""
def __init__(
self, hass: core.HomeAssistant, setup_started: dict[str, float]
self,
hass: core.HomeAssistant,
setup_started: dict[tuple[str, str | None], float],
) -> None:
"""Initialize the WatchPendingSetups class."""
self._hass = hass
@ -612,10 +614,11 @@ class _WatchPendingSetups:
now = monotonic()
self._duration_count += SLOW_STARTUP_CHECK_INTERVAL
remaining_with_setup_started = {
domain: (now - start_time)
for domain, start_time in self._setup_started.items()
}
remaining_with_setup_started: defaultdict[str, float] = defaultdict(float)
for integration_group, start_time in self._setup_started.items():
domain, _ = integration_group
remaining_with_setup_started[domain] += now - start_time
if remaining_with_setup_started:
_LOGGER.debug("Integration remaining: %s", remaining_with_setup_started)
elif waiting_tasks := self._hass._active_tasks: # pylint: disable=protected-access
@ -629,7 +632,7 @@ class _WatchPendingSetups:
# once we take over LOG_SLOW_STARTUP_INTERVAL (60s) to start up
_LOGGER.warning(
"Waiting on integrations to complete setup: %s",
", ".join(self._setup_started),
self._setup_started,
)
_LOGGER.debug("Running timeout Zones: %s", self._hass.timeout.zones)
@ -838,10 +841,8 @@ async def _async_set_up_integrations(
hass: core.HomeAssistant, config: dict[str, Any]
) -> None:
"""Set up all the integrations."""
setup_started: dict[str, float] = {}
setup_started: dict[tuple[str, str | None], float] = {}
hass.data[DATA_SETUP_STARTED] = setup_started
setup_time: dict[str, timedelta] = hass.data.setdefault(DATA_SETUP_TIME, {})
watcher = _WatchPendingSetups(hass, setup_started)
watcher.async_start()
@ -934,7 +935,9 @@ async def _async_set_up_integrations(
watcher.async_stop()
if _LOGGER.isEnabledFor(logging.DEBUG):
setup_time = async_get_setup_timings(hass)
_LOGGER.debug(
"Integration setup times: %s",
dict(sorted(setup_time.items(), key=itemgetter(1))),
dict(sorted(setup_time.items(), key=itemgetter(1), reverse=True)),
)

8
homeassistant/components/device_tracker/legacy.py
View file

@ -50,6 +50,7 @@ from homeassistant.helpers.event import (
from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, GPSType, StateType
from homeassistant.setup import (
SetupPhases,
async_notify_setup_error,
async_prepare_setup_platform,
async_start_setup,
@ -307,7 +308,12 @@ class DeviceTrackerPlatform:
assert self.type == PLATFORM_TYPE_LEGACY
full_name = f"{self.name}.{DOMAIN}"
LOGGER.info("Setting up %s", full_name)
with async_start_setup(hass, [full_name]):
with async_start_setup(
hass,
integration=self.name,
group=str(id(self.config)),
phase=SetupPhases.PLATFORM_SETUP,
):
try:
scanner = None
setup: bool | None = None

8
homeassistant/components/http/__init__.py
View file

@ -43,7 +43,11 @@ from homeassistant.helpers.http import (
from homeassistant.helpers.network import NoURLAvailableError, get_url
from homeassistant.helpers.typing import ConfigType
from homeassistant.loader import bind_hass
from homeassistant.setup import async_start_setup, async_when_setup_or_start
from homeassistant.setup import (
SetupPhases,
async_start_setup,
async_when_setup_or_start,
)
from homeassistant.util import dt as dt_util, ssl as ssl_util
from homeassistant.util.async_ import create_eager_task
from homeassistant.util.json import json_loads
@ -218,7 +222,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
async def start_server(*_: Any) -> None:
"""Start the server."""
with async_start_setup(hass, ["http"]):
with async_start_setup(hass, integration="http", phase=SetupPhases.SETUP):
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, stop_server)
# We already checked it's not None.
assert conf is not None

13
homeassistant/components/notify/legacy.py
View file

@ -16,7 +16,11 @@ from homeassistant.helpers.service import async_set_service_schema
from homeassistant.helpers.template import Template
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.loader import async_get_integration, bind_hass
from homeassistant.setup import async_prepare_setup_platform, async_start_setup
from homeassistant.setup import (
SetupPhases,
async_prepare_setup_platform,
async_start_setup,
)
from homeassistant.util import slugify
from homeassistant.util.yaml import load_yaml_dict
@ -84,7 +88,12 @@ def async_setup_legacy(
full_name = f"{DOMAIN}.{integration_name}"
LOGGER.info("Setting up %s", full_name)
with async_start_setup(hass, [full_name]):
with async_start_setup(
hass,
integration=integration_name,
group=str(id(p_config)),
phase=SetupPhases.PLATFORM_SETUP,
):
notify_service: BaseNotificationService | None = None
try:
if hasattr(platform, "async_get_service"):

5
homeassistant/components/scene/__init__.py
View file

@ -66,7 +66,10 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
await component.async_setup(config)
# Ensure Home Assistant platform always loaded.
await component.async_setup_platform(HA_DOMAIN, {"platform": HA_DOMAIN, STATES: []})
hass.async_create_task(
component.async_setup_platform(HA_DOMAIN, {"platform": HA_DOMAIN, STATES: []}),
eager_start=True,
)
component.async_register_entity_service(
SERVICE_TURN_ON,
{ATTR_TRANSITION: vol.All(vol.Coerce(float), vol.Clamp(min=0, max=6553))},

2
homeassistant/components/stream/__init__.py
View file

@ -35,6 +35,7 @@ from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.typing import ConfigType
from homeassistant.setup import SetupPhases, async_pause_setup
from homeassistant.util.async_ import create_eager_task
from .const import (
@ -221,6 +222,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
logging.getLogger(logging_namespace).setLevel(logging.ERROR)
# This will load av so we run it in the executor
with async_pause_setup(hass, SetupPhases.WAIT_IMPORT_PACKAGES):
await hass.async_add_executor_job(set_pyav_logging, debug_enabled)
# Keep import here so that we can import stream integration without installing reqs

16
homeassistant/components/stt/legacy.py
View file

@ -11,7 +11,11 @@ from homeassistant.config import config_per_platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import discovery
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import async_prepare_setup_platform
from homeassistant.setup import (
SetupPhases,
async_prepare_setup_platform,
async_start_setup,
)
from .const import (
DATA_PROVIDERS,
@ -68,7 +72,15 @@ def async_setup_legacy(
return
try:
provider = await platform.async_get_engine(hass, p_config, discovery_info)
with async_start_setup(
hass,
integration=p_type,
group=str(id(p_config)),
phase=SetupPhases.PLATFORM_SETUP,
):
provider = await platform.async_get_engine(
hass, p_config, discovery_info
)
provider.name = p_type
provider.hass = hass

12
homeassistant/components/tts/legacy.py
View file

@ -31,7 +31,11 @@ from homeassistant.helpers import discovery
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.service import async_set_service_schema
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.setup import async_prepare_setup_platform
from homeassistant.setup import (
SetupPhases,
async_prepare_setup_platform,
async_start_setup,
)
from homeassistant.util.yaml import load_yaml_dict
from .const import (
@ -124,6 +128,12 @@ async def async_setup_legacy(
return
try:
with async_start_setup(
hass,
integration=p_type,
group=str(id(p_config)),
phase=SetupPhases.PLATFORM_SETUP,
):
if hasattr(platform, "async_get_engine"):
provider = await platform.async_get_engine(
hass, p_config, discovery_info

5
homeassistant/components/websocket_api/commands.py
View file

@ -55,7 +55,7 @@ from homeassistant.loader import (
async_get_integration_descriptions,
async_get_integrations,
)
from homeassistant.setup import DATA_SETUP_TIME, async_get_loaded_integrations
from homeassistant.setup import async_get_loaded_integrations, async_get_setup_timings
from homeassistant.util.json import format_unserializable_data
from . import const, decorators, messages
@ -539,12 +539,11 @@ def handle_integration_setup_info(
hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]
) -> None:
"""Handle integrations command."""
setup_time: dict[str, float] = hass.data[DATA_SETUP_TIME]
connection.send_result(
msg["id"],
[
{"domain": integration, "seconds": seconds}
for integration, seconds in setup_time.items()
for integration, seconds in async_get_setup_timings(hass).items()
],
)

23
homeassistant/config_entries.py
View file

@ -57,7 +57,14 @@ from .helpers.frame import report
from .helpers.json import json_bytes, json_fragment
from .helpers.typing import UNDEFINED, ConfigType, DiscoveryInfoType, UndefinedType
from .loader import async_suggest_report_issue
from .setup import DATA_SETUP_DONE, async_process_deps_reqs, async_setup_component
from .setup import (
DATA_SETUP_DONE,
SetupPhases,
async_pause_setup,
async_process_deps_reqs,
async_setup_component,
async_start_setup,
)
from .util import uuid as uuid_util
from .util.async_ import create_eager_task
from .util.decorator import Registry
@ -529,9 +536,16 @@ class ConfigEntry:
self._async_set_state(hass, ConfigEntryState.MIGRATION_ERROR, None)
return
setup_phase = SetupPhases.CONFIG_ENTRY_SETUP
else:
setup_phase = SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP
error_reason = None
try:
with async_start_setup(
hass, integration=self.domain, group=self.entry_id, phase=setup_phase
):
result = await component.async_setup_entry(hass, self)
if not isinstance(result, bool):
@ -1838,6 +1852,8 @@ class ConfigEntries:
) -> None:
"""Forward the setup of an entry to platforms."""
integration = await loader.async_get_integration(self.hass, entry.domain)
if not integration.platforms_are_loaded(platforms):
with async_pause_setup(self.hass, SetupPhases.WAIT_IMPORT_PLATFORMS):
await integration.async_get_platforms(platforms)
await asyncio.gather(
*(
@ -1860,7 +1876,10 @@ class ConfigEntries:
"""
# Setup Component if not set up yet
if domain not in self.hass.config.components:
result = await async_setup_component(self.hass, domain, self._hass_config)
with async_pause_setup(self.hass, SetupPhases.WAIT_BASE_PLATFORM_SETUP):
result = await async_setup_component(
self.hass, domain, self._hass_config
)
if not result:
return False

13
homeassistant/helpers/entity_platform.py
View file

@ -32,7 +32,7 @@ from homeassistant.core import (
)
from homeassistant.exceptions import HomeAssistantError, PlatformNotReady
from homeassistant.generated import languages
from homeassistant.setup import async_start_setup
from homeassistant.setup import SetupPhases, async_start_setup
from homeassistant.util.async_ import create_eager_task
from . import (
@ -284,6 +284,12 @@ class EntityPlatform:
discovery_info,
)
with async_start_setup(
hass,
integration=self.platform_name,
group=str(id(platform_config)),
phase=SetupPhases.PLATFORM_SETUP,
):
await self._async_setup_platform(async_create_setup_awaitable)
@callback
@ -341,7 +347,6 @@ class EntityPlatform:
self.platform_name,
SLOW_SETUP_WARNING,
)
with async_start_setup(hass, [full_name]):
try:
awaitable = async_create_setup_awaitable()
if asyncio.iscoroutine(awaitable):
@ -384,9 +389,7 @@ class EntityPlatform:
async def setup_again(*_args: Any) -> None:
"""Run setup again."""
self._async_cancel_retry_setup = None
await self._async_setup_platform(
async_create_setup_awaitable, tries
)
await self._async_setup_platform(async_create_setup_awaitable, tries)
if hass.state is CoreState.running:
self._async_cancel_retry_setup = async_call_later(

7
homeassistant/loader.py
View file

@ -1172,6 +1172,13 @@ class Integration:
raise self._missing_platforms_cache[full_name]
return None
def platforms_are_loaded(self, platform_names: Iterable[str]) -> bool:
"""Check if a platforms are loaded for an integration."""
return all(
f"{self.domain}.{platform_name}" in self._cache
for platform_name in platform_names
)
def get_platform_cached(self, platform_name: str) -> ModuleType | None:
"""Return a platform for an integration from cache."""
return self._cache.get(f"{self.domain}.{platform_name}") # type: ignore[return-value]

160
homeassistant/setup.py
View file

@ -3,8 +3,11 @@
from __future__ import annotations
import asyncio
from collections.abc import Awaitable, Callable, Generator, Iterable
from collections import defaultdict
from collections.abc import Awaitable, Callable, Generator, Mapping
import contextlib
import contextvars
from enum import StrEnum
import logging.handlers
import time
from timeit import default_timer as timer
@ -29,9 +32,13 @@ from .exceptions import DependencyError, HomeAssistantError
from .helpers import translation
from .helpers.issue_registry import IssueSeverity, async_create_issue
from .helpers.typing import ConfigType
from .util import ensure_unique_string
from .util.async_ import create_eager_task
current_setup_group: contextvars.ContextVar[
tuple[str, str | None] | None
] = contextvars.ContextVar("current_setup_group", default=None)
_LOGGER = logging.getLogger(__name__)
ATTR_COMPONENT: Final = "component"
@ -54,12 +61,12 @@ DATA_SETUP = "setup_tasks"
# is finished, regardless of if the setup was successful or not.
DATA_SETUP_DONE = "setup_done"
# DATA_SETUP_STARTED is a dict [str, float], indicating when an attempt
# DATA_SETUP_STARTED is a dict [tuple[str, str | None], float], indicating when an attempt
# to setup a component started.
DATA_SETUP_STARTED = "setup_started"
# DATA_SETUP_TIME is a dict [str, timedelta], indicating how time was spent
# setting up a component.
# DATA_SETUP_TIME is a defaultdict[str, defaultdict[str | None, defaultdict[SetupPhases, float]]]
# indicating how time was spent setting up a component and each group (config entry).
DATA_SETUP_TIME = "setup_time"
DATA_DEPS_REQS = "deps_reqs_processed"
@ -358,7 +365,7 @@ async def _async_setup_component( # noqa: C901
translation.async_load_integrations(hass, integration_set)
)
with async_start_setup(hass, integration_set):
with async_start_setup(hass, integration=domain, phase=SetupPhases.SETUP):
if hasattr(component, "PLATFORM_SCHEMA"):
# Entity components have their own warning
warn_task = None
@ -626,27 +633,134 @@ def async_get_loaded_integrations(hass: core.HomeAssistant) -> set[str]:
return integrations
class SetupPhases(StrEnum):
"""Constants for setup time measurements."""
SETUP = "setup"
"""Set up of a component in __init__.py."""
CONFIG_ENTRY_SETUP = "config_entry_setup"
"""Set up of a config entry in __init__.py."""
PLATFORM_SETUP = "platform_setup"
"""Set up of a platform integration.
ex async_setup_platform or setup_platform or
a legacy platform like device_tracker.legacy
"""
CONFIG_ENTRY_PLATFORM_SETUP = "config_entry_platform_setup"
"""Set up of a platform in a config entry after the config entry is setup.
This is only for platforms that are not awaited in async_setup_entry.
"""
WAIT_BASE_PLATFORM_SETUP = "wait_base_component"
"""Wait time for the base component to be setup."""
WAIT_IMPORT_PLATFORMS = "wait_import_platforms"
"""Wait time for the platforms to import."""
WAIT_IMPORT_PACKAGES = "wait_import_packages"
"""Wait time for the packages to import."""
@contextlib.contextmanager
def async_pause_setup(
hass: core.HomeAssistant, phase: SetupPhases
) -> Generator[None, None, None]:
"""Keep track of time we are blocked waiting for other operations.
We want to count the time we wait for importing and
setting up the base components so we can subtract it
from the total setup time.
"""
if not (running := current_setup_group.get()):
# This means we are likely in a late platform setup
# that is running in a task so we do not want
# to subtract out the time later as nothing is waiting
# for the code inside the context manager to finish.
yield
return
started = time.monotonic()
try:
yield
finally:
time_taken = time.monotonic() - started
integration, group = running
# Add negative time for the time we waited
_setup_times(hass)[integration][group][phase] = -time_taken
def _setup_times(
hass: core.HomeAssistant,
) -> defaultdict[str, defaultdict[str | None, defaultdict[SetupPhases, float]]]:
"""Return the setup timings default dict."""
if DATA_SETUP_TIME not in hass.data:
hass.data[DATA_SETUP_TIME] = defaultdict(
lambda: defaultdict(lambda: defaultdict(float))
)
return hass.data[DATA_SETUP_TIME] # type: ignore[no-any-return]
@contextlib.contextmanager
def async_start_setup(
hass: core.HomeAssistant, components: Iterable[str]
hass: core.HomeAssistant,
integration: str,
phase: SetupPhases,
group: str | None = None,
) -> Generator[None, None, None]:
"""Keep track of when setup starts and finishes."""
setup_started = hass.data.setdefault(DATA_SETUP_STARTED, {})
started = time.monotonic()
unique_components: dict[str, str] = {}
for domain in components:
unique = ensure_unique_string(domain, setup_started)
unique_components[unique] = domain
setup_started[unique] = started
"""Keep track of when setup starts and finishes.
:param hass: Home Assistant instance
:param integration: The integration that is being setup
:param phase: The phase of setup
:param group: The group (config entry/platform instance) that is being setup
A group is a group of setups that run in parallel.
"""
if hass.is_stopping or hass.state is core.CoreState.running:
# Don't track setup times when we are shutting down or already running
# as we present the timings as "Integration startup time", and we
# don't want to add all the setup retry times to that.
yield
return
setup_time: dict[str, float] = hass.data.setdefault(DATA_SETUP_TIME, {})
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(DATA_SETUP_STARTED, {})
current = (integration, group)
if current in setup_started:
# We are already inside another async_start_setup, this like means we
# are setting up a platform inside async_setup_entry so we should not
# record this as a new setup
yield
return
started = time.monotonic()
current_setup_group.set(current)
setup_started[current] = started
try:
yield
finally:
time_taken = time.monotonic() - started
for unique, domain in unique_components.items():
del setup_started[unique]
integration = domain.partition(".")[0]
if integration in setup_time:
setup_time[integration] += time_taken
else:
setup_time[integration] = time_taken
del setup_started[current]
_setup_times(hass)[integration][group][phase] = time_taken
@callback
def async_get_setup_timings(hass: core.HomeAssistant) -> dict[str, float]:
"""Return timing data for each integration."""
setup_time = _setup_times(hass)
domain_timings: dict[str, float] = {}
top_level_timings: Mapping[SetupPhases, float]
for domain, timings in setup_time.items():
top_level_timings = timings.get(None, {})
total_top_level = sum(top_level_timings.values())
# Groups (config entries/platform instance) are setup in parallel so we
# take the max of the group timings and add it to the top level
group_totals = {
group: sum(group_timings.values())
for group, group_timings in timings.items()
if group is not None
}
group_max = max(group_totals.values(), default=0)
domain_timings[domain] = total_top_level + group_max
return domain_timings

1
tests/components/config/test_scene.py
View file

@ -19,6 +19,7 @@ from tests.typing import ClientSessionGenerator
async def setup_scene(hass, scene_config):
"""Set up scene integration."""
assert await async_setup_component(hass, "scene", {"scene": scene_config})
await hass.async_block_till_done()
@pytest.mark.parametrize("scene_config", ({},))

2
tests/components/homeassistant/test_scene.py
View file

@ -18,6 +18,7 @@ from tests.common import async_capture_events, async_mock_service
async def test_reload_config_service(hass: HomeAssistant) -> None:
"""Test the reload config service."""
assert await async_setup_component(hass, "scene", {})
await hass.async_block_till_done()
test_reloaded_event = async_capture_events(hass, EVENT_SCENE_RELOADED)
@ -175,6 +176,7 @@ async def test_delete_service(
"scene",
{"scene": {"name": "hallo_2", "entities": {"light.kitchen": "on"}}},
)
await hass.async_block_till_done()
await hass.services.async_call(
"scene",

1
tests/components/scene/test_init.py
View file

@ -228,6 +228,7 @@ async def activate(hass, entity_id=ENTITY_MATCH_ALL):
async def test_services_registered(hass: HomeAssistant) -> None:
"""Test we register services with empty config."""
assert await async_setup_component(hass, "scene", {})
await hass.async_block_till_done()
assert hass.services.has_service("scene", "reload")
assert hass.services.has_service("scene", "turn_on")
assert hass.services.has_service("scene", "apply")

11
tests/components/websocket_api/test_commands.py
View file

@ -23,7 +23,7 @@ from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.loader import async_get_integration
from homeassistant.setup import DATA_SETUP_TIME, async_setup_component
from homeassistant.setup import async_setup_component
from homeassistant.util.json import json_loads
from tests.common import (
@ -2491,13 +2491,16 @@ async def test_integration_setup_info(
hass_admin_user: MockUser,
) -> None:
"""Test subscribe/unsubscribe bootstrap_integrations."""
hass.data[DATA_SETUP_TIME] = {
with patch(
"homeassistant.components.websocket_api.commands.async_get_setup_timings",
return_value={
"august": 12.5,
"isy994": 12.8,
}
},
):
await websocket_client.send_json({"id": 7, "type": "integration/setup_info"})
msg = await websocket_client.receive_json()
assert msg["id"] == 7
assert msg["type"] == const.TYPE_RESULT
assert msg["success"]

5
tests/test_bootstrap.py
View file

@ -14,7 +14,7 @@ from homeassistant import bootstrap, loader, runner
import homeassistant.config as config_util
from homeassistant.config_entries import HANDLERS, ConfigEntry
from homeassistant.const import SIGNAL_BOOTSTRAP_INTEGRATIONS
from homeassistant.core import HomeAssistant, async_get_hass, callback
from homeassistant.core import CoreState, HomeAssistant, async_get_hass, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.typing import ConfigType
@ -869,6 +869,9 @@ async def test_empty_integrations_list_is_only_sent_at_the_end_of_bootstrap(
hass: HomeAssistant,
) -> None:
"""Test empty integrations list is only sent at the end of bootstrap."""
# setup times only tracked when not running
hass.set_state(CoreState.not_running)
order = []
def gen_domain_setup(domain):

3
tests/test_loader.py
View file

@ -1520,6 +1520,9 @@ async def test_platforms_exists(
assert integration.platforms_exists(["group"]) == ["group"]
assert integration.platforms_are_loaded(["group"]) is True
assert integration.platforms_are_loaded(["other"]) is False
async def test_async_get_platforms_loads_loop_if_already_in_sys_modules(
hass: HomeAssistant,

254
tests/test_setup.py
View file

@ -2,14 +2,14 @@
import asyncio
import threading
from unittest.mock import AsyncMock, Mock, patch
from unittest.mock import ANY, AsyncMock, Mock, patch
import pytest
import voluptuous as vol
from homeassistant import config_entries, loader, setup
from homeassistant.const import EVENT_COMPONENT_LOADED, EVENT_HOMEASSISTANT_START
from homeassistant.core import HomeAssistant, callback
from homeassistant.core import CoreState, HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import discovery, translation
from homeassistant.helpers.config_validation import (
@ -727,26 +727,244 @@ async def test_integration_only_setup_entry(hass: HomeAssistant) -> None:
assert await setup.async_setup_component(hass, "test_integration_only_entry", {})
async def test_async_start_setup(hass: HomeAssistant) -> None:
"""Test setup started context manager keeps track of setup times."""
with setup.async_start_setup(hass, ["august"]):
assert isinstance(hass.data[setup.DATA_SETUP_STARTED]["august"], float)
with setup.async_start_setup(hass, ["august"]):
assert isinstance(hass.data[setup.DATA_SETUP_STARTED]["august_2"], float)
async def test_async_start_setup_running(hass: HomeAssistant) -> None:
"""Test setup started context manager does nothing when running."""
assert hass.state is CoreState.running
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(setup.DATA_SETUP_STARTED, {})
assert "august" not in hass.data[setup.DATA_SETUP_STARTED]
assert isinstance(hass.data[setup.DATA_SETUP_TIME]["august"], float)
assert "august_2" not in hass.data[setup.DATA_SETUP_TIME]
with setup.async_start_setup(
hass, integration="august", phase=setup.SetupPhases.SETUP
):
assert not setup_started
async def test_async_start_setup_platforms(hass: HomeAssistant) -> None:
"""Test setup started context manager keeps track of setup times for platforms."""
with setup.async_start_setup(hass, ["august.sensor"]):
assert isinstance(hass.data[setup.DATA_SETUP_STARTED]["august.sensor"], float)
async def test_async_start_setup_config_entry(hass: HomeAssistant) -> None:
"""Test setup started keeps track of setup times with a config entry."""
hass.set_state(CoreState.not_running)
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(setup.DATA_SETUP_STARTED, {})
setup_time = setup._setup_times(hass)
assert "august" not in hass.data[setup.DATA_SETUP_STARTED]
assert isinstance(hass.data[setup.DATA_SETUP_TIME]["august"], float)
assert "sensor" not in hass.data[setup.DATA_SETUP_TIME]
with setup.async_start_setup(
hass, integration="august", phase=setup.SetupPhases.SETUP
):
assert isinstance(setup_started[("august", None)], float)
with setup.async_start_setup(
hass,
integration="august",
group="entry_id",
phase=setup.SetupPhases.CONFIG_ENTRY_SETUP,
):
assert isinstance(setup_started[("august", "entry_id")], float)
with setup.async_start_setup(
hass,
integration="august",
group="entry_id",
phase=setup.SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP,
):
assert isinstance(setup_started[("august", "entry_id")], float)
# CONFIG_ENTRY_PLATFORM_SETUP inside of CONFIG_ENTRY_SETUP should not be tracked
assert setup_time["august"] == {
None: {setup.SetupPhases.SETUP: ANY},
"entry_id": {setup.SetupPhases.CONFIG_ENTRY_SETUP: ANY},
}
with setup.async_start_setup(
hass,
integration="august",
group="entry_id",
phase=setup.SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP,
):
assert isinstance(setup_started[("august", "entry_id")], float)
# Platforms outside of CONFIG_ENTRY_SETUP should be tracked
# This simulates a late platform forward
assert setup_time["august"] == {
None: {setup.SetupPhases.SETUP: ANY},
"entry_id": {
setup.SetupPhases.CONFIG_ENTRY_SETUP: ANY,
setup.SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP: ANY,
},
}
with setup.async_start_setup(
hass,
integration="august",
group="entry_id2",
phase=setup.SetupPhases.CONFIG_ENTRY_SETUP,
):
assert isinstance(setup_started[("august", "entry_id2")], float)
# We wrap places where we wait for other components
# or the import of a module with async_freeze_setup
# so we can subtract the time waited from the total setup time
with setup.async_pause_setup(hass, setup.SetupPhases.WAIT_BASE_PLATFORM_SETUP):
await asyncio.sleep(0)
# Wait time should be added if freeze_setup is used
assert setup_time["august"] == {
None: {setup.SetupPhases.SETUP: ANY},
"entry_id": {
setup.SetupPhases.CONFIG_ENTRY_SETUP: ANY,
setup.SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP: ANY,
},
"entry_id2": {
setup.SetupPhases.CONFIG_ENTRY_SETUP: ANY,
setup.SetupPhases.WAIT_BASE_PLATFORM_SETUP: ANY,
},
}
async def test_async_start_setup_top_level_yaml(hass: HomeAssistant) -> None:
"""Test setup started context manager keeps track of setup times with modern yaml."""
hass.set_state(CoreState.not_running)
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(setup.DATA_SETUP_STARTED, {})
setup_time = setup._setup_times(hass)
with setup.async_start_setup(
hass, integration="command_line", phase=setup.SetupPhases.SETUP
):
assert isinstance(setup_started[("command_line", None)], float)
assert setup_time["command_line"] == {
None: {setup.SetupPhases.SETUP: ANY},
}
async def test_async_start_setup_platform_integration(hass: HomeAssistant) -> None:
"""Test setup started keeps track of setup times a platform integration."""
hass.set_state(CoreState.not_running)
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(setup.DATA_SETUP_STARTED, {})
setup_time = setup._setup_times(hass)
with setup.async_start_setup(
hass, integration="sensor", phase=setup.SetupPhases.SETUP
):
assert isinstance(setup_started[("sensor", None)], float)
# Platform integration setups happen in another task
with setup.async_start_setup(
hass,
integration="filter",
group="123456",
phase=setup.SetupPhases.PLATFORM_SETUP,
):
assert isinstance(setup_started[("filter", "123456")], float)
assert setup_time["sensor"] == {
None: {
setup.SetupPhases.SETUP: ANY,
},
}
assert setup_time["filter"] == {
"123456": {
setup.SetupPhases.PLATFORM_SETUP: ANY,
},
}
async def test_async_start_setup_legacy_platform_integration(
hass: HomeAssistant,
) -> None:
"""Test setup started keeps track of setup times for a legacy platform integration."""
hass.set_state(CoreState.not_running)
setup_started: dict[tuple[str, str | None], float]
setup_started = hass.data.setdefault(setup.DATA_SETUP_STARTED, {})
setup_time = setup._setup_times(hass)
with setup.async_start_setup(
hass, integration="notify", phase=setup.SetupPhases.SETUP
):
assert isinstance(setup_started[("notify", None)], float)
with setup.async_start_setup(
hass,
integration="legacy_notify_integration",
group="123456",
phase=setup.SetupPhases.PLATFORM_SETUP,
):
assert isinstance(setup_started[("legacy_notify_integration", "123456")], float)
assert setup_time["notify"] == {
None: {
setup.SetupPhases.SETUP: ANY,
},
}
assert setup_time["legacy_notify_integration"] == {
"123456": {
setup.SetupPhases.PLATFORM_SETUP: ANY,
},
}
async def test_async_start_setup_simple_integration_end_to_end(
hass: HomeAssistant,
) -> None:
"""Test end to end timings for a simple integration with no platforms."""
hass.set_state(CoreState.not_running)
mock_integration(
hass,
MockModule(
"test_integration_no_platforms",
setup=False,
async_setup_entry=AsyncMock(return_value=True),
),
)
assert await setup.async_setup_component(hass, "test_integration_no_platforms", {})
await hass.async_block_till_done()
assert setup.async_get_setup_timings(hass) == {
"test_integration_no_platforms": ANY,
}
async def test_async_get_setup_timings(hass) -> None:
"""Test we can get the setup timings from the setup time data."""
setup_time = setup._setup_times(hass)
# Mock setup time data
setup_time.update(
{
"august": {
None: {setup.SetupPhases.SETUP: 1},
"entry_id": {
setup.SetupPhases.CONFIG_ENTRY_SETUP: 1,
setup.SetupPhases.CONFIG_ENTRY_PLATFORM_SETUP: 4,
},
"entry_id2": {
setup.SetupPhases.CONFIG_ENTRY_SETUP: 7,
setup.SetupPhases.WAIT_BASE_PLATFORM_SETUP: -5,
},
},
"notify": {
None: {
setup.SetupPhases.SETUP: 2,
},
},
"legacy_notify_integration": {
"123456": {
setup.SetupPhases.PLATFORM_SETUP: 3,
},
},
"sensor": {
None: {
setup.SetupPhases.SETUP: 1,
},
},
"filter": {
"123456": {
setup.SetupPhases.PLATFORM_SETUP: 2,
},
},
}
)
assert setup.async_get_setup_timings(hass) == {
"august": 6,
"notify": 2,
"legacy_notify_integration": 3,
"sensor": 1,
"filter": 2,
}
async def test_setup_config_entry_from_yaml(