mai-bot/pytests/webui/test_config_schema.py

from src.config.official_configs import ChatConfig, MessageReceiveConfig
from src.config.config import Config
from src.config.config_base import ConfigBase, Field
from src.webui.config_schema import ConfigSchemaGenerator


def test_field_docs_in_schema():
    """Test that field descriptions are correctly extracted from field_docs (docstrings)."""
    schema = ConfigSchemaGenerator.generate_schema(ChatConfig)
    talk_value = next(f for f in schema["fields"] if f["name"] == "talk_value")

    # Verify description field exists
    assert "description" in talk_value
    # Verify description contains expected Chinese text from the docstring
    assert "聊天频率" in talk_value["description"]


def test_json_schema_extra_merged():
    """Test that json_schema_extra fields are correctly merged into output."""
    schema = ConfigSchemaGenerator.generate_schema(ChatConfig)
    talk_value = next(f for f in schema["fields"] if f["name"] == "talk_value")

    # Verify UI metadata fields from json_schema_extra exist
    assert talk_value.get("x-widget") == "slider"
    assert talk_value.get("x-icon") == "message-circle"
    assert talk_value.get("step") == 0.1


def test_pydantic_constraints_mapped():
    """Test that Pydantic constraints (ge/le) are correctly mapped to minValue/maxValue."""
    schema = ConfigSchemaGenerator.generate_schema(ChatConfig)
    talk_value = next(f for f in schema["fields"] if f["name"] == "talk_value")

    # Verify constraints are mapped to frontend naming convention
    assert "minValue" in talk_value
    assert "maxValue" in talk_value
    assert talk_value["minValue"] == 0  # From ge=0
    assert talk_value["maxValue"] == 1  # From le=1


def test_nested_model_schema():
    """Test that nested models (ConfigBase fields) are correctly handled."""
    schema = ConfigSchemaGenerator.generate_schema(Config)

    # Verify nested structure exists
    assert "nested" in schema
    assert "chat" in schema["nested"]

    # Verify nested chat schema is complete
    chat_schema = schema["nested"]["chat"]
    assert chat_schema["className"] == "ChatConfig"
    assert "fields" in chat_schema

    # Verify nested schema fields include description and metadata
    talk_value = next(f for f in chat_schema["fields"] if f["name"] == "talk_value")
    assert "description" in talk_value
    assert talk_value.get("x-widget") == "slider"
    assert talk_value.get("minValue") == 0


def test_field_without_extra_metadata():
    """Test that fields without json_schema_extra still generate valid schema."""
    schema = ConfigSchemaGenerator.generate_schema(ChatConfig)
    inevitable_at_reply = next(f for f in schema["fields"] if f["name"] == "inevitable_at_reply")

    # Verify basic fields are generated
    assert "name" in inevitable_at_reply
    assert inevitable_at_reply["name"] == "inevitable_at_reply"
    assert "type" in inevitable_at_reply
    assert inevitable_at_reply["type"] == "boolean"
    assert "label" in inevitable_at_reply
    assert "required" in inevitable_at_reply

    # Verify no x-widget or x-icon from json_schema_extra (since field has none)
    # These fields should only be present if explicitly defined in json_schema_extra
    assert not inevitable_at_reply.get("x-widget")
    assert not inevitable_at_reply.get("x-icon")


def test_all_top_level_sections_have_ui_metadata():
    """所有顶层配置节都必须声明 uiParent 或独立 Tab 的标签与图标。"""
    schema = ConfigSchemaGenerator.generate_schema(Config)

    for section_name, section_schema in schema["nested"].items():
        has_parent = bool(section_schema.get("uiParent"))
        has_host_meta = bool(section_schema.get("uiLabel")) and bool(section_schema.get("uiIcon"))
        assert has_parent or has_host_meta, f"{section_name} 缺少 UI 元数据"


def test_maisaka_is_host_tab_and_mcp_is_attached_to_it():
    """MaiSaka 应作为独立 Tab，MCP 作为其子配置挂载。"""
    schema = ConfigSchemaGenerator.generate_schema(Config)

    maisaka_schema = schema["nested"]["maisaka"]
    mcp_schema = schema["nested"]["mcp"]

    assert maisaka_schema.get("uiParent") is None
    assert maisaka_schema.get("uiLabel") == "MaiSaka"
    assert maisaka_schema.get("uiIcon") == "message-circle"
    assert mcp_schema.get("uiParent") == "maisaka"


def test_memory_query_config_fields_are_exposed():
    """query_memory 开关和默认条数应出现在记忆配置 schema 中。"""
    schema = ConfigSchemaGenerator.generate_schema(Config)
    memory_schema = schema["nested"]["memory"]

    assert memory_schema.get("uiParent") == "emoji"

    enable_field = next(field for field in memory_schema["fields"] if field["name"] == "enable_memory_query_tool")
    limit_field = next(field for field in memory_schema["fields"] if field["name"] == "memory_query_default_limit")

    assert enable_field["type"] == "boolean"
    assert enable_field.get("x-widget") == "switch"
    assert enable_field.get("x-icon") == "database"

    assert limit_field["type"] == "integer"
    assert limit_field.get("x-widget") == "input"
    assert limit_field.get("x-icon") == "hash"
    assert limit_field.get("minValue") == 1
    assert limit_field.get("maxValue") == 20


def test_set_field_is_mapped_as_array():
    """set[str] 应映射为前端可识别的 array。"""
    schema = ConfigSchemaGenerator.generate_schema(MessageReceiveConfig)
    ban_words = next(field for field in schema["fields"] if field["name"] == "ban_words")

    assert ban_words["type"] == "array"
    assert ban_words["items"]["type"] == "string"


def test_advanced_fields_are_hidden_from_webui_schema():
    """advanced=True 的字段不应出现在 WebUI 配置 schema 中，未声明时默认展示。"""

    class AdvancedExampleConfig(ConfigBase):
        normal_field: str = Field(default="visible")
        """普通字段"""

        advanced_field: str = Field(default="hidden", json_schema_extra={"advanced": True})
        """高级字段"""

    schema = ConfigSchemaGenerator.generate_schema(AdvancedExampleConfig)
    field_names = {field["name"] for field in schema["fields"]}

    assert "normal_field" in field_names
    assert "advanced_field" not in field_names