import hashlib
import json
import unittest
from pathlib import Path

import generate_offline_migration_report


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


class OfflineMigrationReportTests(unittest.TestCase):
    def test_report(self):
        report = json.loads((ROOT / "coverage/offline-migration.report.json").read_text(encoding="utf-8"))
        self.assertEqual(report["schema"], "offline-migration-report.v1")
        self.assertEqual(report["status"], "verified")
        self.assertTrue(report["packages"]["migration_lossless"])

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


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