28 lines
941 B
Python
28 lines
941 B
Python
import hashlib
|
|
import json
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
import generate_worker_protocol_report
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class WorkerProtocolReportTests(unittest.TestCase):
|
|
def test_report(self):
|
|
report = json.loads((ROOT / "coverage/worker-protocol.report.json").read_text(encoding="utf-8"))
|
|
self.assertEqual(report["schema"], "worker-protocol-report.v1")
|
|
self.assertEqual(report["status"], "verified")
|
|
self.assertTrue(report["client"]["cancel"] == "worker_cancelled")
|
|
|
|
def test_report_self_hash(self):
|
|
report = json.loads((ROOT / "coverage/worker-protocol.report.json").read_text(encoding="utf-8"))
|
|
unsigned = dict(report)
|
|
actual = unsigned.pop("report_sha256")
|
|
self.assertEqual(actual, hashlib.sha256(generate_worker_protocol_report.canonical(unsigned).encode("utf-8")).hexdigest())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|