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

Add service entity context (#71558)

Browse source 
Co-authored-by: Shay Levy <levyshay1@gmail.com>
This commit is contained in:
Paulus Schoutsen 2022-05-09 14:45:53 -07:00 • committed by GitHub
parent d8336a5216
commit 64636a4310
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/helpers/test_service.py
View file

@ -19,12 +19,12 @@ from homeassistant.const import (
STATE_ON,
)
from homeassistant.helpers import (
config_validation as cv,
device_registry as dev_reg,
entity_registry as ent_reg,
service,
template,
)
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import EntityCategory
from homeassistant.setup import async_setup_component
@ -1205,3 +1205,17 @@ async def test_async_extract_config_entry_ids(hass):
)
assert await service.async_extract_config_entry_ids(hass, call) == {"abc"}
async def test_current_entity_context(hass, mock_entities):
"""Test we set the current entity context var."""
async def mock_service(entity, call):
assert entity.entity_id == service.async_get_current_entity()
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_service,
ha.ServiceCall("test_domain", "test_service", {"entity_id": "light.kitchen"}),
)