import unittest
from pathlib import Path

try:
    import tomllib
except ModuleNotFoundError:  # pragma: no cover
    import tomli as tomllib


ROOT = Path(__file__).resolve().parents[1]
AGENT_PATH = ROOT / ".codex" / "agents" / "coordinator-agent.toml"
COORDINATION_README = ROOT / "docs" / "coordination" / "README.md"


class CoordinatorAgentConfigTests(unittest.TestCase):
    def test_coordinator_agent_toml_has_required_codex_fields(self):
        data = tomllib.loads(AGENT_PATH.read_text(encoding="utf-8"))

        self.assertEqual(data["name"], "coordinator-agent")
        self.assertIn("workflow state", data["description"])
        self.assertEqual(data["sandbox_mode"], "workspace-write")
        self.assertEqual(data["model_reasoning_effort"], "high")
        self.assertIn("developer_instructions", data)

    def test_coordinator_agent_instructions_enforce_boundaries(self):
        instructions = AGENT_PATH.read_text(encoding="utf-8")

        for required_text in (
            "Do not implement code.",
            "Do not edit source code.",
            "Do not edit tests.",
            "Do not edit CMake.",
            "Do not run build/test validation.",
            "Do not run Abaqus, Nastran, or any reference solver.",
            "Do not generate reference CSVs.",
            "Do not automatically spawn subagents.",
            "Do not approve release readiness independently.",
        ):
            self.assertIn(required_text, instructions)

    def test_coordinator_agent_instructions_define_workflow_and_status_contract(self):
        instructions = AGENT_PATH.read_text(encoding="utf-8")

        for required_text in (
            "INTAKE -> STATE AUDIT -> GATE DECISION -> HANDOFF PACKAGE -> STATUS REPORT",
            "Requirement Agent",
            "Research Agent",
            "Formulation Agent",
            "Numerical Review Agent",
            "I/O Definition Agent",
            "Reference Model Agent",
            "Implementation Planning Agent",
            "Implementation Agent",
            "Build/Test Executor Agent",
            "Correction Agent",
            "Reference Verification Agent",
            "Physics Evaluation Agent",
            "Release Agent",
            "ready-for-implementation",
            "pass-for-reference-verification",
            "pass-for-physics-evaluation",
            "pass-for-release-agent",
            "ready-for-release",
            "completed",
        ):
            self.assertIn(required_text, instructions)

    def test_coordinator_agent_instructions_define_output_contract(self):
        instructions = AGENT_PATH.read_text(encoding="utf-8")

        for required_text in (
            "Gate Evidence Inventory",
            "Decision Log",
            "Next Agent Handoff",
            "Traceability Snapshot",
            "Risk and Blocker Register",
            "Rework Loop Control",
            "No-Change Assertion",
        ):
            self.assertIn(required_text, instructions)

    def test_coordination_guide_defines_template_and_status_values(self):
        guide = COORDINATION_README.read_text(encoding="utf-8")

        for required_text in (
            "docs/coordination/<feature-id>-coordination.md",
            "Gate Evidence Inventory",
            "Decision Log",
            "Next Agent Handoff",
            "Traceability Snapshot",
            "Risk and Blocker Register",
            "Rework Loop Control",
            "No-Change Assertion",
            "needs-user-decision",
            "blocked",
            "python scripts/validate_workspace.py",
        ):
            self.assertIn(required_text, guide)


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