from __future__ import annotations

import pytest

from schemas.automation.config import AutomationJobConfig, default_memory_job_config


def test_default_memory_job_config_has_expected_defaults() -> None:
    config = default_memory_job_config()

    assert config.agent_type.value == "memory"
    assert config.model_code == "qwen3.5-flash"
    assert config.context.source.value == "latest_chat"


def test_automation_job_config_rejects_non_flash_model() -> None:
    with pytest.raises(ValueError, match="model_code must be qwen3.5-flash"):
        AutomationJobConfig.model_validate(
            {
                "agent_type": "memory",
                "model_code": "qwen-plus",
                "enabled_tools": ["calendar.read"],
                "input_template": "x",
                "context": {
                    "source": "latest_chat",
                    "window_mode": "day",
                    "window_count": 2,
                },
            }
        )