32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
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)
|