from __future__ import annotations

import json
import tempfile
import unittest
from pathlib import Path

from dukaan_saathi.traceability import build_file_ref, file_sha256, jsonl_count, write_manifest


class TraceabilityTest(unittest.TestCase):
    def test_file_ref_includes_hash_and_jsonl_count(self) -> None:
        path = Path(tempfile.gettempdir()) / "dukaan_saathi_traceability_test.jsonl"
        path.write_text('{"x": 1}\n\n{"x": 2}\n', encoding="utf-8")

        ref = build_file_ref(path)

        self.assertEqual(ref["path"], str(path))
        self.assertEqual(ref["records"], 2)
        self.assertEqual(ref["sha256"], file_sha256(path))
        self.assertEqual(jsonl_count(path), 2)

    def test_write_manifest_adds_run_id_and_created_at(self) -> None:
        trace_dir = Path(tempfile.gettempdir()) / "dukaan_saathi_traceability_runs"
        path = write_manifest(
            {
                "run_id": "trace-test",
                "kind": "unit-test",
                "status": "succeeded",
            },
            trace_dir=trace_dir,
        )

        payload = json.loads(path.read_text(encoding="utf-8"))

        self.assertEqual(path.name, "trace-test.json")
        self.assertEqual(payload["kind"], "unit-test")
        self.assertEqual(payload["status"], "succeeded")
        self.assertIn("created_at", payload)


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