core/tests/components/frontend/test_init.py

811 lines
25 KiB
Python
Raw Normal View History

2016-03-09 09:25:50 +00:00
"""The tests for Home Assistant frontend."""
from asyncio import AbstractEventLoop
from http import HTTPStatus
from pathlib import Path
import re
from typing import Any
2021-01-01 21:31:56 +00:00
from unittest.mock import patch
from aiohttp.test_utils import TestClient
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components.frontend import (
CONF_EXTRA_JS_URL_ES5,
CONF_EXTRA_MODULE_URL,
CONF_THEMES,
DEFAULT_THEME_COLOR,
DOMAIN,
2019-07-31 19:25:30 +00:00
EVENT_PANELS_UPDATED,
THEMES_STORAGE_KEY,
add_extra_js_url,
async_register_built_in_panel,
async_remove_panel,
remove_extra_js_url,
2019-07-31 19:25:30 +00:00
)
from homeassistant.components.websocket_api import TYPE_RESULT
from homeassistant.core import HomeAssistant
from homeassistant.loader import async_get_integration
from homeassistant.setup import async_setup_component
from tests.common import MockUser, async_capture_events, async_fire_time_changed
from tests.typing import (
ClientSessionGenerator,
MockHAClientWebSocket,
WebSocketGenerator,
)
MOCK_THEMES = {
"happy": {"primary-color": "red", "app-header-background-color": "blue"},
"dark": {"primary-color": "black"},
"light_only": {
"primary-color": "blue",
"modes": {
"light": {"secondary-color": "black"},
},
},
"dark_only": {
"primary-color": "blue",
"modes": {
"dark": {"secondary-color": "white"},
},
},
"light_and_dark": {
"primary-color": "blue",
"modes": {
"light": {"secondary-color": "black"},
"dark": {"secondary-color": "white"},
},
},
}
CONFIG_THEMES = {DOMAIN: {CONF_THEMES: MOCK_THEMES}}
@pytest.fixture
async def ignore_frontend_deps(hass):
"""Frontend dependencies."""
frontend = await async_get_integration(hass, "frontend")
for dep in frontend.dependencies:
if dep not in ("http", "websocket_api"):
hass.config.components.add(dep)
@pytest.fixture
async def frontend(hass, ignore_frontend_deps):
"""Frontend setup with themes."""
assert await async_setup_component(
hass,
"frontend",
{},
)
@pytest.fixture
async def frontend_themes(hass):
"""Frontend setup with themes."""
assert await async_setup_component(
hass,
"frontend",
CONFIG_THEMES,
)
@pytest.fixture
def aiohttp_client(
event_loop: AbstractEventLoop,
aiohttp_client: ClientSessionGenerator,
socket_enabled: None,
) -> ClientSessionGenerator:
"""Return aiohttp_client and allow opening sockets."""
return aiohttp_client
@pytest.fixture
async def mock_http_client(
hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, frontend
) -> TestClient:
"""Start the Home Assistant HTTP component."""
return await aiohttp_client(hass.http.app)
@pytest.fixture
async def themes_ws_client(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, frontend_themes
) -> MockHAClientWebSocket:
"""Start the Home Assistant HTTP component."""
return await hass_ws_client(hass)
@pytest.fixture
async def ws_client(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, frontend
) -> MockHAClientWebSocket:
"""Start the Home Assistant HTTP component."""
return await hass_ws_client(hass)
@pytest.fixture
async def mock_http_client_with_extra_js(
hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, ignore_frontend_deps
) -> TestClient:
"""Start the Home Assistant HTTP component."""
assert await async_setup_component(
hass,
"frontend",
{
DOMAIN: {
CONF_EXTRA_MODULE_URL: ["/local/my_module.js"],
CONF_EXTRA_JS_URL_ES5: ["/local/my_es5.js"],
}
},
)
return await aiohttp_client(hass.http.app)
@pytest.fixture
def mock_onboarded():
"""Mock that we're onboarded."""
2019-07-31 19:25:30 +00:00
with patch(
"homeassistant.components.onboarding.async_is_onboarded", return_value=True
):
yield
async def test_frontend_and_static(mock_http_client, mock_onboarded) -> None:
"""Test if we can get the frontend."""
resp = await mock_http_client.get("")
assert resp.status == 200
2019-07-31 19:25:30 +00:00
assert "cache-control" not in resp.headers
text = await resp.text()
# Test we can retrieve frontend.js
2023-04-04 04:11:20 +00:00
frontendjs = re.search(r"(?P<app>\/frontend_es5\/app.[A-Za-z0-9_-]{11}.js)", text)
Load requirements and dependencies from manifests. Fallback to current `REQUIREMENTS` and `DEPENDENCIES` (#22717) * Load dependencies from manifests. Fallback to current DEPENDENCIES * Fix typing * Ignore typing correctly * Split out dependency processing to a new method * Fix tests * Only pull from manifest if dependencies is non empty * Inline temporary function * Fix light tests [skip ci] * Fix tests/common * Fix some mqtt tests [skip ci] * Fix tests and component manifests which have only one platform * Fix rflink tests * Fix more tests and manifests * Readability over shorthand format * Fix demo/notify tests * Load dependencies from manifests. Fallback to current DEPENDENCIES * Load requirements from manifests. Fallback to current REQUIREMENTS * Fix typing * Ignore typing correctly * Split out dependency processing to a new method * Only pull from manifest if dependencies is non empty * Inline temporary function * Fix tests and component manifests which have only one platform * Fix rflink tests * Readability over shorthand format * Clean up requirements * Use integration to resolve deps/reqs * Lint * Lint * revert a change * Revert a test change * Fix types * Fix types * Add back cache for load component * Fix test_component_not_found * Move light.test and device_tracker.test into test package instead with manifest to fix tests * Fix broken device_tracker tests * Add docstrings to __init__ * Fix all of the light tests that I broke earlier * Embed the test.switch platform to fix other tests * Embed and fix the test.imagimage_processing platform * Fix tests for nx584 * Add dependencies from platform file's DEPENDENCIES * Try to setup component when entity_platform is setting up Fix tests in helpers folder * Rewrite test_setup * Simplify * Lint * Disable demo component if running in test Temp workaround to unblock CI tests * Skip demo tests * Fix config entry test * Fix repeat test * Clarify doc * One extra guard * Fix import * Lint * Workaround google tts
2019-04-11 08:26:36 +00:00
assert frontendjs is not None, text
resp = await mock_http_client.get(frontendjs.groups(0)[0])
assert resp.status == 200
2019-07-31 19:25:30 +00:00
assert "public" in resp.headers.get("cache-control")
async def test_dont_cache_service_worker(mock_http_client) -> None:
"""Test that we don't cache the service worker."""
resp = await mock_http_client.get("/service_worker.js")
assert resp.status == 200
2019-07-31 19:25:30 +00:00
assert "cache-control" not in resp.headers
async def test_404(mock_http_client) -> None:
"""Test for HTTP 404 error."""
resp = await mock_http_client.get("/not-existing")
assert resp.status == HTTPStatus.NOT_FOUND
2016-05-14 07:58:36 +00:00
async def test_we_cannot_POST_to_root(mock_http_client) -> None:
"""Test that POST is not allow to root."""
resp = await mock_http_client.post("/")
assert resp.status == 405
async def test_themes_api(hass: HomeAssistant, themes_ws_client) -> None:
"""Test that /api/themes returns correct data."""
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["result"]["default_theme"] == "default"
assert msg["result"]["default_dark_theme"] is None
assert msg["result"]["themes"] == MOCK_THEMES
# recovery mode
hass.config.recovery_mode = True
await themes_ws_client.send_json({"id": 6, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_theme"] == "default"
assert msg["result"]["themes"] == {}
# safe mode
hass.config.recovery_mode = False
hass.config.safe_mode = True
await themes_ws_client.send_json({"id": 7, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_theme"] == "default"
assert msg["result"]["themes"] == {}
async def test_themes_persist(
hass: HomeAssistant,
hass_storage: dict[str, Any],
hass_ws_client: WebSocketGenerator,
ignore_frontend_deps,
) -> None:
"""Test that theme settings are restores after restart."""
hass_storage[THEMES_STORAGE_KEY] = {
"key": THEMES_STORAGE_KEY,
"version": 1,
"data": {
"frontend_default_theme": "happy",
"frontend_default_dark_theme": "dark",
},
}
assert await async_setup_component(hass, "frontend", CONFIG_THEMES)
themes_ws_client = await hass_ws_client(hass)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_theme"] == "happy"
assert msg["result"]["default_dark_theme"] == "dark"
async def test_themes_save_storage(
hass: HomeAssistant,
hass_storage: dict[str, Any],
freezer: FrozenDateTimeFactory,
frontend_themes,
) -> None:
"""Test that theme settings are restores after restart."""
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "happy"}, blocking=True
)
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "dark", "mode": "dark"}, blocking=True
)
# To trigger the call_later
freezer.tick(60.0)
async_fire_time_changed(hass)
# To execute the save
await hass.async_block_till_done()
assert hass_storage[THEMES_STORAGE_KEY]["data"] == {
"frontend_default_theme": "happy",
"frontend_default_dark_theme": "dark",
}
async def test_themes_set_theme(hass: HomeAssistant, themes_ws_client) -> None:
"""Test frontend.set_theme service."""
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
DOMAIN, "set_theme", {"name": "happy"}, blocking=True
)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["result"]["default_theme"] == "happy"
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
DOMAIN, "set_theme", {"name": "default"}, blocking=True
)
await themes_ws_client.send_json({"id": 6, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["result"]["default_theme"] == "default"
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "happy"}, blocking=True
)
await hass.services.async_call(DOMAIN, "set_theme", {"name": "none"}, blocking=True)
await themes_ws_client.send_json({"id": 7, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_theme"] == "default"
async def test_themes_set_theme_wrong_name(
hass: HomeAssistant, themes_ws_client
) -> None:
"""Test frontend.set_theme service called with wrong name."""
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
DOMAIN, "set_theme", {"name": "wrong"}, blocking=True
)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["result"]["default_theme"] == "default"
async def test_themes_set_dark_theme(hass: HomeAssistant, themes_ws_client) -> None:
"""Test frontend.set_theme service called with dark mode."""
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "dark", "mode": "dark"}, blocking=True
)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_dark_theme"] == "dark"
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "default", "mode": "dark"}, blocking=True
)
await themes_ws_client.send_json({"id": 6, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_dark_theme"] == "default"
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "none", "mode": "dark"}, blocking=True
)
await themes_ws_client.send_json({"id": 7, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_dark_theme"] is None
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "light_and_dark", "mode": "dark"}, blocking=True
)
await themes_ws_client.send_json({"id": 8, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_dark_theme"] == "light_and_dark"
async def test_themes_set_dark_theme_wrong_name(
hass: HomeAssistant, frontend, themes_ws_client
) -> None:
"""Test frontend.set_theme service called with mode dark and wrong name."""
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "wrong", "mode": "dark"}, blocking=True
)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
assert msg["result"]["default_dark_theme"] is None
async def test_themes_reload_themes(
hass: HomeAssistant, frontend, themes_ws_client
) -> None:
"""Test frontend.reload_themes service."""
2019-07-31 19:25:30 +00:00
with patch(
"homeassistant.components.frontend.async_hass_config_yaml",
2019-07-31 19:25:30 +00:00
return_value={DOMAIN: {CONF_THEMES: {"sad": {"primary-color": "blue"}}}},
):
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
DOMAIN, "set_theme", {"name": "happy"}, blocking=True
)
await hass.services.async_call(DOMAIN, "reload_themes", blocking=True)
await themes_ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await themes_ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["result"]["themes"] == {"sad": {"primary-color": "blue"}}
assert msg["result"]["default_theme"] == "default"
async def test_missing_themes(hass: HomeAssistant, ws_client) -> None:
"""Test that themes API works when themes are not defined."""
await ws_client.send_json({"id": 5, "type": "frontend/get_themes"})
msg = await ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"]["default_theme"] == "default"
assert msg["result"]["themes"] == {}
@pytest.mark.usefixtures("mock_onboarded")
async def test_extra_js(hass: HomeAssistant, mock_http_client_with_extra_js) -> None:
"""Test that extra javascript is loaded."""
async def get_response():
resp = await mock_http_client_with_extra_js.get("")
assert resp.status == 200
assert "cache-control" not in resp.headers
return await resp.text()
text = await get_response()
assert '"/local/my_module.js"' in text
assert '"/local/my_es5.js"' in text
# Test dynamically adding and removing extra javascript
add_extra_js_url(hass, "/local/my_module_2.js", False)
add_extra_js_url(hass, "/local/my_es5_2.js", True)
text = await get_response()
assert '"/local/my_module_2.js"' in text
assert '"/local/my_es5_2.js"' in text
remove_extra_js_url(hass, "/local/my_module_2.js", False)
remove_extra_js_url(hass, "/local/my_es5_2.js", True)
text = await get_response()
assert '"/local/my_module_2.js"' not in text
assert '"/local/my_es5_2.js"' not in text
# Remove again should not raise
remove_extra_js_url(hass, "/local/my_module_2.js", False)
remove_extra_js_url(hass, "/local/my_es5_2.js", True)
text = await get_response()
assert '"/local/my_module_2.js"' not in text
assert '"/local/my_es5_2.js"' not in text
# safe mode
hass.config.safe_mode = True
text = await get_response()
assert '"/local/my_module.js"' not in text
assert '"/local/my_es5.js"' not in text
# Test dynamically adding extra javascript
add_extra_js_url(hass, "/local/my_module_2.js", False)
add_extra_js_url(hass, "/local/my_es5_2.js", True)
text = await get_response()
assert '"/local/my_module_2.js"' not in text
assert '"/local/my_es5_2.js"' not in text
async def test_get_panels(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, mock_http_client
) -> None:
"""Test get_panels command."""
events = async_capture_events(hass, EVENT_PANELS_UPDATED)
2019-07-31 19:25:30 +00:00
resp = await mock_http_client.get("/map")
assert resp.status == HTTPStatus.NOT_FOUND
async_register_built_in_panel(
hass, "map", "Map", "mdi:tooltip-account", require_admin=True
2019-07-31 19:25:30 +00:00
)
2019-07-31 19:25:30 +00:00
resp = await mock_http_client.get("/map")
assert resp.status == 200
assert len(events) == 1
client = await hass_ws_client(hass)
2019-07-31 19:25:30 +00:00
await client.send_json({"id": 5, "type": "get_panels"})
msg = await client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"]["map"]["component_name"] == "map"
assert msg["result"]["map"]["url_path"] == "map"
assert msg["result"]["map"]["icon"] == "mdi:tooltip-account"
assert msg["result"]["map"]["title"] == "Map"
assert msg["result"]["map"]["require_admin"] is True
async_remove_panel(hass, "map")
2019-07-31 19:25:30 +00:00
resp = await mock_http_client.get("/map")
assert resp.status == HTTPStatus.NOT_FOUND
assert len(events) == 2
async def test_get_panels_non_admin(
hass: HomeAssistant, ws_client, hass_admin_user: MockUser
) -> None:
"""Test get_panels command."""
hass_admin_user.groups = []
async_register_built_in_panel(
hass, "map", "Map", "mdi:tooltip-account", require_admin=True
2019-07-31 19:25:30 +00:00
)
async_register_built_in_panel(hass, "history", "History", "mdi:history")
await ws_client.send_json({"id": 5, "type": "get_panels"})
msg = await ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert "history" in msg["result"]
assert "map" not in msg["result"]
async def test_get_translations(hass: HomeAssistant, ws_client) -> None:
"""Test get_translations command."""
2019-07-31 19:25:30 +00:00
with patch(
"homeassistant.components.frontend.async_get_translations",
side_effect=lambda hass, lang, category, integrations, config_flow: {
"lang": lang
},
2019-07-31 19:25:30 +00:00
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_translations",
"language": "nl",
"category": "lang",
}
2019-07-31 19:25:30 +00:00
)
msg = await ws_client.receive_json()
2019-07-31 19:25:30 +00:00
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"] == {"resources": {"lang": "nl"}}
async def test_get_translations_for_integrations(
hass: HomeAssistant, ws_client
) -> None:
"""Test get_translations for integrations command."""
with patch(
"homeassistant.components.frontend.async_get_translations",
side_effect=lambda hass, lang, category, integration, config_flow: {
"lang": lang,
"integration": integration,
},
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_translations",
"integration": ["frontend", "http"],
"language": "nl",
"category": "lang",
}
)
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert set(msg["result"]["resources"]["integration"]) == {"frontend", "http"}
async def test_get_translations_for_single_integration(
hass: HomeAssistant, ws_client
) -> None:
"""Test get_translations for integration command."""
with patch(
"homeassistant.components.frontend.async_get_translations",
side_effect=lambda hass, lang, category, integrations, config_flow: {
"lang": lang,
"integration": integrations,
},
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_translations",
"integration": "http",
"language": "nl",
"category": "lang",
}
)
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"] == {"resources": {"lang": "nl", "integration": ["http"]}}
async def test_auth_load(hass: HomeAssistant) -> None:
"""Test auth component loaded by default."""
frontend = await async_get_integration(hass, "frontend")
assert "auth" in frontend.dependencies
async def test_onboarding_load(hass: HomeAssistant) -> None:
"""Test onboarding component loaded by default."""
frontend = await async_get_integration(hass, "frontend")
assert "onboarding" in frontend.dependencies
2018-08-09 07:27:54 +00:00
async def test_auth_authorize(mock_http_client) -> None:
2018-08-09 07:27:54 +00:00
"""Test the authorize endpoint works."""
resp = await mock_http_client.get(
2019-07-31 19:25:30 +00:00
"/auth/authorize?response_type=code&client_id=https://localhost/&"
"redirect_uri=https://localhost/&state=123%23456"
)
assert resp.status == 200
# No caching of auth page.
2019-07-31 19:25:30 +00:00
assert "cache-control" not in resp.headers
2018-08-09 07:27:54 +00:00
text = await resp.text()
# Test we can retrieve authorize.js
authorizejs = re.search(
2023-04-04 04:11:20 +00:00
r"(?P<app>\/frontend_latest\/authorize.[A-Za-z0-9_-]{11}.js)", text
2019-07-31 19:25:30 +00:00
)
assert authorizejs is not None, text
resp = await mock_http_client.get(authorizejs.groups(0)[0])
assert resp.status == 200
2019-07-31 19:25:30 +00:00
assert "public" in resp.headers.get("cache-control")
async def test_get_version(hass: HomeAssistant, ws_client) -> None:
"""Test get_version command."""
frontend = await async_get_integration(hass, "frontend")
cur_version = next(
req.split("==", 1)[1]
for req in frontend.requirements
if req.startswith("home-assistant-frontend==")
)
await ws_client.send_json({"id": 5, "type": "frontend/get_version"})
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"] == {"version": cur_version}
async def test_static_paths(hass: HomeAssistant, mock_http_client) -> None:
"""Test static paths."""
resp = await mock_http_client.get(
"/.well-known/change-password", allow_redirects=False
)
assert resp.status == 302
assert resp.headers["location"] == "/profile"
async def test_manifest_json(
hass: HomeAssistant, frontend_themes, mock_http_client
) -> None:
"""Test for fetching manifest.json."""
resp = await mock_http_client.get("/manifest.json")
assert resp.status == HTTPStatus.OK
assert "cache-control" not in resp.headers
json = await resp.json()
assert json["theme_color"] == DEFAULT_THEME_COLOR
await hass.services.async_call(
DOMAIN, "set_theme", {"name": "happy"}, blocking=True
)
await hass.async_block_till_done()
resp = await mock_http_client.get("/manifest.json")
assert resp.status == HTTPStatus.OK
assert "cache-control" not in resp.headers
json = await resp.json()
assert json["theme_color"] != DEFAULT_THEME_COLOR
async def test_static_path_cache(hass: HomeAssistant, mock_http_client) -> None:
"""Test static paths cache."""
resp = await mock_http_client.get("/lovelace/default_view", allow_redirects=False)
assert resp.status == 404
resp = await mock_http_client.get("/frontend_latest/", allow_redirects=False)
assert resp.status == 403
resp = await mock_http_client.get(
"/static/icons/favicon.ico", allow_redirects=False
)
assert resp.status == 200
# and again to make sure the cache works
resp = await mock_http_client.get(
"/static/icons/favicon.ico", allow_redirects=False
)
assert resp.status == 200
resp = await mock_http_client.get(
"/static/fonts/roboto/Roboto-Bold.woff2", allow_redirects=False
)
assert resp.status == 200
resp = await mock_http_client.get("/static/does-not-exist", allow_redirects=False)
assert resp.status == 404
# and again to make sure the cache works
resp = await mock_http_client.get("/static/does-not-exist", allow_redirects=False)
assert resp.status == 404
async def test_get_icons(hass: HomeAssistant, ws_client: MockHAClientWebSocket) -> None:
"""Test get_icons command."""
with patch(
"homeassistant.components.frontend.async_get_icons",
side_effect=lambda hass, category, integrations: {},
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_icons",
"category": "entity_component",
}
)
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"] == {"resources": {}}
async def test_get_icons_for_integrations(
hass: HomeAssistant, ws_client: MockHAClientWebSocket
) -> None:
"""Test get_icons for integrations command."""
with patch(
"homeassistant.components.frontend.async_get_icons",
side_effect=lambda hass, category, integrations: {
integration: {} for integration in integrations
},
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_icons",
"integration": ["frontend", "http"],
"category": "entity",
}
)
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert set(msg["result"]["resources"]) == {"frontend", "http"}
async def test_get_icons_for_single_integration(
hass: HomeAssistant, ws_client: MockHAClientWebSocket
) -> None:
"""Test get_icons for integration command."""
with patch(
"homeassistant.components.frontend.async_get_icons",
side_effect=lambda hass, category, integrations: {
integration: {} for integration in integrations
},
):
await ws_client.send_json(
{
"id": 5,
"type": "frontend/get_icons",
"integration": "http",
"category": "entity",
}
)
msg = await ws_client.receive_json()
assert msg["id"] == 5
assert msg["type"] == TYPE_RESULT
assert msg["success"]
assert msg["result"] == {"resources": {"http": {}}}
async def test_www_local_dir(
hass: HomeAssistant, tmp_path: Path, hass_client: ClientSessionGenerator
) -> None:
"""Test local www folder."""
hass.config.config_dir = str(tmp_path)
tmp_path_www = tmp_path / "www"
x_txt_file = tmp_path_www / "x.txt"
def _create_www_and_x_txt():
tmp_path_www.mkdir()
x_txt_file.write_text("any")
await hass.async_add_executor_job(_create_www_and_x_txt)
assert await async_setup_component(hass, "frontend", {})
client = await hass_client()
resp = await client.get("/local/x.txt")
assert resp.status == HTTPStatus.OK