0009be595c
* First draft * Generate device id * No obscure registry * Dont store config_entry_id in device * Storage * Small mistake on rebase * Do storage more like entity registry * Improve device identification * Add tests * Remove deconz device support from PR * Fix hound comments, voff! * Fix comments and clean up * Fix proper indentation * Fix pydoc issues * Fix mochad component to not use self.device * Fix mochad light platform to not use self.device * Fix TankUtilitySensor to not use self.device * Fix Soundtouch to not use self.device * Fix Plex to not use self.device * Fix Emby to not use self.device * Fix Heatmiser to not use self.device * Fix Wemo lights to not use self.device * Fix Lifx to not use self.device * Fix Radiotherm to not use self.device * Fix Juicenet to not use self.device * Fix Qwikswitch to not use self.device * Fix Xiaomi miio to not use self.device * Fix Nest to not use self.device * Fix Tellduslive to not use self.device * Fix Knx to not use self.device * Clean up a small mistake in soundtouch * Fix comment from Ballob * Fix bad indentation * Fix indentatin * Lint * Remove unused variable * Lint
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
"""Tests for the Device Registry."""
|
|
import pytest
|
|
|
|
from homeassistant.helpers import device_registry
|
|
|
|
|
|
def mock_registry(hass, mock_entries=None):
|
|
"""Mock the Device Registry."""
|
|
registry = device_registry.DeviceRegistry(hass)
|
|
registry.devices = mock_entries or []
|
|
|
|
async def _get_reg():
|
|
return registry
|
|
|
|
hass.data[device_registry.DATA_REGISTRY] = \
|
|
hass.loop.create_task(_get_reg())
|
|
return registry
|
|
|
|
|
|
@pytest.fixture
|
|
def registry(hass):
|
|
"""Return an empty, loaded, registry."""
|
|
return mock_registry(hass)
|
|
|
|
|
|
async def test_get_or_create_returns_same_entry(registry):
|
|
"""Make sure we do not duplicate entries."""
|
|
entry = registry.async_get_or_create(
|
|
[['bridgeid', '0123']], 'manufacturer', 'model',
|
|
[['ethernet', '12:34:56:78:90:AB:CD:EF']])
|
|
entry2 = registry.async_get_or_create(
|
|
[['bridgeid', '0123']], 'manufacturer', 'model',
|
|
[['ethernet', '11:22:33:44:55:66:77:88']])
|
|
entry3 = registry.async_get_or_create(
|
|
[['bridgeid', '1234']], 'manufacturer', 'model',
|
|
[['ethernet', '12:34:56:78:90:AB:CD:EF']])
|
|
|
|
assert len(registry.devices) == 1
|
|
assert entry is entry2
|
|
assert entry is entry3
|
|
assert entry.identifiers == [['bridgeid', '0123']]
|
|
|
|
|
|
async def test_loading_from_storage(hass, hass_storage):
|
|
"""Test loading stored devices on start."""
|
|
hass_storage[device_registry.STORAGE_KEY] = {
|
|
'version': device_registry.STORAGE_VERSION,
|
|
'data': {
|
|
'devices': [
|
|
{
|
|
'connection': [
|
|
[
|
|
'Zigbee',
|
|
'01.23.45.67.89'
|
|
]
|
|
],
|
|
'id': 'abcdefghijklm',
|
|
'identifiers': [
|
|
[
|
|
'serial',
|
|
'12:34:56:78:90:AB:CD:EF'
|
|
]
|
|
],
|
|
'manufacturer': 'manufacturer',
|
|
'model': 'model',
|
|
'name': 'name',
|
|
'sw_version': 'version'
|
|
}
|
|
]
|
|
}
|
|
}
|
|
|
|
registry = await device_registry.async_get_registry(hass)
|
|
|
|
entry = registry.async_get_or_create(
|
|
[['serial', '12:34:56:78:90:AB:CD:EF']], 'manufacturer',
|
|
'model', [['Zigbee', '01.23.45.67.89']])
|
|
assert entry.id == 'abcdefghijklm'
|