"""Campaign contracts and transaction boundaries; no paid agent invocations."""

from __future__ import annotations

import copy

import json

import os

from pathlib import Path

import sys

import tempfile

import unittest

from unittest import mock

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

from oakbench import agent_campaign as campaign

class CampaignTest(unittest.TestCase):

    def setUp(self):

        self.tmp = tempfile.TemporaryDirectory()

        self.root = Path(self.tmp.name).resolve()

        self.manifest_path = self.root / "manifest.json"

        self.raw = json.loads((campaign.ROOT / "config/agent-campaign-smoke.json").read_text())

        self.raw.update(agents_config=str(campaign.ROOT / "config/agents.toml"),

                        subjects_config=str(campaign.ROOT / "config/subjects.toml"))

        self.write_manifest()

    def tearDown(self):

        self.tmp.cleanup()

    def write_manifest(self):

        self.manifest_path.write_text(json.dumps(self.raw))

    def manifest(self):

        self.write_manifest()

        return campaign.load_manifest(self.manifest_path)

    def plan(self):

        with mock.patch.object(campaign, "resolve_runtime", return_value=({}, {}, {})):

            return campaign.create_plan(self.manifest_path, self.root / "campaign")

    def test_schedule_determinism_balance_and_identity(self):

        first = campaign.schedule(self.manifest())

        self.assertEqual(first, campaign.schedule(self.manifest()))

        self.assertEqual(len({trial["trial_id"] for trial in first}), len(first))

        self.assertEqual([x["subject"] for x in first[:2]], list(reversed([x["subject"] for x in first[2:]])))

        self.assertEqual(first[0]["block_id"], first[1]["block_id"])

        self.assertNotEqual(first[0]["block_id"], first[2]["block_id"])

        self.raw["instruction_levels"] = ["full-docs", "zero-shot"]

        self.raw["max_trials"] = 8

        self.assertEqual(len(campaign.schedule(self.manifest())), 8)

    def test_invalid_limits_unknown_fields_duplicates_and_paths(self):

        for key, value in (("schema_version", True), ("repetitions", True), ("repetitions", 0),

                           ("max_trials", -1), ("max_wall_seconds", float("nan")),

                           ("trial_timeout_seconds", float("inf")), ("seed", True),

                           ("subjects", ["git", "git"]), ("subjects", ["../escape"]),

                           ("workflows", ["history_archaeology"]), ("typo", 1)):

            with self.subTest(key=key, value=value):

                raw = dict(self.raw)

                raw[key] = value

                self.manifest_path.write_text(json.dumps(raw))

                with self.assertRaises(campaign.CampaignError):

                    campaign.load_manifest(self.manifest_path)

        self.manifest_path.write_text('{"schema_version":1,"schema_version":1}')

        with self.assertRaises(campaign.CampaignError):

            campaign.load_manifest(self.manifest_path)

    def test_huge_schedule_rejected_before_materialization(self):

        self.raw["repetitions"] = 10 ** 30

        with self.assertRaises(campaign.CampaignError):

            campaign.schedule(self.manifest())

    def test_freeze_collision_and_tamper_rejected(self):

        plan = self.plan()

        directory = self.root / "campaign"

        self.assertEqual(campaign.load_plan(directory), plan)

        with mock.patch.object(campaign, "resolve_runtime", return_value=({}, {}, {})):

            with self.assertRaises(campaign.CampaignError):

                campaign.create_plan(self.manifest_path, directory)

        plan["schedule"][0]["subject"] = "wrong"

        (directory / "plan.json").write_text(json.dumps(plan))

        with self.assertRaises(campaign.CampaignError):

            campaign.load_plan(directory)

    def test_interrupted_trial_is_never_pending_and_reserves_budget(self):

        plan = self.plan()

        directory = self.root / "campaign"

        trial = plan["schedule"][0]["trial_id"]

        campaign.append_event(directory, plan, {"event": "started", "trial_id": trial, "timeout_seconds": 60})

        status = campaign.campaign_status(plan, campaign.load_events(directory, plan))

        self.assertEqual(status["interrupted_trial_ids"], [trial])

        self.assertNotIn(trial, status["pending_trial_ids"])

        self.assertEqual(status["charged_episode_seconds"], 60)

    def test_torn_duplicate_unknown_nonfinite_and_bad_duration_journal_fail_closed(self):

        plan = self.plan()

        directory = self.root / "campaign"

        trial = plan["schedule"][0]["trial_id"]

        start = {"event": "started", "trial_id": trial, "plan_sha256": plan["plan_sha256"], "timeout_seconds": 60}

        variants = [b'{"event":', b'[]\n', (json.dumps(start) + '\n') .encode() * 2,

                    (json.dumps({**start, "trial_id": "other"}) + '\n').encode(),

                    (json.dumps({**start, "timeout_seconds": float("nan")}) + '\n').encode()]

        for wall in (True, -1, float("inf")):

            finished = {"event": "finished", "trial_id": trial, "plan_sha256": plan["plan_sha256"],

                        "outcome": "error", "error": "test", "wall_seconds": wall}

            variants.append((json.dumps(start) + '\n' + json.dumps(finished) + '\n').encode())

        for content in variants:

            with self.subTest(content=content):

                (directory / "events.jsonl").write_bytes(content)

                with self.assertRaises(campaign.CampaignError):

                    campaign.load_events(directory, plan)

                self.assertEqual((directory / "events.jsonl").read_bytes(), content)

    def test_torn_new_start_after_old_interruption_reserves_next_trial(self):

        plan = self.plan()

        directory = self.root / "campaign"

        first, second, third = plan["schedule"][:3]

        for trial in (first, second):

            campaign.append_event(directory, plan, {"event": "started", "trial_id": trial["trial_id"], "timeout_seconds": 60})

        campaign.append_event(directory, plan, {"event": "finished", "trial_id": second["trial_id"],

            "outcome": "error", "wall_seconds": 1})

        with (directory / "events.jsonl").open("ab") as stream:

            stream.write(b'{"event":"started"')

        events = campaign.load_events(directory, plan, tolerate_torn=True)

        self.assertEqual(events[-2]["trial_id"], third["trial_id"])

        self.assertTrue(events[-2]["uncertain_torn_start"])

        state = campaign.campaign_status(plan, events)

        self.assertNotIn(third["trial_id"], state["pending_trial_ids"])

    def test_system_tmp_alias_is_accepted_but_arbitrary_symlinks_are_not(self):

        self.assertEqual(campaign.safe_path(Path("/tmp") / "campaign-safe-path-check"),

                         (Path("/tmp") / "campaign-safe-path-check").resolve())

    def test_terminal_row_identity_cannot_be_rebound(self):

        from agent_metrics_schema import example_row

        plan = self.plan()

        directory = self.root / "campaign"

        trial = plan["schedule"][0]

        row = example_row()

        row.update(task_id=trial["trial_id"], bench_id=plan["campaign_id"], subject=trial["subject"],

                   scenario=trial["workflow"], run=trial["repetition"], outcome="pass")

        row["agent"].update(name=trial["agent"], model=trial["model"],

                             instruction_level=trial["instruction_level"], environment="minimal")

        campaign.validate_trial_row(row, trial, plan)

        for key, value in (("subject", "wrong"), ("scenario", "wrong"), ("run", 999), ("bench_id", "other")):

            wrong = copy.deepcopy(row)

            wrong[key] = value

            with self.subTest(key=key), self.assertRaises(campaign.CampaignError):

                campaign.validate_trial_row(wrong, trial, plan)

        for key in ("name", "model", "instruction_level"):

            wrong = copy.deepcopy(row)

            wrong["agent"][key] = "wrong"

            with self.subTest(agent_key=key), self.assertRaises(campaign.CampaignError):

                campaign.validate_trial_row(wrong, trial, plan)

        campaign.append_event(directory, plan, {"event": "started", "trial_id": trial["trial_id"], "timeout_seconds": 60})

        row["subject"] = "wrong"

        campaign.append_event(directory, plan, {"event": "finished", "trial_id": trial["trial_id"],

            "outcome": "pass", "row": row, "wall_seconds": 1})

        with self.assertRaises(campaign.CampaignError):

            campaign.load_events(directory, plan)

    def test_symlink_destinations_and_lock_rejected(self):

        alias = self.root / "alias"

        alias.symlink_to(self.root, target_is_directory=True)

        with self.assertRaises(campaign.CampaignError):

            campaign.safe_path(alias / "campaign")

        plan = self.plan()

        directory = self.root / "campaign"

        for name in ("events.jsonl", ".run.lock", "trials"):

            path = directory / name

            path.symlink_to(self.root / "outside")

            with self.assertRaises(campaign.CampaignError):

                campaign.safe_path(path)

            path.unlink()

    def test_exclusive_runner_ownership(self):

        self.plan()

        with campaign.campaign_lock(self.root / "campaign"):

            with self.assertRaises(campaign.CampaignError):

                with campaign.campaign_lock(self.root / "campaign"):

                    pass

    def test_preflight_drift_real_optin_and_attestation(self):

        plan = self.plan()

        directory = self.root / "campaign"

        with mock.patch.object(campaign, "resolve_runtime", return_value=({}, {}, {"changed": True})):

            with self.assertRaises(campaign.CampaignError):

                campaign.preflight(plan, directory)

        from agent_workflow import AgentConfig

        real = AgentConfig("real", "codex_exec_json", "openai", Path("/binary"), "pinned", "1")

        with mock.patch.object(campaign, "resolve_runtime", return_value=({"real": real}, {}, {})):

            with self.assertRaises(campaign.CampaignError):

                campaign.preflight(plan, directory)

            with self.assertRaises(campaign.CampaignError):

                campaign.preflight(plan, directory, enable_real=True)

            plan["manifest"]["isolation_attestation"] = {

                "worker_root": str(self.root), "unrestricted_cli_acknowledged": True}

            self.assertEqual(campaign.preflight(plan, directory, enable_real=True)[0]["real"], real)

    def test_lock_provenance_drift_names_the_key(self):

        plan = self.plan()

        plan["provenance"] = {"measurement_lock_path": "/old/lock"}

        with mock.patch.object(campaign, "resolve_runtime", return_value=({}, {}, {"measurement_lock_path": "/new/lock"})):

            with self.assertRaisesRegex(campaign.CampaignError, "measurement_lock_path"):

                campaign.preflight(plan, self.root / "campaign")

    def test_fixture_cache_reuses_generation_and_rejects_mutated_payload(self):

        import workflow_ab

        from dataclasses import replace

        counter = mock.Mock(side_effect=workflow_ab.WORKFLOWS["wide_config_refactor"].make_fixture)

        workflow = replace(workflow_ab.WORKFLOWS["wide_config_refactor"], make_fixture=counter)

        plan = self.plan()

        directory = self.root / "campaign"

        fixture = campaign.cached_fixture(directory, plan, workflow)

        self.assertEqual(fixture, directory / "fixtures" / workflow.name)

        self.assertEqual(campaign.cached_fixture(directory, plan, workflow), fixture)

        self.assertEqual(counter.call_count, 1)

        payload = next(fixture.glob("services/*/config.toml"))

        payload.write_text("corrupt fixture")

        with self.assertRaisesRegex(campaign.CampaignError, "fixture.*drift"):

            campaign.cached_fixture(directory, plan, workflow)

        self.assertEqual(counter.call_count, 1)

    def test_completed_infrastructure_error_not_retried_on_resume(self):

        self.plan()

        directory = self.root / "campaign"

        import agent_workflow

        import workflow_ab

        from oakbench.runlock import LockInfo

        from contextlib import contextmanager

        @contextmanager

        def locked(label):

            yield LockInfo(0, True)

        agent = agent_workflow.AgentConfig("mock", "deterministic_mock", "local", None, None, "1")

        subject = workflow_ab.Subject("git", "git", "Git", Path("/git"))

        with mock.patch.object(campaign, "preflight", return_value=({"mock": agent}, {"git": subject, "oak_installed": subject})), \

             mock.patch("oakbench.runlock.measurement_lock", locked), \

             mock.patch.object(campaign, "cached_fixture", return_value=self.root), \

             mock.patch.object(agent_workflow, "run_one", side_effect=RuntimeError("fixture setup failed")) as run:

            state = campaign.run_campaign(directory, limit=1)

            self.assertEqual(state["finished"], 1)

            campaign.run_campaign(directory, resume=True, limit=1)

            self.assertEqual(run.call_count, 2)

            events = campaign.load_events(directory, campaign.load_plan(directory))

            self.assertEqual(len({event["trial_id"] for event in events if event["event"] == "started"}), 2)

            self.assertEqual([event["outcome"] for event in events if event["event"] == "finished"], ["error", "error"])

    def test_mkdir_crash_after_durable_start_is_not_pending(self):

        from contextlib import contextmanager

        from oakbench.runlock import LockInfo

        self.plan()

        directory = self.root / "campaign"

        @contextmanager

        def locked(label):

            yield LockInfo(0, True)

        original = Path.mkdir

        def crash(path, *args, **kwargs):

            if path.parent.name == "trials":

                raise KeyboardInterrupt("simulated process crash")

            return original(path, *args, **kwargs)

        with mock.patch.object(campaign, "preflight", return_value=({}, {})), \

             mock.patch("oakbench.runlock.measurement_lock", locked), \

             mock.patch.object(Path, "mkdir", crash):

            with self.assertRaises(KeyboardInterrupt):

                campaign.run_campaign(directory)

        plan = campaign.load_plan(directory)

        events = campaign.load_events(directory, plan)

        self.assertEqual(len(events), 1)

        state = campaign.campaign_status(plan, events)

        self.assertEqual(state["interrupted_trial_ids"], [plan["schedule"][0]["trial_id"]])

        self.assertNotIn(plan["schedule"][0]["trial_id"], state["pending_trial_ids"])

    def test_schema_rejected_row_remains_in_terminal_evidence(self):

        import agent_workflow

        from contextlib import contextmanager

        from oakbench.runlock import LockInfo

        plan = self.plan()

        directory = self.root / "campaign"

        @contextmanager

        def locked(label):

            yield LockInfo(0, True)

        subjects = {trial["subject"]: object() for trial in plan["schedule"]}

        with mock.patch.object(campaign, "preflight", return_value=({"mock": object()}, subjects)), \

             mock.patch("oakbench.runlock.measurement_lock", locked), \

             mock.patch.object(campaign, "cached_fixture", return_value=self.root), \

             mock.patch.object(agent_workflow, "run_one", return_value={"invalid": "original evidence"}):

            campaign.run_campaign(directory, limit=1)

        events = campaign.load_events(directory, plan)

        self.assertEqual(events[1]["outcome"], "error")

        self.assertEqual(events[1]["rejected_row"]["invalid"], "original evidence")

        self.assertEqual((directory / "rows.jsonl").read_bytes(), b"")

    def test_noop_resume_repairs_missing_or_stale_rows_from_finished_journal(self):

        from agent_metrics_schema import example_row

        self.raw.update(subjects=["git"], repetitions=1)

        self.write_manifest()

        plan = self.plan()

        directory = self.root / "campaign"

        trial = plan["schedule"][0]

        row = example_row()

        row.update(task_id=trial["trial_id"], bench_id=plan["campaign_id"], subject=trial["subject"],

                   scenario=trial["workflow"], run=trial["repetition"], outcome="pass")

        row["agent"].update(name=trial["agent"], model=trial["model"],

                             instruction_level=trial["instruction_level"], environment="minimal")

        campaign.append_event(directory, plan, {"event": "started", "trial_id": trial["trial_id"], "timeout_seconds": 60})

        campaign.append_event(directory, plan, {"event": "finished", "trial_id": trial["trial_id"],

            "outcome": "pass", "row": row, "wall_seconds": 1})

        journal = (directory / "events.jsonl").read_bytes()

        for state in ("missing", "stale"):

            with self.subTest(state=state):

                if state == "stale":

                    (directory / "rows.jsonl").write_bytes(b"stale output")

                with mock.patch.object(campaign, "preflight", return_value=({}, {})), \

                     mock.patch("agent_workflow.run_one") as run:

                    status = campaign.run_campaign(directory, resume=True)

                    run.assert_not_called()

                self.assertEqual(status["finished"], 1)

                self.assertEqual((directory / "rows.jsonl").read_bytes(), campaign.canonical(row) + b"\n")

                self.assertEqual((directory / "events.jsonl").read_bytes(), journal)

    def test_disabled_measurement_lock_starts_nothing(self):

        self.plan()

        directory = self.root / "campaign"

        with mock.patch.object(campaign, "preflight", return_value=({}, {})), \

             mock.patch.dict(os.environ, {"OAK_BENCH_LOCK": "off"}):

            with self.assertRaises(campaign.CampaignError):

                campaign.run_campaign(directory)

        self.assertFalse((directory / "events.jsonl").exists())

    def test_artifact_hashes_exclude_links(self):

        trial = self.root / "trial"

        (trial / "artifacts").mkdir(parents=True)

        (trial / "artifacts/prompt.txt").write_text("fixed prompt")

        (trial / "artifacts/escape").symlink_to(self.manifest_path)

        hashes = campaign.artifact_hashes(trial)

        self.assertEqual(set(hashes), {"artifacts/prompt.txt"})

        self.assertEqual(hashes["artifacts/prompt.txt"], campaign.file_hash(trial / "artifacts/prompt.txt"))

if __name__ == "__main__":

    unittest.main()