jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Add core APIs to migrate device identifiers and entity unique_id (#23481)

Browse source 
* Add device identifiers migration

* Add entity unique_id migration

* Update per arch issue

* Move to existing update methods
This commit is contained in:
Andrew Sayre 2019-04-30 12:04:37 -05:00 • committed by Paulus Schoutsen
parent 41f0066e76
commit cfaaae661a
4 changed files with 61 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/helpers/test_entity_registry.py
View file

@ -271,3 +271,29 @@ async def test_loading_race_condition(hass):
mock_load.assert_called_once_with()
assert results[0] == results[1]
async def test_update_entity_unique_id(registry):
"""Test entity's unique_id is updated."""
entry = registry.async_get_or_create(
'light', 'hue', '5678', config_entry_id='mock-id-1')
new_unique_id = '1234'
with patch.object(registry, 'async_schedule_save') as mock_schedule_save:
updated_entry = registry.async_update_entity(
entry.entity_id, new_unique_id=new_unique_id)
assert updated_entry != entry
assert updated_entry.unique_id == new_unique_id
assert mock_schedule_save.call_count == 1
async def test_update_entity_unique_id_conflict(registry):
"""Test migration raises when unique_id already in use."""
entry = registry.async_get_or_create(
'light', 'hue', '5678', config_entry_id='mock-id-1')
entry2 = registry.async_get_or_create(
'light', 'hue', '1234', config_entry_id='mock-id-1')
with patch.object(registry, 'async_schedule_save') as mock_schedule_save, \
pytest.raises(ValueError):
registry.async_update_entity(
entry.entity_id, new_unique_id=entry2.unique_id)
assert mock_schedule_save.call_count == 0