82 lines
2.6 KiB
Python
82 lines
2.6 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(
|
|
connection=[['ethernet', '12:34:56:78:90:AB:CD:EF']],
|
|
identifiers=[['bridgeid', '0123']],
|
|
manufacturer='manufacturer', model='model')
|
|
entry2 = registry.async_get_or_create(
|
|
connection=[['ethernet', '11:22:33:44:55:66:77:88']],
|
|
identifiers=[['bridgeid', '0123']],
|
|
manufacturer='manufacturer', model='model')
|
|
entry3 = registry.async_get_or_create(
|
|
connection=[['ethernet', '12:34:56:78:90:AB:CD:EF']],
|
|
identifiers=[['bridgeid', '1234']],
|
|
manufacturer='manufacturer', model='model')
|
|
|
|
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(
|
|
connection=[['Zigbee', '01.23.45.67.89']],
|
|
identifiers=[['serial', '12:34:56:78:90:AB:CD:EF']],
|
|
manufacturer='manufacturer', model='model')
|
|
assert entry.id == 'abcdefghijklm'
|