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

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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"