feat(update): Firmware-Update aus dem Ordner mit Einmal-URLs und WLAN-Sicherung

tests/test_http.py

@@ -0,0 +1,84 @@
+"""One-time firmware URLs and the address lamps use to reach Home Assistant."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from homeassistant.core import HomeAssistant
+from homeassistant.core_config import async_process_ha_core_config
+from homeassistant.exceptions import HomeAssistantError
+import pytest
+from pytest_homeassistant_custom_component.typing import ClientSessionGenerator
+
+from custom_components.ninlight.const import CONF_LAMP_BASE_URL
+from custom_components.ninlight.http import OtaRegistry, lamp_base_url
+from custom_components.ninlight.shared import DATA_SHARED
+from tests_unit.images import build_image
+
+
+def test_registry_keys_are_unique_and_43_characters() -> None:
+    registry = OtaRegistry()
+    keys = {registry.create(Path("/tmp/a.img")) for _ in range(50)}
+    assert len(keys) == 50
+    assert all(len(key) == 43 for key in keys)
+    assert len(registry) == 50
+
+
+async def test_view_serves_only_registered_keys(
+    hass: HomeAssistant,
+    hass_client_no_auth: ClientSessionGenerator,
+    init_integration,
+    firmware_dir: Path,
+) -> None:
+    path = firmware_dir / "a.img"
+    path.write_bytes(build_image())
+    registry = hass.data[DATA_SHARED].registry
+    other = firmware_dir / "b.img"
+    other.write_bytes(build_image(version="0.3.9-rc1"))
+    key = registry.create(path)
+    client = await hass_client_no_auth()
+
+    response = await client.get(f"/api/ninlight/firmware/{key}")
+    assert response.status == 200
+    assert await response.read() == path.read_bytes()
+
+    assert (await client.get("/api/ninlight/firmware/" + "x" * 43)).status == 404
+    registry.revoke(key)
+    assert (await client.get(f"/api/ninlight/firmware/{key}")).status == 404
+
+
+async def test_view_returns_404_when_file_vanished(
+    hass: HomeAssistant,
+    hass_client_no_auth: ClientSessionGenerator,
+    init_integration,
+    firmware_dir: Path,
+) -> None:
+    key = hass.data[DATA_SHARED].registry.create(firmware_dir / "gone.img")
+    client = await hass_client_no_auth()
+    assert (await client.get(f"/api/ninlight/firmware/{key}")).status == 404
+
+
+async def test_base_url_prefers_option(hass: HomeAssistant) -> None:
+    await async_process_ha_core_config(hass, {"internal_url": "http://192.168.1.60:8123"})
+    assert (
+        lamp_base_url(hass, {CONF_LAMP_BASE_URL: "http://192.168.1.50:8123"})
+        == "http://192.168.1.50:8123"
+    )
+
+
+async def test_base_url_falls_back_to_internal_http_url(hass: HomeAssistant) -> None:
+    await async_process_ha_core_config(hass, {"internal_url": "http://192.168.1.60:8123"})
+    assert lamp_base_url(hass, {}) == "http://192.168.1.60:8123"
+
+
+async def test_base_url_rejects_https(hass: HomeAssistant) -> None:
+    await async_process_ha_core_config(hass, {"internal_url": "https://ha.local:8123"})
+    with pytest.raises(HomeAssistantError) as raised:
+        lamp_base_url(hass, {})
+    assert raised.value.translation_key == "no_http_url"
+
+
+async def test_base_url_without_any_url(hass: HomeAssistant) -> None:
+    with pytest.raises(HomeAssistantError) as raised:
+        lamp_base_url(hass, {})
+    assert raised.value.translation_key == "no_http_url"