27 lines
820 B
Python
27 lines
820 B
Python
|
"""Tests for instance ID helper."""
|
||
|
|
from tests.async_mock import patch
|
||
|
|
|
||
|
|
|
||
|
|
async def test_get_id_empty(hass, hass_storage):
|
||
|
|
"""Get unique ID."""
|
||
|
|
uuid = await hass.helpers.instance_id.async_get()
|
||
|
|
assert uuid is not None
|
||
|
|
# Assert it's stored
|
||
|
|
assert hass_storage["core.uuid"]["data"]["uuid"] == uuid
|
||
|
|
|
||
|
|
|
||
|
|
async def test_get_id_migrate(hass, hass_storage):
|
||
|
|
"""Migrate existing file."""
|
||
|
|
with patch(
|
||
|
|
"homeassistant.util.json.load_json", return_value={"uuid": "1234"}
|
||
|
|
), patch("os.path.isfile", return_value=True), patch("os.remove") as mock_remove:
|
||
|
|
uuid = await hass.helpers.instance_id.async_get()
|
||
|
|
|
||
|
|
assert uuid == "1234"
|
||
|
|
|
||
|
|
# Assert it's stored
|
||
|
|
assert hass_storage["core.uuid"]["data"]["uuid"] == uuid
|
||
|
|
|
||
|
|
# assert old deleted
|
||
|
|
assert len(mock_remove.mock_calls) == 1
|