import json from edge.cloudclient import CloudClient from edge.alert import sign class _Resp: def __init__(self, code, body=""): self.status_code = code; self.text = body def json(self): return json.loads(self.text) def test_fetch_config_signs_and_parses(): captured = {} def fake_get(url, headers=None, timeout=None): captured["url"] = url; captured["headers"] = headers return _Resp(200, json.dumps({"version": 3, "zones": [], "thresholds": {}})) c = CloudClient("https://x", "pi-1", "sek", nonce_fn=lambda: "n1", now_fn=lambda: 1000.0, http_get=fake_get) cfg = c.fetch_config() assert cfg["version"] == 3 assert "device_id=pi-1" in captured["url"] assert captured["headers"]["X-Signature"] == sign("sek", "n1", "1000.0", "") def test_fetch_config_none_on_error(): c = CloudClient("https://x", "pi-1", "sek", lambda: "n", lambda: 1.0, http_get=lambda *a, **k: _Resp(401)) assert c.fetch_config() is None def test_send_heartbeat_signs_body(): captured = {} def fake_post(url, data=None, headers=None, timeout=None): captured["data"] = data; captured["headers"] = headers; return _Resp(200) c = CloudClient("https://x", "pi-1", "sek", lambda: "n1", lambda: 1000.0, http_post=fake_post) assert c.send_heartbeat({"fps": 12.0}) is True body = captured["data"] assert json.loads(body)["device_id"] == "pi-1" assert captured["headers"]["X-Signature"] == sign("sek", "n1", "1000.0", body)