ForcePilot/test/test_chat_stream_interrupt.py

"""测试 chat_stream_service 中的 interrupt 相关函数"""
import pytest
import sys
import os

sys.path.insert(0, os.getcwd())

from src.services.chat_stream_service import (
    _normalize_interrupt_options,
    _build_ask_user_question_payload,
    _coerce_interrupt_payload,
)


class TestNormalizeInterruptOptions:
    """测试 _normalize_interrupt_options 函数"""

    def test_empty_input(self):
        assert _normalize_interrupt_options(None) == []
        assert _normalize_interrupt_options([]) == []

    def test_dict_options(self):
        raw = [
            {"label": "选项1", "value": "option1"},
            {"label": "选项2", "value": "option2"},
        ]
        result = _normalize_interrupt_options(raw)
        assert len(result) == 2
        assert result[0] == {"label": "选项1", "value": "option1"}
        assert result[1] == {"label": "选项2", "value": "option2"}

    def test_string_options(self):
        raw = ["选项1", "选项2", "选项3"]
        result = _normalize_interrupt_options(raw)
        assert len(result) == 3
        assert result[0] == {"label": "选项1", "value": "选项1"}

    def test_mixed_options(self):
        raw = [{"label": "选项1", "value": "option1"}, "选项2"]
        result = _normalize_interrupt_options(raw)
        assert len(result) == 2
        assert result[0] == {"label": "选项1", "value": "option1"}
        assert result[1] == {"label": "选项2", "value": "选项2"}

    def test_invalid_options(self):
        raw = [{"label": "只有label"}, {}, "  "]
        result = _normalize_interrupt_options(raw)
        assert len(result) == 1  # 只有有效的选项
        assert result[0] == {"label": "只有label", "value": "只有label"}

    def test_value_only(self):
        raw = [{"value": "only_value"}]
        result = _normalize_interrupt_options(raw)
        assert len(result) == 1
        assert result[0] == {"label": "only_value", "value": "only_value"}


class TestBuildAskUserQuestionPayload:
    """测试 _build_ask_user_question_payload 函数"""

    def test_basic_question(self):
        info = {
            "question": "请确认是否继续？",
            "options": [
                {"label": "确认", "value": "yes"},
                {"label": "取消", "value": "no"},
            ],
        }
        result = _build_ask_user_question_payload(info, "thread-123")

        assert result["question"] == "请确认是否继续？"
        assert len(result["options"]) == 2
        assert result["options"][0] == {"label": "确认", "value": "yes"}
        assert result["options"][1] == {"label": "取消", "value": "no"}
        assert result["source"] == "interrupt"
        assert result["thread_id"] == "thread-123"
        assert result["multi_select"] is False
        assert result["allow_other"] is True

    def test_question_with_source(self):
        info = {
            "question": "选择一个选项",
            "options": ["A", "B", "C"],
            "source": "ask_user_question",
        }
        result = _build_ask_user_question_payload(info, "thread-456")

        assert result["source"] == "ask_user_question"
        assert len(result["options"]) == 3

    def test_multi_select(self):
        info = {
            "question": "选择多个",
            "options": ["A", "B", "C"],
            "multi_select": True,
        }
        result = _build_ask_user_question_payload(info, "thread-789")

        assert result["multi_select"] is True

    def test_disable_allow_other(self):
        info = {
            "question": "只能选择",
            "options": ["A", "B"],
            "allow_other": False,
        }
        result = _build_ask_user_question_payload(info, "thread-000")

        assert result["allow_other"] is False

    def test_with_operation(self):
        info = {
            "question": "是否执行操作？",
            "operation": "删除文件",
            "options": [{"label": "批准", "value": "approve"}, {"label": "拒绝", "value": "reject"}],
        }
        result = _build_ask_user_question_payload(info, "thread-op")

        assert result["operation"] == "删除文件"

    def test_no_options(self):
        """测试没有 options 的情况 - 不再自动填充 legacy 选项"""
        info = {
            "question": "请确认？",
        }
        result = _build_ask_user_question_payload(info, "thread-no-opt")

        # 不再有默认的 approve/reject 选项
        assert result["options"] == []
        assert result["source"] == "interrupt"

    def test_question_id_generation(self):
        """测试 question_id 自动生成"""
        info = {"question": "测试？"}
        result = _build_ask_user_question_payload(info, "thread-id")

        # 应该生成了 UUID
        assert result["question_id"] != ""
        assert len(result["question_id"]) > 0


class TestCoerceInterruptPayload:
    """测试 _coerce_interrupt_payload 函数"""

    def test_dict_input(self):
        info = {"question": "test?", "options": ["a", "b"]}
        result = _coerce_interrupt_payload(info)
        assert result == info

    def test_string_input(self):
        info = "just a string"
        result = _coerce_interrupt_payload(info)
        assert isinstance(result, dict)

    def test_none_input(self):
        result = _coerce_interrupt_payload(None)
        assert isinstance(result, dict)


if __name__ == "__main__":
    pytest.main([__file__, "-v"])