"""主模块测试套件。

本文件包含三个测试类，分别覆盖：
1. 配置加载功能（从 config.json 读取项目设置）
2. 语言规则生成（根据配置或自动检测确定输出语言）
3. 分析提示构建（组装完整的 AI 分析提示词）
"""

import sys
from pathlib import Path

# 将 src 目录加入 Python 路径，使测试可以导入项目模块
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))

from config import get_language, load_config
from main import build_analysis_prompt, language_rule


class TestConfig:
    """测试配置加载模块。

    验证 config.json 能否被正确读取，以及语言设置是否能正常获取。
    """

    def test_load_config_returns_dict(self) -> None:
        """测试 load_config 返回字典且包含 language 键。

        验证配置文件的解析结果是一个字典结构，
        并且其中包含必需的 language 字段。
        """
        config = load_config()
        assert isinstance(config, dict)
        assert "language" in config

    def test_get_language_from_config(self) -> None:
        """测试 get_language 能正常返回语言设置。

        当 config.json 中 language 不为 "auto" 时，
        get_language 应返回非 None 值。
        """
        language = get_language()
        assert language is not None


class TestLanguageRule:
    """测试语言规则生成函数。

    language_rule 负责根据配置或源文档内容，
    生成指导 AI 输出语言的提示文本。
    """

    def test_explicit_language(self) -> None:
        """测试显式指定语言时，提示中包含该语言名称。

        当 configured_language 为 "Chinese" 时，
        返回的提示应包含 "Chinese" 字样。
        """
        result = language_rule("test content", "Chinese")
        assert "Chinese" in result

    def test_auto_with_chinese_content(self) -> None:
        """测试中文内容自动检测。

        当源文档包含中文字符且未指定语言时，
        应自动选择简体中文作为输出语言。
        """
        result = language_rule("这是一段中文内容", None)
        assert "简体中文" in result

    def test_auto_with_english_content(self) -> None:
        """测试英文内容自动检测。

        当源文档为纯英文且未指定语言时，
        应返回跟随源文档语言的提示。
        """
        result = language_rule("This is English content", None)
        assert "源文档" in result

    def test_auto_detection_string(self) -> None:
        """测试传入 "auto" 字符串时触发自动检测。

        "auto" 和 None 等效，都表示启用自动语言检测。
        """
        result = language_rule("test", "auto")
        assert "源文档" in result


class TestBuildAnalysisPrompt:
    """测试分析提示构建函数。

    build_analysis_prompt 负责将 wiki 目的、索引、
    源文档内容和语言设置组装成完整的 AI 提示词。
    """

    def test_basic_prompt(self) -> None:
        """测试基础提示词包含核心结构。

        即使目的和索引都为空，提示词仍应包含
        角色定义（专业研究分析师）和分析框架（关键实体）。
        """
        prompt = build_analysis_prompt(purpose="", index="")
        assert "专业研究分析师" in prompt
        assert "关键实体" in prompt

    def test_with_purpose(self) -> None:
        """测试传入目的时，提示词包含 Wiki 目的部分。

        当 purpose 参数非空时，生成的提示词应包含
        "Wiki 目的" 标题和传入的目的内容。
        """
        prompt = build_analysis_prompt(purpose="测试目的", index="")
        assert "Wiki 目的" in prompt
        assert "测试目的" in prompt

    def test_with_index(self) -> None:
        """测试传入索引时，提示词包含 Wiki 索引部分。

        当 index 参数非空时，生成的提示词应包含
        "当前 Wiki 索引" 标题和传入的索引内容。
        """
        prompt = build_analysis_prompt(purpose="", index="测试索引")
        assert "当前 Wiki 索引" in prompt
        assert "测试索引" in prompt

    def test_empty_content(self) -> None:
        """测试空内容时，可选部分不出现在提示词中。

        当 purpose 和 index 都为空时，
        提示词不应包含 "Wiki 目的" 和 "当前 Wiki 索引" 部分，
        以保持提示词简洁。
        """
        prompt = build_analysis_prompt(purpose="", index="")
        assert "Wiki 目的" not in prompt
        assert "当前 Wiki 索引" not in prompt

    def test_language_from_config(self) -> None:
        """测试语言设置从配置文件读取。

        当 configured_language 为 None 时，
        build_analysis_prompt 会从 config.json 读取语言设置，
        生成的提示词应包含语言规则。
        """
        prompt = build_analysis_prompt(purpose="", index="", configured_language=None)
        assert "使用" in prompt
        assert "语言输出分析报告" in prompt