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" / "io-definition-agent.toml"
IO_DEFINITIONS_README = ROOT / "docs" / "io-definitions" / "README.md"


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

        self.assertEqual(data["name"], "io-definition-agent")
        self.assertIn("Abaqus input-file subsets", data["description"])
        self.assertEqual(data["sandbox_mode"], "read-only")
        self.assertEqual(data["model_reasoning_effort"], "extra high")
        self.assertIn("developer_instructions", data)

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

        for required_text in (
            "Do not implement parsers.",
            "Do not design C++ APIs",
            "Do not run Abaqus, Nastran, or any reference solver.",
            "Do not generate reference HDF5 artifacts or reference CSVs.",
            "Do not approve release readiness.",
            "Do not claim full Abaqus compatibility",
        ):
            self.assertIn(required_text, instructions)

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

        for required_text in (
            "FESA solver input files are Abaqus input files.",
            "Abaqus input files use keyword lines, data lines, and comment lines.",
            "Model data and history data",
            "supported Abaqus keyword subset",
            "HDF5 result/reference schemas",
        ):
            self.assertIn(required_text, instructions)

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

        for required_text in (
            "Abaqus Input Scope",
            "Syntax Policy",
            "Model Data Mapping",
            "History Data Mapping",
            "Internal Model Contract",
            "HDF5 Result and Reference Schemas",
            "Validation Rules",
            "Downstream Handoff",
        ):
            self.assertIn(required_text, instructions)

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

        for required_text in (
            "*NODE",
            "*ELEMENT",
            "*MATERIAL",
            "*BOUNDARY",
            "*STEP",
            "*OUTPUT",
            "*NODE OUTPUT",
            "*ELEMENT OUTPUT",
        ):
            self.assertIn(required_text, instructions)

    def test_io_definition_document_guide_defines_output_contract(self):
        guide = IO_DEFINITIONS_README.read_text(encoding="utf-8")

        for required_text in (
            "Abaqus Input Scope",
            "Syntax Policy",
            "Model Data Mapping",
            "History Data Mapping",
            "Internal Model Contract",
            "HDF5 Result and Reference Schemas",
            "Validation Rules",
            "Downstream Handoff",
            "FESA 솔버의 입력 파일은 Abaqus input file이다.",
        ):
            self.assertIn(required_text, guide)


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