jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-07 18:09:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Index entities by domain for entity services (#106759)

Browse Source
This commit is contained in:
J. Nick Koston
2024-01-02 04:28:58 -10:00
committed by GitHub
parent bf0d891f68
commit 09b65f14b9
6 changed files with 63 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/helpers/test_service.py
View File

@@ -802,7 +802,7 @@ async def test_call_with_required_features(hass: HomeAssistant, mock_entities) -
test_service_mock = AsyncMock(return_value=None)
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
test_service_mock,
ServiceCall("test_domain", "test_service", {"entity_id": "all"}),
required_features=[SUPPORT_A],
@@ -821,7 +821,7 @@ async def test_call_with_required_features(hass: HomeAssistant, mock_entities) -
with pytest.raises(exceptions.HomeAssistantError):
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
test_service_mock,
ServiceCall(
"test_domain", "test_service", {"entity_id": "light.living_room"}
@@ -838,7 +838,7 @@ async def test_call_with_both_required_features(
test_service_mock = AsyncMock(return_value=None)
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
test_service_mock,
ServiceCall("test_domain", "test_service", {"entity_id": "all"}),
required_features=[SUPPORT_A | SUPPORT_B],
@@ -857,7 +857,7 @@ async def test_call_with_one_of_required_features(
test_service_mock = AsyncMock(return_value=None)
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
test_service_mock,
ServiceCall("test_domain", "test_service", {"entity_id": "all"}),
required_features=[SUPPORT_A, SUPPORT_C],
@@ -878,7 +878,7 @@ async def test_call_with_sync_func(hass: HomeAssistant, mock_entities) -> None:
test_service_mock = Mock(return_value=None)
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
test_service_mock,
ServiceCall("test_domain", "test_service", {"entity_id": "light.kitchen"}),
)
@@ -890,7 +890,7 @@ async def test_call_with_sync_attr(hass: HomeAssistant, mock_entities) -> None:
mock_method = mock_entities["light.kitchen"].sync_method = Mock(return_value=None)
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
"sync_method",
ServiceCall(
"test_domain",
@@ -908,7 +908,7 @@ async def test_call_context_user_not_exist(hass: HomeAssistant) -> None:
with pytest.raises(exceptions.UnknownUser) as err:
await service.entity_service_call(
hass,
[],
{},
Mock(),
ServiceCall(
"test_domain",
@@ -935,7 +935,7 @@ async def test_call_context_target_all(
):
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall(
"test_domain",
@@ -963,7 +963,7 @@ async def test_call_context_target_specific(
):
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall(
"test_domain",
@@ -987,7 +987,7 @@ async def test_call_context_target_specific_no_auth(
):
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall(
"test_domain",
@@ -1007,7 +1007,7 @@ async def test_call_no_context_target_all(
"""Check we target all if no user context given."""
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall(
"test_domain", "test_service", data={"entity_id": ENTITY_MATCH_ALL}
@@ -1026,7 +1026,7 @@ async def test_call_no_context_target_specific(
"""Check we can target specified entities."""
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall(
"test_domain",
@@ -1048,7 +1048,7 @@ async def test_call_with_match_all(
"""Check we only target allowed entities if targeting all."""
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall("test_domain", "test_service", {"entity_id": "all"}),
)
@@ -1065,7 +1065,7 @@ async def test_call_with_omit_entity_id(
"""Check service call if we do not pass an entity ID."""
await service.entity_service_call(
hass,
[Mock(entities=mock_entities)],
mock_entities,
Mock(),
ServiceCall("test_domain", "test_service"),
)