Fix 'all' entity_id in service call extraction (#20281)

pull/20285/head
Anders Melchiorsen 2019-01-21 01:33:11 +01:00 committed by Paulus Schoutsen
parent 08a57959b9
commit 1218127d83
2 changed files with 4 additions and 3 deletions

View File

@ -7,7 +7,8 @@ import logging
from homeassistant import config as conf_util
from homeassistant.setup import async_prepare_setup_platform
from homeassistant.const import (
ATTR_ENTITY_ID, CONF_SCAN_INTERVAL, CONF_ENTITY_NAMESPACE, MATCH_ALL)
ATTR_ENTITY_ID, CONF_SCAN_INTERVAL, CONF_ENTITY_NAMESPACE,
ENTITY_MATCH_ALL)
from homeassistant.core import callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_per_platform, discovery
@ -163,7 +164,7 @@ class EntityComponent:
"""
data_ent_id = service.data.get(ATTR_ENTITY_ID)
if data_ent_id in (None, MATCH_ALL):
if data_ent_id in (None, ENTITY_MATCH_ALL):
if data_ent_id is None:
self.logger.warning(
'Not passing an entity ID to a service to target all '

View File

@ -479,7 +479,7 @@ async def test_extract_all_use_match_all(hass, caplog):
MockEntity(name='test_2'),
])
call = ha.ServiceCall('test', 'service', {'entity_id': '*'})
call = ha.ServiceCall('test', 'service', {'entity_id': 'all'})
assert ['test_domain.test_1', 'test_domain.test_2'] == \
sorted(ent.entity_id for ent in