mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-15 22:10:12 +00:00
f6faa18409
It seems that the codebase is not formatted with the latest ruff version. This PR reformats the codebase with ruff 0.5.7.
87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
"""Test websocket."""
|
|
|
|
# pylint: disable=protected-access, import-error
|
|
import asyncio
|
|
import logging
|
|
|
|
from awesomeversion import AwesomeVersion
|
|
|
|
from supervisor.const import CoreState
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.homeassistant.const import WSEvent, WSType
|
|
|
|
|
|
async def test_send_command(coresys: CoreSys):
|
|
"""Test websocket error on listen."""
|
|
client = coresys.homeassistant.websocket._client
|
|
await coresys.homeassistant.websocket.async_send_command({"type": "test"})
|
|
client.async_send_command.assert_called_with({"type": "test"})
|
|
|
|
await coresys.homeassistant.websocket.async_supervisor_update_event(
|
|
"test", {"lorem": "ipsum"}
|
|
)
|
|
client.async_send_command.assert_called_with(
|
|
{
|
|
"type": WSType.SUPERVISOR_EVENT,
|
|
"data": {
|
|
"event": WSEvent.SUPERVISOR_UPDATE,
|
|
"update_key": "test",
|
|
"data": {"lorem": "ipsum"},
|
|
},
|
|
}
|
|
)
|
|
|
|
|
|
async def test_send_command_old_core_version(coresys: CoreSys, caplog):
|
|
"""Test websocket error on listen."""
|
|
caplog.set_level(logging.INFO)
|
|
client = coresys.homeassistant.websocket._client
|
|
client.ha_version = AwesomeVersion("1970.1.1")
|
|
|
|
await coresys.homeassistant.websocket.async_send_command(
|
|
{"type": "supervisor/event"}
|
|
)
|
|
|
|
assert (
|
|
"WebSocket command supervisor/event is not supported until core-2021.2.4"
|
|
in caplog.text
|
|
)
|
|
|
|
await coresys.homeassistant.websocket.async_supervisor_update_event(
|
|
"test", {"lorem": "ipsum"}
|
|
)
|
|
client.async_send_command.assert_not_called()
|
|
|
|
|
|
async def test_send_message_during_startup(coresys: CoreSys):
|
|
"""Test websocket messages queue during startup."""
|
|
client = coresys.homeassistant.websocket._client
|
|
await coresys.homeassistant.websocket.load()
|
|
coresys.core.state = CoreState.SETUP
|
|
|
|
await coresys.homeassistant.websocket.async_supervisor_update_event(
|
|
"test", {"lorem": "ipsum"}
|
|
)
|
|
client.async_send_command.assert_not_called()
|
|
|
|
coresys.core.state = CoreState.RUNNING
|
|
await asyncio.sleep(0)
|
|
|
|
assert client.async_send_command.call_count == 2
|
|
assert client.async_send_command.call_args_list[0][0][0] == {
|
|
"type": WSType.SUPERVISOR_EVENT,
|
|
"data": {
|
|
"event": WSEvent.SUPERVISOR_UPDATE,
|
|
"update_key": "test",
|
|
"data": {"lorem": "ipsum"},
|
|
},
|
|
}
|
|
assert client.async_send_command.call_args_list[1][0][0] == {
|
|
"type": WSType.SUPERVISOR_EVENT,
|
|
"data": {
|
|
"event": WSEvent.SUPERVISOR_UPDATE,
|
|
"update_key": "info",
|
|
"data": {"state": "running"},
|
|
},
|
|
}
|