import json
import os
import sys
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch

ROOT = Path(__file__).resolve().parents[1]
SRC = ROOT / "src"
if str(SRC) not in sys.path:
    sys.path.insert(0, str(SRC))

from agent_app.revenue_assurance.aml_submit import AmlSubmitConfig, build_job_spec, load_submit_config


class AmlSubmitTests(unittest.TestCase):
    def test_build_job_spec_uses_config_path_and_defaults(self) -> None:
        with tempfile.TemporaryDirectory() as tmpdir, patch.dict(
            os.environ,
            {
                "AML_SUBSCRIPTION_ID": "",
                "AML_RESOURCE_GROUP": "",
                "AML_WORKSPACE_NAME": "",
                "AML_COMPUTE_NAME": "",
            },
            clear=False,
        ):
            model_config_path = Path(tmpdir) / "model_config.json"
            model_config_path.write_text(
                json.dumps(
                    {
                        "model_endpoint": "https://example.services.ai.azure.com/openai/v1/responses",
                        "model_deployment": "gpt-5.6-sol",
                        "api_version": "2026-07-09",
                        "api_key": "secret",
                        "use_agentic": True,
                    }
                )
            )

            submit_config_path = Path(tmpdir) / "submit_config.json"
            submit_config_path.write_text(
                json.dumps(
                    {
                        "subscription_id": "sub-id",
                        "resource_group": "rg-name",
                        "workspace_name": "ws-name",
                        "compute_name": "cpu-cluster",
                        "code_path": ".",
                        "environment_image": "python:3.12-slim",
                        "experiment_name": "revenue-assurance",
                        "display_name": "revenue-assurance-agent",
                        "revenue_assurance_config_path": str(model_config_path),
                    }
                )
            )

            submit_config = load_submit_config(str(submit_config_path))
            self.assertIsInstance(submit_config, AmlSubmitConfig)

            spec = build_job_spec(submit_config=submit_config)

            self.assertEqual(spec["compute"], "cpu-cluster")
            self.assertTrue(spec["environment_variables"]["REVENUE_ASSURANCE_USE_AGENTIC"])
            self.assertIn("python -m agent_app.revenue_assurance.aml_runner", spec["command"])
            self.assertEqual(spec["model_deployment"], "gpt-5.6-sol")


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