mirror of
https://github.com/home-assistant/core.git
synced 2026-09-17 14:08:42 +01:00
128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
"""Test the Lovelace initialization."""
|
|
|
|
from collections.abc import Generator
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import probatio
|
|
import pytest
|
|
|
|
from homeassistant.components.lovelace import DOMAIN, _validate_url_slug
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.typing import WebSocketGenerator
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_onboarding_not_done() -> Generator[MagicMock]:
|
|
"""Mock that Home Assistant is currently onboarding."""
|
|
with patch(
|
|
"homeassistant.components.onboarding.async_is_onboarded",
|
|
return_value=False,
|
|
) as mock_onboarding:
|
|
yield mock_onboarding
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_onboarding_done() -> Generator[MagicMock]:
|
|
"""Mock that Home Assistant is currently onboarding."""
|
|
with patch(
|
|
"homeassistant.components.onboarding.async_is_onboarded",
|
|
return_value=True,
|
|
) as mock_onboarding:
|
|
yield mock_onboarding
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_add_onboarding_listener() -> Generator[MagicMock]:
|
|
"""Mock that Home Assistant is currently onboarding."""
|
|
with patch(
|
|
"homeassistant.components.onboarding.async_add_listener",
|
|
) as mock_add_onboarding_listener:
|
|
yield mock_add_onboarding_listener
|
|
|
|
|
|
async def test_create_dashboards_when_onboarded(
|
|
hass: HomeAssistant,
|
|
hass_ws_client: WebSocketGenerator,
|
|
hass_storage: dict[str, Any],
|
|
mock_onboarding_done,
|
|
) -> None:
|
|
"""Test we don't create dashboards when onboarded."""
|
|
client = await hass_ws_client(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
|
|
# List dashboards
|
|
await client.send_json_auto_id({"type": "lovelace/dashboards/list"})
|
|
response = await client.receive_json()
|
|
assert response["success"]
|
|
assert response["result"] == []
|
|
|
|
|
|
async def test_create_dashboards_when_not_onboarded(
|
|
hass: HomeAssistant,
|
|
hass_ws_client: WebSocketGenerator,
|
|
hass_storage: dict[str, Any],
|
|
mock_add_onboarding_listener,
|
|
mock_onboarding_not_done,
|
|
) -> None:
|
|
"""Test we automatically create dashboards when not onboarded."""
|
|
client = await hass_ws_client(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
|
|
# Call onboarding listener
|
|
mock_add_onboarding_listener.mock_calls[0][1][1]()
|
|
await hass.async_block_till_done()
|
|
|
|
# List dashboards
|
|
await client.send_json_auto_id({"type": "lovelace/dashboards/list"})
|
|
response = await client.receive_json()
|
|
assert response["success"]
|
|
assert response["result"] == [
|
|
{
|
|
"icon": "mdi:map",
|
|
"id": "map",
|
|
"mode": "storage",
|
|
"require_admin": False,
|
|
"show_in_sidebar": True,
|
|
"title": "Map",
|
|
"url_path": "map",
|
|
}
|
|
]
|
|
|
|
# List map dashboard config
|
|
await client.send_json_auto_id({"type": "lovelace/config", "url_path": "map"})
|
|
response = await client.receive_json()
|
|
assert response["success"]
|
|
assert response["result"] == {"strategy": {"type": "map"}}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("value", "expected"),
|
|
[
|
|
("lovelace", "lovelace"),
|
|
("my-dashboard", "my-dashboard"),
|
|
("my-cool-dashboard", "my-cool-dashboard"),
|
|
],
|
|
)
|
|
def test_validate_url_slug_valid(value: str, expected: str) -> None:
|
|
"""Test _validate_url_slug with valid values."""
|
|
assert _validate_url_slug(value) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("value", "error_message"),
|
|
[
|
|
(None, r"Slug should not be None"),
|
|
("nodash", r"Url path needs to contain a hyphen \(-\)"),
|
|
("my-dash board", r"invalid slug my-dash board \(try my-dash-board\)"),
|
|
],
|
|
)
|
|
def test_validate_url_slug_invalid(value: Any, error_message: str) -> None:
|
|
"""Test _validate_url_slug with invalid values."""
|
|
with pytest.raises(probatio.Invalid, match=error_message):
|
|
_validate_url_slug(value)
|