mirror of
https://github.com/home-assistant/core.git
synced 2026-04-03 00:46:46 +01:00
108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
"""The tests for the hassio component."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Literal
|
|
|
|
from aiohttp import hdrs, web
|
|
import pytest
|
|
|
|
from homeassistant.components.hassio.handler import HassIO, HassioAPIError
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
|
|
async def test_api_ingress_panels(
|
|
hassio_handler: HassIO, aioclient_mock: AiohttpClientMocker
|
|
) -> None:
|
|
"""Test setup with API Ingress panels."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/ingress/panels",
|
|
json={
|
|
"result": "ok",
|
|
"data": {
|
|
"panels": {
|
|
"slug": {
|
|
"enable": True,
|
|
"title": "Test",
|
|
"icon": "mdi:test",
|
|
"admin": False,
|
|
}
|
|
}
|
|
},
|
|
},
|
|
)
|
|
|
|
data = await hassio_handler.get_ingress_panels()
|
|
assert aioclient_mock.call_count == 1
|
|
assert data["panels"]
|
|
assert "slug" in data["panels"]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("api_call", "method", "payload"),
|
|
[
|
|
("get_ingress_panels", "GET", None),
|
|
],
|
|
)
|
|
@pytest.mark.usefixtures("socket_enabled")
|
|
async def test_api_headers(
|
|
aiohttp_raw_server, # 'aiohttp_raw_server' must be before 'hass'!
|
|
hass: HomeAssistant,
|
|
api_call: str,
|
|
method: Literal["GET", "POST"],
|
|
payload: Any,
|
|
) -> None:
|
|
"""Test headers are forwarded correctly."""
|
|
received_request = None
|
|
|
|
async def mock_handler(request):
|
|
"""Return OK."""
|
|
nonlocal received_request
|
|
received_request = request
|
|
return web.json_response({"result": "ok", "data": None})
|
|
|
|
server = await aiohttp_raw_server(mock_handler)
|
|
hassio_handler = HassIO(
|
|
hass.loop,
|
|
async_get_clientsession(hass),
|
|
f"{server.host}:{server.port}",
|
|
)
|
|
|
|
api_func = getattr(hassio_handler, api_call)
|
|
if payload:
|
|
await api_func(payload)
|
|
else:
|
|
await api_func()
|
|
assert received_request is not None
|
|
|
|
assert received_request.method == method
|
|
assert received_request.headers.get("X-Hass-Source") == "core.handler"
|
|
|
|
if method == "GET":
|
|
assert hdrs.CONTENT_TYPE not in received_request.headers
|
|
return
|
|
|
|
assert hdrs.CONTENT_TYPE in received_request.headers
|
|
if payload:
|
|
assert received_request.headers[hdrs.CONTENT_TYPE] == "application/json"
|
|
else:
|
|
assert received_request.headers[hdrs.CONTENT_TYPE] == "application/octet-stream"
|
|
|
|
|
|
@pytest.mark.usefixtures("hassio_stubs")
|
|
async def test_send_command_invalid_command(hass: HomeAssistant) -> None:
|
|
"""Test send command fails when command is invalid."""
|
|
hassio: HassIO = hass.data["hassio"]
|
|
with pytest.raises(HassioAPIError):
|
|
# absolute path
|
|
await hassio.send_command("/test/../bad")
|
|
with pytest.raises(HassioAPIError):
|
|
# relative path
|
|
await hassio.send_command("test/../bad")
|
|
with pytest.raises(HassioAPIError):
|
|
# relative path with percent encoding
|
|
await hassio.send_command("test/%2E%2E/bad")
|