Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Track primary integration (#119741)

Browse Source 
* Track primary integration

* Update snapshots

* More snapshots updated

* Uno mas

* Update snapshot
This commit is contained in:
Paulus Schoutsen
2024-06-16 13:26:06 -04:00
committed by GitHub
parent 59ca5b04fa
commit 836abe68c7
62 changed files with 1559 additions and 807 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/helpers/test_device_registry.py
View File

@@ -2628,3 +2628,39 @@ async def test_async_remove_device_thread_safety(
await hass.async_add_executor_job(
device_registry.async_remove_device, device.id
)
async def test_primary_integration(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
mock_config_entry: MockConfigEntry,
) -> None:
"""Test the primary integration field."""
# Update existing
device = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
identifiers=set(),
manufacturer="manufacturer",
model="model",
)
assert device.primary_integration is None
device = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
model="model 2",
domain="test_domain",
)
assert device.primary_integration == "test_domain"
# Create new
device = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
identifiers=set(),
manufacturer="manufacturer",
model="model",
domain="test_domain",
)
assert device.primary_integration == "test_domain"