Compare commits
5
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b6a2b59d1c | ||
|
|
7f085b140d | ||
|
|
ec0fc29160 | ||
|
|
3b9ffa383f | ||
|
|
78f783cfad |
@@ -13,11 +13,15 @@ NestQuest helps households coordinate daily life, chore tracking, custody schedu
|
|||||||
1. Ensure [HACS (Home Assistant Community Store)](https://hacs.xyz/) is installed.
|
1. Ensure [HACS (Home Assistant Community Store)](https://hacs.xyz/) is installed.
|
||||||
2. In Home Assistant, open **HACS** > **Integrations**.
|
2. In Home Assistant, open **HACS** > **Integrations**.
|
||||||
3. Select the three-dots menu in the top right and click **Custom repositories**.
|
3. Select the three-dots menu in the top right and click **Custom repositories**.
|
||||||
4. Enter the repository URL: `https://code.cubecraftlabs.com/Maniacs_Mansion/NestQuest`
|
4. Enter the repository URL: `https://github.com/talon2king/NestQuest`
|
||||||
5. Select category **Integration** and click **Add**.
|
5. Select category **Integration** and click **Add**.
|
||||||
6. Search for **NestQuest** and click **Download**.
|
6. Search for **NestQuest** and click **Download**.
|
||||||
7. Restart Home Assistant.
|
7. Restart Home Assistant.
|
||||||
|
|
||||||
|
> **Note:** Development happens in the private Gitea repository
|
||||||
|
> (`https://code.cubecraftlabs.com/Maniacs_Mansion/NestQuest`). The GitHub
|
||||||
|
> repository is a public mirror maintained for HACS distribution.
|
||||||
|
|
||||||
### Manual Installation
|
### Manual Installation
|
||||||
|
|
||||||
1. Download the latest release from the repository.
|
1. Download the latest release from the repository.
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
"""The NestQuest integration."""
|
"""The NestQuest integration."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import inspect
|
||||||
|
from collections.abc import Callable
|
||||||
|
from dataclasses import dataclass
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
@@ -11,6 +14,15 @@ from .const import DOMAIN, LOGGER
|
|||||||
_LOGGER = LOGGER
|
_LOGGER = LOGGER
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class NestQuestRuntimeData:
|
||||||
|
"""Runtime data stored on a NestQuest config entry."""
|
||||||
|
|
||||||
|
entry_id: str
|
||||||
|
options: dict[str, Any]
|
||||||
|
remove_update_listener: Callable[[], Any]
|
||||||
|
|
||||||
|
|
||||||
async def async_setup(hass: HomeAssistant, config: dict[str, Any]) -> bool:
|
async def async_setup(hass: HomeAssistant, config: dict[str, Any]) -> bool:
|
||||||
"""Set up the NestQuest integration. YAML configuration is not used, returns True."""
|
"""Set up the NestQuest integration. YAML configuration is not used, returns True."""
|
||||||
return True
|
return True
|
||||||
@@ -19,9 +31,34 @@ async def async_setup(hass: HomeAssistant, config: dict[str, Any]) -> bool:
|
|||||||
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||||
"""Set up NestQuest from a config entry."""
|
"""Set up NestQuest from a config entry."""
|
||||||
hass.data.setdefault(DOMAIN, {})
|
hass.data.setdefault(DOMAIN, {})
|
||||||
|
|
||||||
|
async def _async_update_listener(
|
||||||
|
listener_hass: HomeAssistant, listener_entry: ConfigEntry
|
||||||
|
) -> None:
|
||||||
|
"""Reload the entry when its options change."""
|
||||||
|
await listener_hass.config_entries.async_reload(listener_entry.entry_id)
|
||||||
|
|
||||||
|
runtime_data = NestQuestRuntimeData(
|
||||||
|
entry_id=entry.entry_id,
|
||||||
|
options=dict(entry.options),
|
||||||
|
remove_update_listener=entry.add_update_listener(_async_update_listener),
|
||||||
|
)
|
||||||
|
entry.runtime_data = runtime_data
|
||||||
|
hass.data[DOMAIN][entry.entry_id] = runtime_data
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||||
"""Unload a config entry."""
|
"""Unload a config entry."""
|
||||||
|
runtime_data = hass.data.setdefault(DOMAIN, {}).pop(entry.entry_id, None)
|
||||||
|
if runtime_data is None:
|
||||||
|
runtime_data = getattr(entry, "runtime_data", None)
|
||||||
|
if runtime_data is not None:
|
||||||
|
remove_update_listener = getattr(runtime_data, "remove_update_listener", None)
|
||||||
|
if remove_update_listener is not None:
|
||||||
|
result = remove_update_listener()
|
||||||
|
if inspect.isawaitable(result):
|
||||||
|
await result
|
||||||
|
if getattr(entry, "runtime_data", None) is not None:
|
||||||
|
entry.runtime_data = None
|
||||||
return True
|
return True
|
||||||
|
|||||||
@@ -0,0 +1,20 @@
|
|||||||
|
"""SQLite storage path resolution for the NestQuest integration."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
|
from .const import SQLITE_DB_FILENAME
|
||||||
|
|
||||||
|
|
||||||
|
def _ensure_dir(path: Path) -> None:
|
||||||
|
"""Create the given directory and any missing parents."""
|
||||||
|
path.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
|
||||||
|
async def async_get_db_path(hass: HomeAssistant) -> Path:
|
||||||
|
"""Return the absolute path to the NestQuest SQLite database file."""
|
||||||
|
db_path = Path(hass.config.path(SQLITE_DB_FILENAME)).absolute()
|
||||||
|
await hass.async_add_executor_job(_ensure_dir, db_path.parent)
|
||||||
|
return db_path
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
{
|
{
|
||||||
"name": "NestQuest",
|
"name": "NestQuest",
|
||||||
"homeassistant": "2024.1.0",
|
"homeassistant": "2024.6.0",
|
||||||
"render_readme": true
|
"render_readme": true
|
||||||
}
|
}
|
||||||
|
|||||||
+17
-1
@@ -25,10 +25,16 @@ def test_hacs_json_exists_and_is_valid(hacs_config: dict) -> None:
|
|||||||
"""Verify hacs.json exists and is valid JSON."""
|
"""Verify hacs.json exists and is valid JSON."""
|
||||||
assert isinstance(hacs_config, dict)
|
assert isinstance(hacs_config, dict)
|
||||||
assert hacs_config.get("name") == "NestQuest"
|
assert hacs_config.get("name") == "NestQuest"
|
||||||
assert hacs_config.get("homeassistant") == "2024.1.0"
|
assert hacs_config.get("homeassistant") == "2024.6.0"
|
||||||
assert hacs_config.get("render_readme") is True
|
assert hacs_config.get("render_readme") is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_hacs_json_minimum_version_supports_runtime_data(hacs_config: dict) -> None:
|
||||||
|
"""Advertised HA floor must support ConfigEntry.runtime_data (added in 2024.6)."""
|
||||||
|
advertised = tuple(int(part) for part in hacs_config.get("homeassistant", "").split("."))
|
||||||
|
assert advertised >= (2024, 6, 0)
|
||||||
|
|
||||||
|
|
||||||
def test_license_exists_and_non_empty() -> None:
|
def test_license_exists_and_non_empty() -> None:
|
||||||
"""Verify LICENSE file exists and is non-empty."""
|
"""Verify LICENSE file exists and is non-empty."""
|
||||||
assert LICENSE_PATH.exists(), f"LICENSE file does not exist at {LICENSE_PATH}"
|
assert LICENSE_PATH.exists(), f"LICENSE file does not exist at {LICENSE_PATH}"
|
||||||
@@ -58,6 +64,16 @@ def test_readme_exists_and_contains_required_sections() -> None:
|
|||||||
assert "uv run pytest" in content
|
assert "uv run pytest" in content
|
||||||
|
|
||||||
|
|
||||||
|
def test_readme_hacs_points_at_public_github_mirror() -> None:
|
||||||
|
"""Verify README directs HACS users to the public GitHub mirror, not the private Gitea repo."""
|
||||||
|
content = README_PATH.read_text(encoding="utf-8")
|
||||||
|
assert "https://github.com/talon2king/NestQuest" in content
|
||||||
|
assert "https://code.cubecraftlabs.com/Maniacs_Mansion/NestQuest" in content
|
||||||
|
assert content.index("https://github.com/talon2king/NestQuest") < content.index(
|
||||||
|
"https://code.cubecraftlabs.com/Maniacs_Mansion/NestQuest"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_hacs_directory_structure() -> None:
|
def test_hacs_directory_structure() -> None:
|
||||||
"""Verify directory structure matches HACS integration expectations."""
|
"""Verify directory structure matches HACS integration expectations."""
|
||||||
assert MANIFEST_PATH.exists(), f"Integration manifest does not exist at {MANIFEST_PATH}"
|
assert MANIFEST_PATH.exists(), f"Integration manifest does not exist at {MANIFEST_PATH}"
|
||||||
|
|||||||
@@ -0,0 +1,250 @@
|
|||||||
|
"""Tests for the setup/unload lifecycle of NestQuest."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import inspect
|
||||||
|
import sys
|
||||||
|
from types import SimpleNamespace
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
|
# Mock homeassistant modules if not installed in environment
|
||||||
|
for _mod in (
|
||||||
|
"homeassistant",
|
||||||
|
"homeassistant.core",
|
||||||
|
"homeassistant.config_entries",
|
||||||
|
):
|
||||||
|
sys.modules.setdefault(_mod, MagicMock())
|
||||||
|
|
||||||
|
# Wire parent package attributes to the actual module mocks so that
|
||||||
|
# "from homeassistant import config_entries" resolves to the mocked module.
|
||||||
|
sys.modules["homeassistant"].config_entries = sys.modules["homeassistant.config_entries"]
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from custom_components.nestquest import DOMAIN, async_setup_entry, async_unload_entry
|
||||||
|
from custom_components.nestquest.const import DOMAIN as DOMAIN_CONST
|
||||||
|
|
||||||
|
|
||||||
|
def _run(coro):
|
||||||
|
return asyncio.new_event_loop().run_until_complete(coro)
|
||||||
|
|
||||||
|
|
||||||
|
class _ListenerRegistry:
|
||||||
|
"""Models HA's update-listener bookkeeping: appends to a list per entry, with per-listener removers."""
|
||||||
|
|
||||||
|
def __init__(self, hass=None):
|
||||||
|
self._hass = hass
|
||||||
|
self._listeners: dict[str, list[object]] = {}
|
||||||
|
self.reloaded: list[str] = []
|
||||||
|
self._loop = asyncio.new_event_loop()
|
||||||
|
|
||||||
|
def add(self, entry_id: str, listener) -> object:
|
||||||
|
def _remove():
|
||||||
|
self._listeners.get(entry_id, []).remove(listener)
|
||||||
|
|
||||||
|
self._listeners.setdefault(entry_id, []).append(listener)
|
||||||
|
return _remove
|
||||||
|
|
||||||
|
def dispatch_options_update(self, entry) -> None:
|
||||||
|
for listener in list(self._listeners.get(entry.entry_id, [])):
|
||||||
|
result = listener(self._hass, entry)
|
||||||
|
if inspect.isawaitable(result):
|
||||||
|
self._loop.run_until_complete(result)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def size(self) -> int:
|
||||||
|
return sum(len(listeners) for listeners in self._listeners.values())
|
||||||
|
|
||||||
|
|
||||||
|
def _make_hass():
|
||||||
|
hass = SimpleNamespace(
|
||||||
|
data={},
|
||||||
|
config_entries=SimpleNamespace(
|
||||||
|
async_reload=None,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
registry = _ListenerRegistry(hass)
|
||||||
|
async def _async_reload(entry_id: str) -> None:
|
||||||
|
registry.reloaded.append(entry_id)
|
||||||
|
|
||||||
|
hass.config_entries.async_reload = MagicMock(side_effect=_async_reload)
|
||||||
|
return hass, registry
|
||||||
|
|
||||||
|
|
||||||
|
def _make_entry(entry_id: str = "test_entry", options: dict | None = None):
|
||||||
|
entry = SimpleNamespace(entry_id=entry_id, options=dict(options or {}))
|
||||||
|
entry.add_update_listener = None
|
||||||
|
return entry
|
||||||
|
|
||||||
|
|
||||||
|
def _wire_entry(entry, registry: _ListenerRegistry):
|
||||||
|
entry.add_update_listener = lambda listener: registry.add(entry.entry_id, listener)
|
||||||
|
return entry
|
||||||
|
|
||||||
|
|
||||||
|
def test_setup_stores_runtime_data_in_hass_data() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), _ListenerRegistry())
|
||||||
|
assert _run(async_setup_entry(hass, entry)) is True
|
||||||
|
assert entry.entry_id in hass.data[DOMAIN]
|
||||||
|
assert hass.data[DOMAIN][entry.entry_id] is entry.runtime_data
|
||||||
|
|
||||||
|
|
||||||
|
def test_setup_runtime_data_has_entry_id_and_options() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(options={"horizon_days": 7}), _ListenerRegistry())
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
runtime = entry.runtime_data
|
||||||
|
assert runtime.entry_id == entry.entry_id
|
||||||
|
assert runtime.options == {"horizon_days": 7}
|
||||||
|
assert callable(runtime.remove_update_listener)
|
||||||
|
|
||||||
|
|
||||||
|
def test_unload_returns_true_and_removes_hass_data() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
registry = _ListenerRegistry()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
assert _run(async_unload_entry(hass, entry)) is True
|
||||||
|
assert entry.entry_id not in hass.data[DOMAIN]
|
||||||
|
|
||||||
|
|
||||||
|
def test_unload_is_idempotent() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
registry = _ListenerRegistry()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
assert _run(async_unload_entry(hass, entry)) is True
|
||||||
|
assert _run(async_unload_entry(hass, entry)) is True
|
||||||
|
assert entry.entry_id not in hass.data[DOMAIN]
|
||||||
|
|
||||||
|
|
||||||
|
def test_unload_without_setup_returns_true() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
entry = _make_entry()
|
||||||
|
assert _run(async_unload_entry(hass, entry)) is True
|
||||||
|
assert DOMAIN not in hass.data or entry.entry_id not in hass.data[DOMAIN]
|
||||||
|
|
||||||
|
|
||||||
|
def test_unload_clears_entry_runtime_data() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
registry = _ListenerRegistry()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
assert entry.runtime_data is None
|
||||||
|
|
||||||
|
|
||||||
|
def test_unload_removes_update_listener() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
registry = _ListenerRegistry()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
assert registry.size == 1
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
assert registry.size == 0
|
||||||
|
|
||||||
|
|
||||||
|
def test_options_update_triggers_reload_once() -> None:
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
entry.options = {"horizon_days": 30}
|
||||||
|
registry.dispatch_options_update(entry)
|
||||||
|
assert hass.config_entries.async_reload.call_count == 1
|
||||||
|
assert registry.reloaded == [entry.entry_id]
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_reload_after_unload() -> None:
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
registry.dispatch_options_update(entry)
|
||||||
|
assert hass.config_entries.async_reload.call_count == 0
|
||||||
|
|
||||||
|
|
||||||
|
def test_three_reload_cycles_keep_single_listener() -> None:
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
for _ in range(3):
|
||||||
|
assert _run(async_setup_entry(hass, entry)) is True
|
||||||
|
assert registry.size == 1
|
||||||
|
assert _run(async_unload_entry(hass, entry)) is True
|
||||||
|
assert registry.size == 0
|
||||||
|
assert hass.config_entries.async_reload.call_count == 0
|
||||||
|
|
||||||
|
|
||||||
|
def test_three_setup_unload_cycles_leave_no_listeners() -> None:
|
||||||
|
"""Listeners accumulate per entry (HA semantics); unload must remove them all."""
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
for _ in range(3):
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
assert registry._listeners.get(entry.entry_id, []) == []
|
||||||
|
assert registry.size == 0
|
||||||
|
|
||||||
|
|
||||||
|
def test_options_update_after_each_setup_invokes_exactly_one_listener() -> None:
|
||||||
|
"""After each setup, an options update dispatches to exactly one listener."""
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
for _ in range(3):
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
registry.dispatch_options_update(entry)
|
||||||
|
assert hass.config_entries.async_reload.call_count == 1
|
||||||
|
assert registry.reloaded == [entry.entry_id]
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
hass.config_entries.async_reload.reset_mock()
|
||||||
|
registry.reloaded.clear()
|
||||||
|
|
||||||
|
|
||||||
|
def test_reload_setup_cycles_dispatch_reload_each_time() -> None:
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
for _ in range(3):
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
registry.dispatch_options_update(entry)
|
||||||
|
_run(async_unload_entry(hass, entry))
|
||||||
|
assert hass.config_entries.async_reload.call_count == 3
|
||||||
|
assert registry.size == 0
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_orphaned_tasks_after_unload() -> None:
|
||||||
|
async def _scenario():
|
||||||
|
hass, registry = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), registry)
|
||||||
|
await async_setup_entry(hass, entry)
|
||||||
|
await async_unload_entry(hass, entry)
|
||||||
|
pending = [task for task in asyncio.all_tasks() if task is not asyncio.current_task()]
|
||||||
|
assert pending == []
|
||||||
|
|
||||||
|
_run(_scenario())
|
||||||
|
|
||||||
|
|
||||||
|
def test_setup_entry_uses_domain_constant() -> None:
|
||||||
|
hass, _ = _make_hass()
|
||||||
|
entry = _wire_entry(_make_entry(), _ListenerRegistry())
|
||||||
|
_run(async_setup_entry(hass, entry))
|
||||||
|
assert set(hass.data) == {DOMAIN_CONST}
|
||||||
|
|
||||||
|
|
||||||
|
def test_setup_awaitable_listener_removal_supported() -> None:
|
||||||
|
"""A coroutine-returning remover is awaited during unload."""
|
||||||
|
|
||||||
|
async def _scenario():
|
||||||
|
removed = []
|
||||||
|
|
||||||
|
async def _remover():
|
||||||
|
removed.append(True)
|
||||||
|
|
||||||
|
entry = SimpleNamespace(entry_id="awaitable", options={})
|
||||||
|
entry.add_update_listener = lambda listener: _remover
|
||||||
|
hass = SimpleNamespace(data={}, config_entries=SimpleNamespace(async_reload=None))
|
||||||
|
await async_setup_entry(hass, entry)
|
||||||
|
result = await async_unload_entry(hass, entry)
|
||||||
|
assert result is True
|
||||||
|
assert removed == [True]
|
||||||
|
|
||||||
|
_run(_scenario())
|
||||||
@@ -0,0 +1,160 @@
|
|||||||
|
"""Tests for store.py of NestQuest."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import ast
|
||||||
|
import asyncio
|
||||||
|
from pathlib import Path
|
||||||
|
import sys
|
||||||
|
from unittest.mock import AsyncMock, MagicMock
|
||||||
|
|
||||||
|
# Mock homeassistant modules if not installed in environment
|
||||||
|
for _mod in (
|
||||||
|
"homeassistant",
|
||||||
|
"homeassistant.core",
|
||||||
|
"homeassistant.config_entries",
|
||||||
|
):
|
||||||
|
sys.modules.setdefault(_mod, MagicMock())
|
||||||
|
|
||||||
|
from custom_components.nestquest import store
|
||||||
|
from custom_components.nestquest.const import SQLITE_DB_FILENAME
|
||||||
|
|
||||||
|
|
||||||
|
def _run(coro):
|
||||||
|
return asyncio.new_event_loop().run_until_complete(coro)
|
||||||
|
|
||||||
|
|
||||||
|
def _make_hass(config_dir: Path, execute: bool = True):
|
||||||
|
hass = MagicMock()
|
||||||
|
hass.config.path = MagicMock(return_value=str(config_dir / SQLITE_DB_FILENAME))
|
||||||
|
hass.async_add_executor_job = AsyncMock(
|
||||||
|
side_effect=(lambda fn, *a: fn(*a)) if execute else None
|
||||||
|
)
|
||||||
|
return hass
|
||||||
|
|
||||||
|
|
||||||
|
def test_returns_absolute_path_from_config_path() -> None:
|
||||||
|
hass = _make_hass(Path("/config"), execute=False)
|
||||||
|
db_path = _run(store.async_get_db_path(hass))
|
||||||
|
hass.config.path.assert_called_once_with(SQLITE_DB_FILENAME)
|
||||||
|
assert db_path == Path("/config") / SQLITE_DB_FILENAME
|
||||||
|
assert db_path.is_absolute()
|
||||||
|
|
||||||
|
|
||||||
|
def test_relative_config_path_is_normalized_to_absolute(tmp_path, monkeypatch) -> None:
|
||||||
|
monkeypatch.chdir(tmp_path)
|
||||||
|
hass = MagicMock()
|
||||||
|
hass.config.path = MagicMock(return_value="relative-config/" + SQLITE_DB_FILENAME)
|
||||||
|
hass.async_add_executor_job = AsyncMock(side_effect=(lambda fn, *a: fn(*a)))
|
||||||
|
db_path = _run(store.async_get_db_path(hass))
|
||||||
|
assert db_path.is_absolute()
|
||||||
|
assert db_path.name == SQLITE_DB_FILENAME
|
||||||
|
assert str(db_path).endswith("relative-config/" + SQLITE_DB_FILENAME)
|
||||||
|
assert db_path.parent.is_dir()
|
||||||
|
assert (tmp_path / "relative-config").is_dir()
|
||||||
|
assert not db_path.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_creates_missing_parent_directory(tmp_path) -> None:
|
||||||
|
nested = tmp_path / "homeassistant" / "config"
|
||||||
|
hass = _make_hass(nested)
|
||||||
|
db_path = _run(store.async_get_db_path(hass))
|
||||||
|
assert db_path.parent == nested
|
||||||
|
assert nested.is_dir()
|
||||||
|
assert db_path.parent.is_dir()
|
||||||
|
assert not db_path.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_two_calls_return_same_path(tmp_path) -> None:
|
||||||
|
hass = _make_hass(tmp_path)
|
||||||
|
first = _run(store.async_get_db_path(hass))
|
||||||
|
second = _run(store.async_get_db_path(hass))
|
||||||
|
assert first == second
|
||||||
|
assert first.parent.is_dir()
|
||||||
|
assert not first.exists()
|
||||||
|
assert not second.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_existing_directory_does_not_fail(tmp_path) -> None:
|
||||||
|
nested = tmp_path / "config"
|
||||||
|
nested.mkdir()
|
||||||
|
hass = _make_hass(nested)
|
||||||
|
db_path = _run(store.async_get_db_path(hass))
|
||||||
|
assert db_path == nested / SQLITE_DB_FILENAME
|
||||||
|
assert db_path.parent.is_dir()
|
||||||
|
assert not db_path.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_mkdir_runs_via_executor(tmp_path) -> None:
|
||||||
|
nested = tmp_path / "deep" / "nested"
|
||||||
|
hass = _make_hass(nested)
|
||||||
|
db_path = _run(store.async_get_db_path(hass))
|
||||||
|
assert nested.is_dir()
|
||||||
|
hass.async_add_executor_job.assert_called_once()
|
||||||
|
args, kwargs = hass.async_add_executor_job.call_args
|
||||||
|
fn, fn_args = args[0], args[1:]
|
||||||
|
assert callable(fn)
|
||||||
|
assert fn is store._ensure_dir
|
||||||
|
assert fn_args == (nested,)
|
||||||
|
assert kwargs == {}
|
||||||
|
assert db_path.parent == nested
|
||||||
|
assert db_path.parent.is_dir()
|
||||||
|
assert not db_path.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_executor_helper_creates_directory(tmp_path) -> None:
|
||||||
|
nested = tmp_path / "executor" / "made"
|
||||||
|
store._ensure_dir(nested)
|
||||||
|
assert nested.is_dir()
|
||||||
|
store._ensure_dir(nested)
|
||||||
|
assert nested.is_dir()
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_sqlite3_import_in_package() -> None:
|
||||||
|
pkg_dir = Path(store.__file__).parent
|
||||||
|
py_files = list(pkg_dir.glob("*.py"))
|
||||||
|
assert py_files
|
||||||
|
for py_file in py_files:
|
||||||
|
tree = ast.parse(py_file.read_text(encoding="utf-8"), filename=str(py_file))
|
||||||
|
for node in ast.walk(tree):
|
||||||
|
if isinstance(node, ast.Import):
|
||||||
|
for alias in node.names:
|
||||||
|
assert alias.name != "sqlite3", (
|
||||||
|
f"Found sqlite3 import in {py_file.name}:{node.lineno}"
|
||||||
|
)
|
||||||
|
elif isinstance(node, ast.ImportFrom) and node.module == "sqlite3":
|
||||||
|
raise AssertionError(
|
||||||
|
f"Found sqlite3 import in {py_file.name}:{node.lineno}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_file_creation_calls_in_store() -> None:
|
||||||
|
source = Path(store.__file__).read_text(encoding="utf-8")
|
||||||
|
tree = ast.parse(source, filename=str(store.__file__))
|
||||||
|
for node in ast.walk(tree):
|
||||||
|
if isinstance(node, ast.Call):
|
||||||
|
func = node.func
|
||||||
|
if isinstance(func, ast.Name) and func.id == "open":
|
||||||
|
raise AssertionError(
|
||||||
|
f"Found open() call in store.py:{node.lineno}"
|
||||||
|
)
|
||||||
|
if (
|
||||||
|
isinstance(func, ast.Attribute)
|
||||||
|
and func.attr == "touch"
|
||||||
|
):
|
||||||
|
raise AssertionError(
|
||||||
|
f"Found touch() call in store.py:{node.lineno}"
|
||||||
|
)
|
||||||
|
assert "touch(" not in source
|
||||||
|
assert "open(" not in source
|
||||||
|
|
||||||
|
|
||||||
|
def test_container_style_path_resolves() -> None:
|
||||||
|
hass = _make_hass(Path("/config"), execute=False)
|
||||||
|
assert _run(store.async_get_db_path(hass)) == Path("/config") / SQLITE_DB_FILENAME
|
||||||
|
|
||||||
|
|
||||||
|
def test_core_style_path_resolves(tmp_path) -> None:
|
||||||
|
config_dir = tmp_path / ".homeassistant"
|
||||||
|
hass = _make_hass(config_dir)
|
||||||
|
assert _run(store.async_get_db_path(hass)) == config_dir / SQLITE_DB_FILENAME
|
||||||
|
assert (config_dir / SQLITE_DB_FILENAME).parent.is_dir()
|
||||||
Reference in New Issue
Block a user