jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-24 01:08:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Context to service_calls fixture (#120923)

Browse Source
This commit is contained in:
epenet 2024-07-01 14:05:30 +02:00 committed by GitHub
parent 546d6b22f1
commit 61b2e4ca32
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/conftest.py
View File

@ -55,6 +55,7 @@ from homeassistant.config import YAML_CONFIG_FILE
from homeassistant.config_entries import ConfigEntries, ConfigEntry, ConfigEntryState
from homeassistant.const import HASSIO_USER_NAME
from homeassistant.core import (
Context,
CoreState,
HassJob,
HomeAssistant,
@ -1661,7 +1662,7 @@ def label_registry(hass: HomeAssistant) -> lr.LabelRegistry:
@pytest.fixture
def service_calls(hass: HomeAssistant) -> Generator[None, None, list[ServiceCall]]:
def service_calls(hass: HomeAssistant) -> Generator[list[ServiceCall]]:
"""Track all service calls."""
calls = []
@ -1672,15 +1673,23 @@ def service_calls(hass: HomeAssistant) -> Generator[None, None, list[ServiceCall
domain: str,
service: str,
service_data: dict[str, Any] | None = None,
**kwargs: Any,
blocking: bool = False,
context: Context | None = None,
target: dict[str, Any] | None = None,
return_response: bool = False,
) -> ServiceResponse:
calls.append(ServiceCall(domain, service, service_data))
calls.append(
ServiceCall(domain, service, service_data, context, return_response)
)
try:
return await _original_async_call(
domain,
service,
service_data,
**kwargs,
blocking,
context,
target,
return_response,
)
except ha.ServiceNotFound:
_LOGGER.debug("Ignoring unknown service call to %s.%s", domain, service)
@ -1697,7 +1706,7 @@ def snapshot(snapshot: SnapshotAssertion) -> SnapshotAssertion:
@pytest.fixture
def disable_block_async_io() -> Generator[Any, Any, None]:
def disable_block_async_io() -> Generator[None]:
"""Fixture to disable the loop protection from block_async_io."""
yield
calls = block_async_io._BLOCKED_CALLS.calls