import json
import tempfile
from pathlib import Path

from minisweagent.agents.default import DefaultAgent
from minisweagent.environments.local import LocalEnvironment
from minisweagent.models.test_models import DeterministicModel, make_output


def test_agent_save_includes_class_names():
    """Test that agent.save includes the full class names with import paths."""
    import yaml

    config_path = Path("src/minisweagent/config/default.yaml")
    with open(config_path) as f:
        default_config = yaml.safe_load(f)["agent"]

    model = DeterministicModel(outputs=[make_output("echo 'test'", [])])
    env = LocalEnvironment()
    agent = DefaultAgent(model, env, **default_config)

    agent.add_messages({"role": "system", "content": "test system message"})
    agent.add_messages({"role": "user", "content": "test user message"})

    with tempfile.TemporaryDirectory() as temp_dir:
        temp_path = Path(temp_dir) / "test_trajectory.json"

        agent.save(temp_path, {"info": {"exit_status": "Submitted", "submission": "test result"}})

        with temp_path.open() as f:
            saved_data = json.load(f)

        assert "info" in saved_data
        assert "config" in saved_data["info"]

        config = saved_data["info"]["config"]

        assert "agent_type" in config
        assert "model_type" in config
        assert "environment_type" in config

        assert config["agent_type"] == "minisweagent.agents.default.DefaultAgent"
        assert config["model_type"] == "minisweagent.models.test_models.DeterministicModel"
        assert config["environment_type"] == "minisweagent.environments.local.LocalEnvironment"

        assert saved_data["info"]["exit_status"] == "Submitted"
        assert saved_data["info"]["submission"] == "test result"
        assert saved_data["trajectory_format"] == "mini-swe-agent-1.1"


def test_agent_serialize():
    """Test that agent.serialize returns the correct structure."""
    import yaml

    config_path = Path("src/minisweagent/config/default.yaml")
    with open(config_path) as f:
        default_config = yaml.safe_load(f)["agent"]

    model = DeterministicModel(outputs=[make_output("echo 'test'", [])])
    env = LocalEnvironment()
    agent = DefaultAgent(model, env, **default_config)

    agent.add_messages({"role": "system", "content": "test system message"})
    agent.add_messages({"role": "user", "content": "test user message"})

    data = agent.serialize()

    assert "info" in data
    assert "config" in data["info"]
    assert "messages" in data