"""Adversarial saved-baseline comparisons and claim eligibility controls."""

from dataclasses import replace

import contextlib

import io

import json

from pathlib import Path

import subprocess

import sys

import tempfile

import unittest

from unittest.mock import patch

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))

import benchmark_stats

from oakbench.baseline_book import Target, book_json, cells_from_rows, load_baseline_book

from oakbench.populations import comparison_identity

from oakbench.scorecard import evaluate_targets

def rows(subject="git", **overrides):

    result = []

    for index in range(30):

        row = dict(subject=subject, subject_kind="git" if subject == "git" else "oak",

                   bench_id="campaign", run=index, lane="core", profile="standard",

                   scenario="tiny", operation="status.clean", returncode=0,

                   elapsed_ms=10.0 if subject == "git" else 1.0,

                   benchmark_track="core-equivalent", command_semantics_version=1,

                   fixture_version=1, scenario_version=1, runner_class="mac-arm",

                   cache_state="warm", env_isolation_version="1", load_tier="idle",

                   binary_sha256="git-sha" if subject == "git" else "oak-sha")

        row.update(overrides)

        result.append(row)

    return result

TARGET = Target("t", "core", "standard", "tiny", "status.clean", "oak_local", "git",

                "elapsed_ms", "median", "ratio_at_most", .8, {})

def baseline(source=None):

    return cells_from_rows(rows() if source is None else source, lane="core", metrics=("elapsed_ms",),

                           statistics_names=("median",), noise_floor_pct=1.0, bootstrap_samples=20)[0]

def evaluate(candidate=None, cell=None, mode="claim"):

    cell = baseline() if cell is None else cell

    return evaluate_targets(rows("oak_local") if candidate is None else candidate,

                            [TARGET], {cell.key: cell}, mode=mode, bootstrap_samples=20)[0]

class ClaimEligibilityTest(unittest.TestCase):

    def test_comparable_independent_trials_allow_different_vcs_binary_hashes(self):

        result = evaluate()

        self.assertEqual(result["verdict"], "GREEN")

        self.assertTrue(result["claim_eligible"])

    def test_saved_baseline_mismatch_is_invalid_even_in_diagnostic_mode(self):

        for field, value in [("runner_class", "linux"), ("fixture_version", 2),

                             ("benchmark_track", "agent-default"), ("cache_state", "cold"),

                             ("remote_transport", "network"), ("load_tier", "cpu50"),

                             ("output_oracle_version", "different")]:

            with self.subTest(field=field):

                for mode in ("claim", "diagnostic"):

                    result = evaluate(rows("oak_local", **{field: value}), mode=mode)

                    self.assertEqual(result["verdict"], "INVALID")

                    self.assertFalse(result["claim_eligible"])

    def test_missing_and_duplicate_trials_cannot_manufacture_precision(self):

        candidates = [rows("oak_local")[:1], rows("oak_local", run=0),

                      rows("oak_local") + [rows("oak_local")[0]] * 1000]

        missing = rows("oak_local")

        del missing[0]["bench_id"]

        candidates.append(missing)

        for candidate in candidates:

            with self.subTest(n=len(candidate)):

                result = evaluate(candidate)

                self.assertEqual(result["verdict"], "AMBER")

                self.assertFalse(result["claim_eligible"])

    def test_unknown_baseline_evidence_cannot_be_public_green(self):

        for changes in ({"noise_floor_pct": None}, {"noise_floor_pct": float("nan")},

                        {"noise_floor_pct": -1}, {"comparison_identity": None},

                        {"comparison_identity": {}}, {"independent_n": 1},

                        {"attempted_n": None}, {"failure_n": 1}, {"ci_low": None}):

            with self.subTest(changes=changes):

                result = evaluate(cell=replace(baseline(), **changes))

                self.assertNotEqual(result["verdict"], "GREEN")

                self.assertFalse(result["claim_eligible"])

    def test_uniformly_unknown_cache_cannot_be_claim_evidence(self):

        result = evaluate(rows("oak_local", cache_state="unknown"), baseline(rows(cache_state="unknown")))

        self.assertEqual(result["verdict"], "AMBER")

        self.assertIn("cache_state", " ".join(result["messages"]))

    def test_legacy_diagnostic_is_explicitly_not_claim_eligible(self):

        cell = replace(baseline(), comparison_identity=None, noise_floor_pct=None)

        result = evaluate(rows("oak_local")[:1], cell, mode="diagnostic")

        self.assertEqual(result["verdict"], "GREEN")

        self.assertFalse(result["claim_eligible"])

        self.assertIn("diagnostic only", " ".join(result["messages"]))

    def test_baseline_builder_preserves_failures_and_unmeasured_denominators(self):

        source = rows() + [dict(rows()[0], returncode=1), dict(rows()[0], elapsed_ms=None)]

        cell = baseline(source)

        self.assertEqual(cell.attempted_n, 32)

        self.assertEqual(cell.failure_n, 1)

        self.assertFalse(evaluate(cell=cell)["claim_eligible"])

    def test_baseline_builder_rejects_mixed_populations(self):

        with self.assertRaisesRegex(ValueError, "incompatible baseline population"):

            baseline(rows() + rows(runner_class="linux"))

    def test_book_v2_roundtrip_and_duplicate_keys_rejected(self):

        book = book_json([baseline()])

        self.assertEqual(book["schema_version"], 2)

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "book.json"

            path.write_text(json.dumps(book))

            cell = next(iter(load_baseline_book(path).values()))

            self.assertTrue(evaluate(cell=cell)["claim_eligible"])

            book["cells"].append(book["cells"][0])

            path.write_text(json.dumps(book))

            with self.assertRaisesRegex(ValueError, "duplicate baseline cell"):

                load_baseline_book(path)

    def test_unknown_identity_version_rejected(self):

        book = book_json([baseline()])

        book["cells"][0]["comparison_identity"]["schema_version"] = 99

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "book.json"

            path.write_text(json.dumps(book))

            with self.assertRaisesRegex(ValueError, "unsupported baseline comparison_identity"):

                load_baseline_book(path)

    def test_versioned_patch_guard_requires_actual_success(self):

        target = replace(TARGET, guards={"patch_apply_ok": True})

        cell = baseline(rows(patch_oracle_version="apply-tree-v1"))

        for evidence in (None, False, True):

            candidate = rows("oak_local", patch_oracle_version="apply-tree-v1", patch_apply_ok=evidence)

            result = evaluate_targets(candidate, [target], {cell.key: cell}, mode="claim", bootstrap_samples=20)[0]

            self.assertEqual(result["verdict"], "GREEN" if evidence is True else "INVALID")

    def test_failure_only_recall_guard_fails_without_crashing(self):

        target = replace(TARGET, guards={"information_recall_at_least": 1.0})

        cell = baseline()

        result = evaluate_targets(rows("oak_local", returncode=1), [target], {cell.key: cell})[0]

        self.assertEqual(result["verdict"], "INVALID")

    def test_nested_fixture_version_normalizes_but_subject_prompt_does_not_compare(self):

        row = rows("oak_local")[0]

        del row["fixture_version"]

        row["fixture"] = {"fixture_version": 1}

        row["task_prompt_id"] = "oak-specific-label"

        self.assertEqual(comparison_identity(row), comparison_identity(rows()[0]))

    def test_cli_defaults_to_claim_and_refuses_legacy_evidence(self):

        target = {"schema_version": 1, "targets": [dict(id="t", lane="core", profile="standard",

                  scenario="tiny", operation="status.clean", subject="oak_local", comparator="git",

                  metric="elapsed_ms", goal=dict(type="ratio_at_most", value=.8))]}

        with tempfile.TemporaryDirectory() as tmp:

            root = Path(tmp)

            (root / "rows.jsonl").write_text("\n".join(json.dumps(row) for row in rows("oak_local")))

            (root / "book.json").write_text(json.dumps(book_json([replace(baseline(), comparison_identity=None)])))

            (root / "targets.json").write_text(json.dumps(target))

            command = [sys.executable, "scripts/scorecard.py", str(root / "rows.jsonl"),

                       "--baseline-book", str(root / "book.json"), "--targets", str(root / "targets.json"), "--json"]

            proc = subprocess.run(command, capture_output=True, text=True)

            self.assertEqual(proc.returncode, 1)

            self.assertEqual(json.loads(proc.stdout)[0]["mode"], "claim")

class OutcomeReportingTest(unittest.TestCase):

    def test_failure_only_cells_and_unknown_outcomes_stay_visible(self):

        source = [dict(rows()[0], returncode=1, elapsed_ms=999),

                  dict(rows()[0], returncode=None), dict(rows()[0], returncode=77)]

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "rows.jsonl"

            path.write_text("\n".join(json.dumps(row) for row in source))

            output = io.StringIO()

            with patch.object(sys, "argv", ["benchmark_stats.py", str(path)]), contextlib.redirect_stdout(output):

                self.assertEqual(benchmark_stats.main(), 0)

            self.assertIn("| `git` | `tiny/status.clean` | 2 | 0 | 1 | 1 | 1 | 0 |", output.getvalue())

            self.assertIn("| 0 | unmeasured |", output.getvalue())

            self.assertNotIn("999", output.getvalue())

if __name__ == "__main__":

    unittest.main()