import json
import unittest

from cockpit_agent.intent.parser import ModelIntentParser


EXPECTED = {
    "主驾23度": {
        "domain": "climate",
        "function": "temperature",
        "zone": "driver",
        "operation": "set",
        "state": None,
        "value": 23,
        "unit": "celsius",
    },
    "把驾驶位温度调成23℃": {
        "domain": "climate",
        "function": "temperature",
        "zone": "driver",
        "operation": "set",
        "state": None,
        "value": 23,
        "unit": "celsius",
    },
    "副驾温度降低2度": {
        "domain": "climate",
        "function": "temperature",
        "zone": "passenger",
        "operation": "decrease",
        "state": None,
        "value": 2,
        "unit": "celsius",
    },
    "打开内循环": {
        "domain": "climate",
        "function": "air_recirculation",
        "zone": "all",
        "operation": "set",
        "state": "on",
        "value": None,
        "unit": None,
    },
    "关闭主驾空调": {
        "domain": "climate",
        "function": "climate_power",
        "zone": "driver",
        "operation": "set",
        "state": "off",
        "value": None,
        "unit": None,
    },
    "后备箱最大开度80%": {
        "domain": "body",
        "function": "tailgate_max_opening",
        "zone": None,
        "operation": "set",
        "state": None,
        "value": 80,
        "unit": "percent",
    },
    "解锁全部车门": {
        "domain": "body",
        "function": "unlock_mode",
        "zone": None,
        "operation": "set",
        "state": None,
        "value": "all_doors",
        "unit": None,
    },
}


class FakeTextGenerator:
    def generate_text(self, prompt: str, max_new_tokens: int = 256) -> str:
        del max_new_tokens
        instruction = prompt.rsplit("用户指令：", maxsplit=1)[1].strip()
        return "```json\n" + json.dumps(EXPECTED[instruction]) + "\n```"


class IntentParserTest(unittest.TestCase):
    def test_required_instruction_variants(self) -> None:
        parser = ModelIntentParser(FakeTextGenerator())
        for instruction, expected in EXPECTED.items():
            with self.subTest(instruction=instruction):
                self.assertEqual(parser.parse(instruction).to_dict(), expected)

    def test_empty_instruction_is_rejected(self) -> None:
        parser = ModelIntentParser(FakeTextGenerator())
        with self.assertRaisesRegex(ValueError, "non-empty"):
            parser.parse("  ")

    def test_invalid_model_schema_is_rejected(self) -> None:
        class InvalidGenerator:
            def generate_text(self, prompt: str, max_new_tokens: int = 256) -> str:
                del prompt, max_new_tokens
                return '{"domain": "climate"}'

        with self.assertRaisesRegex(ValueError, "missing fields"):
            ModelIntentParser(InvalidGenerator()).parse("打开内循环")


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