piphi-runtime-testkit-python 0.1.1__tar.gz → 0.1.2__tar.gz

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: piphi-runtime-testkit-python
-Version: 0.1.1
+Version: 0.1.2
 Summary: Pytest-first helpers for testing PiPhi runtime integrations
 Keywords: piphi,pytest,testing,runtime,iot
 Author-Email: KelvinSan <support@piphi.network>

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "piphi-runtime-testkit-python"
-version = "0.1.1"
+version = "0.1.2"
 description = "Pytest-first helpers for testing PiPhi runtime integrations"
 authors = [
     { name = "KelvinSan", email = "support@piphi.network" },
@@ -55,7 +55,7 @@ release-version = "python scripts/release.py"
 dev = [
     "fastapi>=0.115,<1.0",
     "httpx>=0.27,<1.0",
-    "piphi-runtime-kit-python[fastapi]==0.4.2",
+    "piphi-runtime-kit-python[fastapi]==0.4.5",
     "pytest>=8.3,<9.0",
     "pytest-asyncio>=0.24,<1.0",
     "twine>=6.2.0",

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/src/piphi_runtime_testkit_python/__init__.py

@@ -1,16 +1,23 @@
 """Pytest-first helpers for testing PiPhi runtime integrations."""
 
 from .assertions import assert_entities_response, assert_event_sent, assert_telemetry_sent
-from .builders import build_config_payload, build_config_snapshot, build_runtime_headers
-from .mock_core import CapturedRequest, MockCoreServer
+from .builders import (
+    build_config_payload,
+    build_config_snapshot,
+    build_core_config_row,
+    build_runtime_headers,
+)
+from .mock_core import CapturedRequest, MockCoreServer, RUNTIME_CONFIG_FETCH_PATH
 
 __all__ = [
     "CapturedRequest",
     "MockCoreServer",
+    "RUNTIME_CONFIG_FETCH_PATH",
     "assert_entities_response",
     "assert_event_sent",
     "assert_telemetry_sent",
     "build_config_payload",
     "build_config_snapshot",
+    "build_core_config_row",
     "build_runtime_headers",
 ]

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/src/piphi_runtime_testkit_python/builders.py

@@ -50,16 +50,55 @@ def build_config_snapshot(
     container_id: str = "test-container",
     integration_id: str = "test-integration",
     generation: int = 1,
+    deleted_config_ids: list[str] | None = None,
+    config_hash: str | None = None,
+    schema_version: int = 1,
+    updated_at: str | None = None,
     extra: dict[str, Any] | None = None,
 ) -> dict[str, Any]:
     """Build a snapshot payload for `/config/sync` style tests."""
 
     payload: dict[str, Any] = {
+        "schema_version": schema_version,
         "container_id": container_id,
         "integration_id": integration_id,
         "generation": generation,
         "configs": configs or [],
+        "deleted_config_ids": deleted_config_ids or [],
     }
+    if config_hash is not None:
+        payload["config_hash"] = config_hash
+    if updated_at is not None:
+        payload["updated_at"] = updated_at
     if extra:
         payload.update(extra)
     return payload
+
+
+def build_core_config_row(
+    *,
+    config_data: dict[str, Any] | None = None,
+    config_id: str = "config-1",
+    container_id: str = "test-container",
+    integration_id: str = "test-integration",
+    generation: int = 1,
+    extra: dict[str, Any] | None = None,
+) -> dict[str, Any]:
+    """Build a Core database-shaped config row for startup rehydrate tests."""
+
+    payload = build_config_payload(
+        config_id=config_id,
+        container_id=container_id,
+        integration_id=integration_id,
+        extra=config_data,
+    )
+    row: dict[str, Any] = {
+        "id": config_id,
+        "container_id": container_id,
+        "integration_id": integration_id,
+        "generation": generation,
+        "config_data": payload,
+    }
+    if extra:
+        row.update(extra)
+    return row

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/src/piphi_runtime_testkit_python/mock_core.py

@@ -9,6 +9,7 @@ from typing import Any
 
 TELEMETRY_PATH = "/api/v2/integrations/telemetry"
 EVENT_PATH = "/api/v2/events/ingest"
+RUNTIME_CONFIG_FETCH_PATH = "/api/v2/integrations/config/fetch/all/by/container/internal"
 
 
 @dataclass(slots=True)
@@ -24,23 +25,48 @@ class _RequestStore:
     def __init__(self) -> None:
         self.telemetry_requests: list[CapturedRequest] = []
         self.event_requests: list[CapturedRequest] = []
+        self.runtime_config_requests: list[CapturedRequest] = []
         self.telemetry_status = 200
         self.event_status = 200
+        self.runtime_config_status = 200
         self.telemetry_json: Any = {"ok": True}
         self.event_json: Any = {"ok": True}
+        self.runtime_config_json: Any = []
 
     def reset(self) -> None:
         self.telemetry_requests.clear()
         self.event_requests.clear()
+        self.runtime_config_requests.clear()
         self.telemetry_status = 200
         self.event_status = 200
+        self.runtime_config_status = 200
         self.telemetry_json = {"ok": True}
         self.event_json = {"ok": True}
+        self.runtime_config_json = []
 
 
 class _MockCoreRequestHandler(BaseHTTPRequestHandler):
     store: _RequestStore
 
+    def do_GET(self) -> None:  # noqa: N802
+        request = CapturedRequest(
+            method="GET",
+            path=self.path,
+            headers={key: value for key, value in self.headers.items()},
+            body=b"",
+            json_body=None,
+        )
+
+        if self.path.split("?", 1)[0] == RUNTIME_CONFIG_FETCH_PATH:
+            self.store.runtime_config_requests.append(request)
+            self._send_json(
+                self.store.runtime_config_status,
+                self.store.runtime_config_json,
+            )
+            return
+
+        self._send_json(404, {"ok": False, "detail": "unknown path"})
+
     def do_POST(self) -> None:  # noqa: N802
         content_length = int(self.headers.get("Content-Length", "0"))
         body = self.rfile.read(content_length)
@@ -119,6 +145,10 @@ class MockCoreServer:
     def event_requests(self) -> list[CapturedRequest]:
         return self._store.event_requests
 
+    @property
+    def runtime_config_requests(self) -> list[CapturedRequest]:
+        return self._store.runtime_config_requests
+
     def set_telemetry_response(self, *, status_code: int = 200, json_body: Any = None) -> None:
         self._store.telemetry_status = status_code
         self._store.telemetry_json = {"ok": status_code < 400} if json_body is None else json_body
@@ -127,6 +157,13 @@ class MockCoreServer:
         self._store.event_status = status_code
         self._store.event_json = {"ok": status_code < 400} if json_body is None else json_body
 
+    def set_runtime_config_response(self, *, status_code: int = 200, json_body: Any = None) -> None:
+        self._store.runtime_config_status = status_code
+        self._store.runtime_config_json = [] if json_body is None else json_body
+
+    def set_runtime_config_rows(self, rows: list[dict[str, Any]]) -> None:
+        self.set_runtime_config_response(status_code=200, json_body=rows)
+
     def reset(self) -> None:
         self._store.reset()
 

{piphi_runtime_testkit_python-0.1.1 → piphi_runtime_testkit_python-0.1.2}/tests/test_mock_core.py

@@ -4,7 +4,13 @@ import json
 from urllib import error
 from urllib import request
 
-from piphi_runtime_testkit_python.mock_core import EVENT_PATH, TELEMETRY_PATH, MockCoreServer
+from piphi_runtime_testkit_python import build_core_config_row, build_runtime_headers
+from piphi_runtime_testkit_python.mock_core import (
+    EVENT_PATH,
+    RUNTIME_CONFIG_FETCH_PATH,
+    TELEMETRY_PATH,
+    MockCoreServer,
+)
 
 
 def _post_json(url: str, payload: dict) -> tuple[int, dict]:
@@ -19,6 +25,16 @@ def _post_json(url: str, payload: dict) -> tuple[int, dict]:
         return response.status, json.loads(response.read().decode("utf-8"))
 
 
+def _get_json(url: str, headers: dict[str, str] | None = None) -> tuple[int, list]:
+    req = request.Request(
+        url,
+        headers=headers or {},
+        method="GET",
+    )
+    with request.urlopen(req) as response:  # noqa: S310
+        return response.status, json.loads(response.read().decode("utf-8"))
+
+
 def test_mock_core_captures_telemetry_and_events():
     mock_core = MockCoreServer()
     try:
@@ -170,3 +186,40 @@ def test_mock_core_invalid_json_is_captured_with_none_json_body():
         assert mock_core.telemetry_requests[0].body == b"{invalid-json"
     finally:
         mock_core.shutdown()
+
+
+def test_mock_core_serves_runtime_config_rows_and_captures_auth_headers():
+    mock_core = MockCoreServer()
+    try:
+        mock_core.set_runtime_config_rows(
+            [
+                build_core_config_row(
+                    config_id="config-1",
+                    container_id="runtime-123",
+                    integration_id="test-integration",
+                    config_data={"host": "10.0.0.10"},
+                )
+            ]
+        )
+        headers = build_runtime_headers(
+            container_id="runtime-123",
+            internal_token="secret-token",
+        )
+
+        status, rows = _get_json(
+            f"{mock_core.base_url}{RUNTIME_CONFIG_FETCH_PATH}?container_id=runtime-123",
+            headers=headers,
+        )
+
+        assert status == 200
+        assert rows[0]["config_data"]["id"] == "config-1"
+        assert rows[0]["config_data"]["host"] == "10.0.0.10"
+        assert len(mock_core.runtime_config_requests) == 1
+        captured_headers = {
+            key.lower(): value
+            for key, value in mock_core.runtime_config_requests[0].headers.items()
+        }
+        assert captured_headers["x-container-id"] == "runtime-123"
+        assert captured_headers["x-piphi-integration-token"] == "secret-token"
+    finally:
+        mock_core.shutdown()

piphi_runtime_testkit_python-0.1.2/tests/test_runtime_config_contract.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+import httpx
+import pytest
+
+from piphi_runtime_kit_python import (
+    RuntimeConfigSnapshot,
+    RuntimeContext,
+    rehydrate_runtime_configs,
+)
+from piphi_runtime_testkit_python import (
+    build_core_config_row,
+)
+
+
+@pytest.mark.asyncio
+async def test_mock_core_supports_sdk_startup_rehydrate_contract(mock_core) -> None:
+    mock_core.set_runtime_config_rows(
+        [
+            build_core_config_row(
+                config_id="sensor-1",
+                container_id="runtime-123",
+                integration_id="test-integration",
+                config_data={"device_id": "sensor-1", "host": "10.0.0.42"},
+            )
+        ]
+    )
+    runtime = RuntimeContext()
+    runtime.auth.update(container_id="runtime-123", internal_token="secret-token")
+    applied_snapshots: list[RuntimeConfigSnapshot] = []
+
+    async def apply_snapshot(snapshot: RuntimeConfigSnapshot) -> None:
+        applied_snapshots.append(snapshot)
+
+    async with httpx.AsyncClient() as client:
+        result = await rehydrate_runtime_configs(
+            runtime_context=runtime,
+            client=client,
+            apply_snapshot=apply_snapshot,
+            core_base_url=mock_core.base_url,
+        )
+
+    assert result.core_applied is True
+    assert result.core_config_count == 1
+    assert applied_snapshots[0].container_id == "runtime-123"
+    assert applied_snapshots[0].configs[0].id == "sensor-1"
+    assert applied_snapshots[0].configs[0].device_id == "sensor-1"
+
+    captured_request = mock_core.runtime_config_requests[0]
+    captured_headers = {
+        key.lower(): value
+        for key, value in captured_request.headers.items()
+    }
+    assert "container_id=runtime-123" in captured_request.path
+    assert captured_headers["x-container-id"] == "runtime-123"
+    assert captured_headers["x-piphi-integration-token"] == "secret-token"