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

Cleanups in device registry tests (#110786)

Browse Source
This commit is contained in:
Franck Nijhof 2024-02-17 14:06:53 +01:00 committed by GitHub
parent 42cf081582
commit 0fbadc274a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 212 additions and 183 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
tests/components/config/test_device_registry.py
View File

@ -7,7 +7,7 @@ from homeassistant.helpers import device_registry as dr
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import MockConfigEntry, MockModule, mock_integration from tests.common import MockConfigEntry, MockModule, mock_integration
from tests.typing import WebSocketGenerator from tests.typing import MockHAClientWebSocket, WebSocketGenerator
@pytest.fixture(autouse=True, name="stub_blueprint_populate") @pytest.fixture(autouse=True, name="stub_blueprint_populate")
@ -15,15 +15,19 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
"""Stub copying the blueprints to the config folder.""" """Stub copying the blueprints to the config folder."""
@pytest.fixture @pytest.fixture(name="client")
async def client(hass, hass_ws_client): async def client_fixture(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator
) -> MockHAClientWebSocket:
"""Fixture that can interact with the config manager API.""" """Fixture that can interact with the config manager API."""
device_registry.async_setup(hass) device_registry.async_setup(hass)
return await hass_ws_client(hass) return await hass_ws_client(hass)
async def test_list_devices( async def test_list_devices(
hass: HomeAssistant, client, device_registry: dr.DeviceRegistry hass: HomeAssistant,
client: MockHAClientWebSocket,
device_registry: dr.DeviceRegistry,
) -> None: ) -> None:
"""Test list entries.""" """Test list entries."""
entry = MockConfigEntry(title=None) entry = MockConfigEntry(title=None)
@ -44,10 +48,10 @@ async def test_list_devices(
entry_type=dr.DeviceEntryType.SERVICE, entry_type=dr.DeviceEntryType.SERVICE,
) )
await client.send_json({"id": 5, "type": "config/device_registry/list"}) await client.send_json_auto_id({"type": "config/device_registry/list"})
msg = await client.receive_json() msg = await client.receive_json()
dev1, dev2 = (entry.pop("id") for entry in msg["result"]) dev1, _ = (entry.pop("id") for entry in msg["result"])
assert msg["result"] == [ assert msg["result"] == [
{ {
@ -92,7 +96,7 @@ async def test_list_devices(
device_registry.async_update_device(device2.id, name=Unserializable()) device_registry.async_update_device(device2.id, name=Unserializable())
await hass.async_block_till_done() await hass.async_block_till_done()
await client.send_json({"id": 6, "type": "config/device_registry/list"}) await client.send_json_auto_id({"type": "config/device_registry/list"})
msg = await client.receive_json() msg = await client.receive_json()
assert msg["result"] == [ assert msg["result"] == [
@ -134,10 +138,10 @@ async def test_list_devices(
) )
async def test_update_device( async def test_update_device(
hass: HomeAssistant, hass: HomeAssistant,
client, client: MockHAClientWebSocket,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
payload_key, payload_key: str,
payload_value, payload_value: str | None | dr.DeviceEntryDisabler,
) -> None: ) -> None:
"""Test update entry.""" """Test update entry."""
entry = MockConfigEntry(title=None) entry = MockConfigEntry(title=None)
@ -152,9 +156,8 @@ async def test_update_device(
assert not getattr(device, payload_key) assert not getattr(device, payload_key)
await client.send_json( await client.send_json_auto_id(
{ {
"id": 1,
"type": "config/device_registry/update", "type": "config/device_registry/update",
"device_id": device.id, "device_id": device.id,
payload_key: payload_value, payload_key: payload_value,
@ -231,9 +234,8 @@ async def test_remove_config_entry_from_device(
# Try removing a config entry from the device, it should fail because # Try removing a config entry from the device, it should fail because
# async_remove_config_entry_device returns False # async_remove_config_entry_device returns False
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 5,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_1.entry_id, "config_entry_id": entry_1.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -248,9 +250,8 @@ async def test_remove_config_entry_from_device(
can_remove = True can_remove = True
# Remove the 1st config entry # Remove the 1st config entry
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 6,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_1.entry_id, "config_entry_id": entry_1.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -267,9 +268,8 @@ async def test_remove_config_entry_from_device(
} }
# Remove the 2nd config entry # Remove the 2nd config entry
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 7,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_2.entry_id, "config_entry_id": entry_2.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -354,9 +354,8 @@ async def test_remove_config_entry_from_device_fails(
assert device_entry.id != fake_device_id assert device_entry.id != fake_device_id
# Try removing a non existing config entry from the device # Try removing a non existing config entry from the device
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 5,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": fake_entry_id, "config_entry_id": fake_entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -369,9 +368,8 @@ async def test_remove_config_entry_from_device_fails(
assert response["error"]["message"] == "Unknown config entry" assert response["error"]["message"] == "Unknown config entry"
# Try removing a config entry which does not support removal from the device # Try removing a config entry which does not support removal from the device
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 6,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_1.entry_id, "config_entry_id": entry_1.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -386,9 +384,8 @@ async def test_remove_config_entry_from_device_fails(
) )
# Try removing a config entry from a device which does not exist # Try removing a config entry from a device which does not exist
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 7,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_2.entry_id, "config_entry_id": entry_2.entry_id,
"device_id": fake_device_id, "device_id": fake_device_id,
@ -401,9 +398,8 @@ async def test_remove_config_entry_from_device_fails(
assert response["error"]["message"] == "Unknown device" assert response["error"]["message"] == "Unknown device"
# Try removing a config entry from a device which it's not connected to # Try removing a config entry from a device which it's not connected to
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 8,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_2.entry_id, "config_entry_id": entry_2.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -417,9 +413,8 @@ async def test_remove_config_entry_from_device_fails(
entry_3.entry_id, entry_3.entry_id,
} }
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 9,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_2.entry_id, "config_entry_id": entry_2.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,
@ -432,9 +427,8 @@ async def test_remove_config_entry_from_device_fails(
assert response["error"]["message"] == "Config entry not in device" assert response["error"]["message"] == "Config entry not in device"
# Try removing a config entry which can't be loaded from a device - allowed # Try removing a config entry which can't be loaded from a device - allowed
await ws_client.send_json( await ws_client.send_json_auto_id(
{ {
"id": 10,
"type": "config/device_registry/remove_config_entry", "type": "config/device_registry/remove_config_entry",
"config_entry_id": entry_3.entry_id, "config_entry_id": entry_3.entry_id,
"device_id": device_entry.id, "device_id": device_entry.id,

341
tests/helpers/test_device_registry.py
View File

@ -9,7 +9,7 @@ from yarl import URL
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.const import EVENT_HOMEASSISTANT_STARTED from homeassistant.const import EVENT_HOMEASSISTANT_STARTED
from homeassistant.core import CoreState, HomeAssistant, callback from homeassistant.core import CoreState, HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import ( from homeassistant.helpers import (
area_registry as ar, area_registry as ar,
@ -19,26 +19,13 @@ from homeassistant.helpers import (
from tests.common import ( from tests.common import (
MockConfigEntry, MockConfigEntry,
async_capture_events,
flush_store, flush_store,
help_test_all, help_test_all,
import_and_test_deprecated_constant_enum, import_and_test_deprecated_constant_enum,
) )
@pytest.fixture
def update_events(hass):
"""Capture update events."""
events = []
@callback
def async_capture(event):
events.append(event.data)
hass.bus.async_listen(dr.EVENT_DEVICE_REGISTRY_UPDATED, async_capture)
return events
@pytest.fixture @pytest.fixture
def mock_config_entry(hass: HomeAssistant) -> MockConfigEntry: def mock_config_entry(hass: HomeAssistant) -> MockConfigEntry:
"""Create a mock config entry and add it to hass.""" """Create a mock config entry and add it to hass."""
@ -52,9 +39,9 @@ async def test_get_or_create_returns_same_entry(
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
area_registry: ar.AreaRegistry, area_registry: ar.AreaRegistry,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
update_events,
) -> None: ) -> None:
"""Make sure we do not duplicate entries.""" """Make sure we do not duplicate entries."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id, config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
@ -101,13 +88,14 @@ async def test_get_or_create_returns_same_entry(
# Only 2 update events. The third entry did not generate any changes. # Only 2 update events. The third entry did not generate any changes.
assert len(update_events) == 2 assert len(update_events) == 2
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry.id assert update_events[1].data == {
assert update_events[1]["changes"] == { "action": "update",
"connections": {("mac", "12:34:56:ab:cd:ef")} "device_id": entry.id,
"changes": {"connections": {("mac", "12:34:56:ab:cd:ef")}},
} }
@ -656,9 +644,10 @@ async def test_migration_1_3_to_1_4(
async def test_removing_config_entries( async def test_removing_config_entries(
hass: HomeAssistant, device_registry: dr.DeviceRegistry, update_events hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Make sure we do not get duplicate entries.""" """Make sure we do not get duplicate entries."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
config_entry_1 = MockConfigEntry() config_entry_1 = MockConfigEntry()
config_entry_1.add_to_hass(hass) config_entry_1.add_to_hass(hass)
config_entry_2 = MockConfigEntry() config_entry_2 = MockConfigEntry()
@ -703,29 +692,37 @@ async def test_removing_config_entries(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 5 assert len(update_events) == 5
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry2.id assert update_events[1].data == {
assert update_events[1]["changes"] == {"config_entries": {config_entry_1.entry_id}} "action": "update",
assert update_events[2]["action"] == "create" "device_id": entry.id,
assert update_events[2]["device_id"] == entry3.id "changes": {"config_entries": {config_entry_1.entry_id}},
assert "changes" not in update_events[2] }
assert update_events[3]["action"] == "update" assert update_events[2].data == {
assert update_events[3]["device_id"] == entry.id "action": "create",
assert update_events[3]["changes"] == { "device_id": entry3.id,
"config_entries": {config_entry_1.entry_id, config_entry_2.entry_id} }
assert update_events[3].data == {
"action": "update",
"device_id": entry.id,
"changes": {
"config_entries": {config_entry_1.entry_id, config_entry_2.entry_id}
},
}
assert update_events[4].data == {
"action": "remove",
"device_id": entry3.id,
} }
assert update_events[4]["action"] == "remove"
assert update_events[4]["device_id"] == entry3.id
assert "changes" not in update_events[4]
async def test_deleted_device_removing_config_entries( async def test_deleted_device_removing_config_entries(
hass: HomeAssistant, device_registry: dr.DeviceRegistry, update_events hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Make sure we do not get duplicate entries.""" """Make sure we do not get duplicate entries."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
config_entry_1 = MockConfigEntry() config_entry_1 = MockConfigEntry()
config_entry_1.add_to_hass(hass) config_entry_1.add_to_hass(hass)
config_entry_2 = MockConfigEntry() config_entry_2 = MockConfigEntry()
@ -767,21 +764,27 @@ async def test_deleted_device_removing_config_entries(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 5 assert len(update_events) == 5
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry2.id assert update_events[1].data == {
assert update_events[1]["changes"] == {"config_entries": {config_entry_1.entry_id}} "action": "update",
assert update_events[2]["action"] == "create" "device_id": entry2.id,
assert update_events[2]["device_id"] == entry3.id "changes": {"config_entries": {config_entry_1.entry_id}},
assert "changes" not in update_events[2]["device_id"] }
assert update_events[3]["action"] == "remove" assert update_events[2].data == {
assert update_events[3]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[3] "device_id": entry3.id,
assert update_events[4]["action"] == "remove" }
assert update_events[4]["device_id"] == entry3.id assert update_events[3].data == {
assert "changes" not in update_events[4] "action": "remove",
"device_id": entry.id,
}
assert update_events[4].data == {
"action": "remove",
"device_id": entry3.id,
}
device_registry.async_clear_config_entry(config_entry_1.entry_id) device_registry.async_clear_config_entry(config_entry_1.entry_id)
assert len(device_registry.devices) == 0 assert len(device_registry.devices) == 0
@ -1105,9 +1108,9 @@ async def test_update(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
update_events,
) -> None: ) -> None:
"""Verify that we can update some attributes of a device.""" """Verify that we can update some attributes of a device."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id, config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
@ -1180,33 +1183,37 @@ async def test_update(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 2 assert len(update_events) == 2
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry.id assert update_events[1].data == {
assert update_events[1]["changes"] == { "action": "update",
"area_id": None, "device_id": entry.id,
"configuration_url": None, "changes": {
"disabled_by": None, "area_id": None,
"entry_type": None, "configuration_url": None,
"hw_version": None, "disabled_by": None,
"identifiers": {("bla", "123"), ("hue", "456")}, "entry_type": None,
"manufacturer": None, "hw_version": None,
"model": None, "identifiers": {("bla", "123"), ("hue", "456")},
"name": None, "manufacturer": None,
"name_by_user": None, "model": None,
"serial_number": None, "name": None,
"suggested_area": None, "name_by_user": None,
"sw_version": None, "serial_number": None,
"via_device_id": None, "suggested_area": None,
"sw_version": None,
"via_device_id": None,
},
} }
async def test_update_remove_config_entries( async def test_update_remove_config_entries(
hass: HomeAssistant, device_registry: dr.DeviceRegistry, update_events hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Make sure we do not get duplicate entries.""" """Make sure we do not get duplicate entries."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
config_entry_1 = MockConfigEntry() config_entry_1 = MockConfigEntry()
config_entry_1.add_to_hass(hass) config_entry_1.add_to_hass(hass)
config_entry_2 = MockConfigEntry() config_entry_2 = MockConfigEntry()
@ -1256,23 +1263,30 @@ async def test_update_remove_config_entries(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 5 assert len(update_events) == 5
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry2.id assert update_events[1].data == {
assert update_events[1]["changes"] == {"config_entries": {config_entry_1.entry_id}} "action": "update",
assert update_events[2]["action"] == "create" "device_id": entry2.id,
assert update_events[2]["device_id"] == entry3.id "changes": {"config_entries": {config_entry_1.entry_id}},
assert "changes" not in update_events[2] }
assert update_events[3]["action"] == "update" assert update_events[2].data == {
assert update_events[3]["device_id"] == entry.id "action": "create",
assert update_events[3]["changes"] == { "device_id": entry3.id,
"config_entries": {config_entry_1.entry_id, config_entry_2.entry_id} }
assert update_events[3].data == {
"action": "update",
"device_id": entry.id,
"changes": {
"config_entries": {config_entry_1.entry_id, config_entry_2.entry_id}
},
}
assert update_events[4].data == {
"action": "remove",
"device_id": entry3.id,
} }
assert update_events[4]["action"] == "remove"
assert update_events[4]["device_id"] == entry3.id
assert "changes" not in update_events[4]
async def test_update_suggested_area( async def test_update_suggested_area(
@ -1280,9 +1294,9 @@ async def test_update_suggested_area(
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
area_registry: ar.AreaRegistry, area_registry: ar.AreaRegistry,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
update_events,
) -> None: ) -> None:
"""Verify that we can update the suggested area version of a device.""" """Verify that we can update the suggested area version of a device."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id, config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
@ -1310,12 +1324,15 @@ async def test_update_suggested_area(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 2 assert len(update_events) == 2
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update" }
assert update_events[1]["device_id"] == entry.id assert update_events[1].data == {
assert update_events[1]["changes"] == {"area_id": None, "suggested_area": None} "action": "update",
"device_id": entry.id,
"changes": {"area_id": None, "suggested_area": None},
}
# Do not save or fire the event if the suggested # Do not save or fire the event if the suggested
# area does not result in a change of area # area does not result in a change of area
@ -1457,9 +1474,9 @@ async def test_restore_device(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
update_events,
) -> None: ) -> None:
"""Make sure device id is stable.""" """Make sure device id is stable."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id, config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
@ -1503,27 +1520,31 @@ async def test_restore_device(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 4 assert len(update_events) == 4
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "remove" }
assert update_events[1]["device_id"] == entry.id assert update_events[1].data == {
assert "changes" not in update_events[1] "action": "remove",
assert update_events[2]["action"] == "create" "device_id": entry.id,
assert update_events[2]["device_id"] == entry2.id }
assert "changes" not in update_events[2] assert update_events[2].data == {
assert update_events[3]["action"] == "create" "action": "create",
assert update_events[3]["device_id"] == entry3.id "device_id": entry2.id,
assert "changes" not in update_events[3] }
assert update_events[3].data == {
"action": "create",
"device_id": entry3.id,
}
async def test_restore_simple_device( async def test_restore_simple_device(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
mock_config_entry: MockConfigEntry, mock_config_entry: MockConfigEntry,
update_events,
) -> None: ) -> None:
"""Make sure device id is stable.""" """Make sure device id is stable."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
config_entry_id=mock_config_entry.entry_id, config_entry_id=mock_config_entry.entry_id,
connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")}, connections={(dr.CONNECTION_NETWORK_MAC, "12:34:56:AB:CD:EF")},
@ -1557,24 +1578,29 @@ async def test_restore_simple_device(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 4 assert len(update_events) == 4
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "remove" }
assert update_events[1]["device_id"] == entry.id assert update_events[1].data == {
assert "changes" not in update_events[1] "action": "remove",
assert update_events[2]["action"] == "create" "device_id": entry.id,
assert update_events[2]["device_id"] == entry2.id }
assert "changes" not in update_events[2] assert update_events[2].data == {
assert update_events[3]["action"] == "create" "action": "create",
assert update_events[3]["device_id"] == entry3.id "device_id": entry2.id,
assert "changes" not in update_events[3] }
assert update_events[3].data == {
"action": "create",
"device_id": entry3.id,
}
async def test_restore_shared_device( async def test_restore_shared_device(
hass: HomeAssistant, device_registry: dr.DeviceRegistry, update_events hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Make sure device id is stable for shared devices.""" """Make sure device id is stable for shared devices."""
update_events = async_capture_events(hass, dr.EVENT_DEVICE_REGISTRY_UPDATED)
config_entry_1 = MockConfigEntry() config_entry_1 = MockConfigEntry()
config_entry_1.add_to_hass(hass) config_entry_1.add_to_hass(hass)
config_entry_2 = MockConfigEntry() config_entry_2 = MockConfigEntry()
@ -1660,32 +1686,41 @@ async def test_restore_shared_device(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 7 assert len(update_events) == 7
assert update_events[0]["action"] == "create" assert update_events[0].data == {
assert update_events[0]["device_id"] == entry.id "action": "create",
assert "changes" not in update_events[0] "device_id": entry.id,
assert update_events[1]["action"] == "update"
assert update_events[1]["device_id"] == entry.id
assert update_events[1]["changes"] == {
"config_entries": {config_entry_1.entry_id},
"identifiers": {("entry_123", "0123")},
} }
assert update_events[2]["action"] == "remove" assert update_events[1].data == {
assert update_events[2]["device_id"] == entry.id "action": "update",
assert "changes" not in update_events[2] "device_id": entry.id,
assert update_events[3]["action"] == "create" "changes": {
assert update_events[3]["device_id"] == entry.id "config_entries": {config_entry_1.entry_id},
assert "changes" not in update_events[3] "identifiers": {("entry_123", "0123")},
assert update_events[4]["action"] == "remove" },
assert update_events[4]["device_id"] == entry.id }
assert "changes" not in update_events[4] assert update_events[2].data == {
assert update_events[5]["action"] == "create" "action": "remove",
assert update_events[5]["device_id"] == entry.id "device_id": entry.id,
assert "changes" not in update_events[5] }
assert update_events[6]["action"] == "update" assert update_events[3].data == {
assert update_events[6]["device_id"] == entry.id "action": "create",
assert update_events[6]["changes"] == { "device_id": entry.id,
"config_entries": {config_entry_2.entry_id}, }
"identifiers": {("entry_234", "2345")}, assert update_events[4].data == {
"action": "remove",
"device_id": entry.id,
}
assert update_events[5].data == {
"action": "create",
"device_id": entry.id,
}
assert update_events[6].data == {
"action": "update",
"device_id": entry.id,
"changes": {
"config_entries": {config_entry_2.entry_id},
"identifiers": {("entry_234", "2345")},
},
} }