from __future__ import annotations

import json

import subprocess

import sys

import tempfile

import unittest

from pathlib import Path

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

SCRIPT = ROOT / "scripts" / "runner_calibration.py"

EXPECTED_KEYS = {

    "schema_version",

    "runner_id",

    "runner_class",

    "cpu_sha256_mib_per_s",

    "io_write_read_mib_per_s",

    "spawn_ms_median",

    "elapsed_s",

}

class RunnerCalibrationCliTest(unittest.TestCase):

    def test_quick_run_emits_schema_with_runner_identity(self) -> None:

        with tempfile.TemporaryDirectory() as workdir:

            completed = subprocess.run(

                [sys.executable, str(SCRIPT), "--quick", "--path", workdir],

                capture_output=True,

                text=True,

                timeout=60,

            )

        self.assertEqual(completed.returncode, 0, completed.stderr)

        row = json.loads(completed.stdout)

        self.assertEqual(set(row), EXPECTED_KEYS)

        self.assertEqual(row["schema_version"], 1)

        self.assertIsNotNone(row["runner_id"])

        self.assertGreater(row["elapsed_s"], 0)

        for key in ("cpu_sha256_mib_per_s", "io_write_read_mib_per_s", "spawn_ms_median"):

            if row[key] is not None:

                self.assertGreater(row[key], 0)

if __name__ == "__main__":

    unittest.main()