import hashlib
import json
import subprocess
import unittest
from pathlib import Path

import generate_task_program_state_report


ROOT = Path(__file__).resolve().parents[1]


class TaskProgramStateReportTests(unittest.TestCase):
    def test_worker_state_machine(self):
        result = subprocess.run(["node", "web/worker/test_task_program_state.mjs"], cwd=ROOT, check=False)
        self.assertEqual(result.returncode, 0)

    def test_report_self_hash(self):
        report = json.loads((ROOT / "coverage/task-program-state.report.json").read_text(encoding="utf-8"))
        self.assertEqual(report["schema"], "task-program-state-report.v1")
        self.assertEqual(report["status"], "verified")
        unsigned = dict(report)
        actual = unsigned.pop("report_sha256")
        self.assertEqual(actual, hashlib.sha256(generate_task_program_state_report.canonical(unsigned).encode("utf-8")).hexdigest())


if __name__ == "__main__":
    unittest.main()