import hashlib
import json
import unittest
from pathlib import Path

import generate_trace_export_report


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


class TraceExportReportTests(unittest.TestCase):
    def test_report(self):
        report = json.loads((ROOT / "coverage/trace-export.report.json").read_text(encoding="utf-8"))
        self.assertEqual(report["schema"], "trace-export-report.v1")
        self.assertEqual(report["status"], "verified")
        self.assertIn("event_hash", report["trace"]["fields"])

    def test_report_self_hash(self):
        report = json.loads((ROOT / "coverage/trace-export.report.json").read_text(encoding="utf-8"))
        unsigned = dict(report)
        actual = unsigned.pop("report_sha256")
        self.assertEqual(actual, hashlib.sha256(generate_trace_export_report.canonical(unsigned).encode("utf-8")).hexdigest())


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