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

Add type to device registry (#35095)

This commit is contained in:
Paulus Schoutsen
2020-05-03 13:56:58 -07:00
committed by GitHub
parent 6f6c670b3b
commit b90cb09fd1
7 changed files with 42 additions and 13 deletions

View File

@@ -149,6 +149,7 @@ async def test_loading_from_storage(hass, hass_storage):
"model": "model",
"name": "name",
"sw_version": "version",
"entry_type": "service",
"area_id": "12345A",
"name_by_user": "Test Friendly Name",
}
@@ -168,6 +169,7 @@ async def test_loading_from_storage(hass, hass_storage):
assert entry.id == "abcdefghijklm"
assert entry.area_id == "12345A"
assert entry.name_by_user == "Test Friendly Name"
assert entry.entry_type == "service"
assert isinstance(entry.config_entries, set)
@@ -304,6 +306,9 @@ async def test_loading_saving_data(hass, registry):
identifiers={("hue", "0123")},
manufacturer="manufacturer",
model="via",
name="Original Name",
sw_version="Orig SW 1",
entry_type="device",
)
orig_light = registry.async_get_or_create(
@@ -317,6 +322,10 @@ async def test_loading_saving_data(hass, registry):
assert len(registry.devices) == 2
orig_via = registry.async_update_device(
orig_via.id, area_id="mock-area-id", name_by_user="mock-name-by-user"
)
# Now load written data in new registry
registry2 = device_registry.DeviceRegistry(hass)
await flush_store(registry._store)