Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Unifiprotect replace direct mocks with MockConfigEntry for test_async_ufp_instance_for_config_entry_ids (#131736) 

Co-authored-by: J. Nick Koston <nick@koston.org>
pull/131702/head^2
Raphael Hehl 2024-11-27 18:03:34 +01:00 committed by GitHub
parent fda178da23
commit 1f1fdf80db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
tests/components/unifiprotect/test_init.py
View File

@ -297,15 +297,15 @@ async def test_device_remove_devices_nvr(
[
pytest.param(
[
Mock(
spec=ConfigEntry,
MockConfigEntry(
domain=DOMAIN,
runtime_data=Mock(api="mock_api_instance_1"),
entry_id="1",
data={},
),
Mock(
spec=ConfigEntry,
MockConfigEntry(
domain="other_domain",
runtime_data=Mock(api="mock_api_instance_2"),
entry_id="2",
data={},
),
],
"mock_api_instance_1",
@ -313,15 +313,15 @@ async def test_device_remove_devices_nvr(
),
pytest.param(
[
Mock(
spec=ConfigEntry,
MockConfigEntry(
domain="other_domain",
runtime_data=Mock(api="mock_api_instance_1"),
entry_id="1",
data={},
),
Mock(
spec=ConfigEntry,
MockConfigEntry(
domain="other_domain",
runtime_data=Mock(api="mock_api_instance_2"),
entry_id="2",
data={},
),
],
None,
@ -330,22 +330,17 @@ async def test_device_remove_devices_nvr(
],
)
async def test_async_ufp_instance_for_config_entry_ids(
mock_entries, expected_result
hass: HomeAssistant,
mock_entries: list[MockConfigEntry],
expected_result: str | None,
) -> None:
"""Test async_ufp_instance_for_config_entry_ids with various entry configurations."""
hass = Mock(spec=HomeAssistant)
for index, entry in enumerate(mock_entries):
entry.add_to_hass(hass)
entry.runtime_data = Mock(api=f"mock_api_instance_{index + 1}")
mock_entry_mapping = {
str(index): entry for index, entry in enumerate(mock_entries, start=1)
}
def mock_async_get_entry(entry_id):
return mock_entry_mapping.get(entry_id)
hass.config_entries.async_get_entry = Mock(side_effect=mock_async_get_entry)
entry_ids = set(mock_entry_mapping.keys())
entry_ids = {entry.entry_id for entry in mock_entries}
result = async_ufp_instance_for_config_entry_ids(hass, entry_ids)