mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-12 20:40:21 +00:00
96d5fc244e
* Separate startup event from update check event * Add a queue for messages sent during startup
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
"""Test Homeassistant module."""
|
|
|
|
import asyncio
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from supervisor.const import CoreState
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.docker.interface import DockerInterface
|
|
from supervisor.homeassistant.secrets import HomeAssistantSecrets
|
|
|
|
|
|
async def test_load(coresys: CoreSys, tmp_supervisor_data: Path):
|
|
"""Test homeassistant module load."""
|
|
client = coresys.homeassistant.websocket._client # pylint: disable=protected-access
|
|
with open(tmp_supervisor_data / "homeassistant" / "secrets.yaml", "w") as secrets:
|
|
secrets.write("hello: world\n")
|
|
|
|
# Unwrap read_secrets to prevent throttling between tests
|
|
with patch.object(DockerInterface, "attach") as attach, patch.object(
|
|
HomeAssistantSecrets,
|
|
"_read_secrets",
|
|
new=HomeAssistantSecrets._read_secrets.__wrapped__,
|
|
):
|
|
await coresys.homeassistant.load()
|
|
|
|
attach.assert_called_once()
|
|
|
|
assert coresys.homeassistant.secrets.secrets == {"hello": "world"}
|
|
|
|
coresys.core.state = CoreState.SETUP
|
|
await coresys.homeassistant.websocket.async_send_message({"lorem": "ipsum"})
|
|
client.async_send_command.assert_not_called()
|
|
|
|
coresys.core.state = CoreState.RUNNING
|
|
await asyncio.sleep(0)
|
|
assert client.async_send_command.call_args_list[0][0][0] == {"lorem": "ipsum"}
|