refactor(tests): reorganize tests into unittest/ and e2e/ directories

- Move all unit tests from tests/ to tests/unittest/ - Add tests/e2e/ directory for end-to-end tests - Update conftest.py for new test structure - Add new tests for auth dependencies, policies, route injection - Add new tests for run callbacks, create store, execution artifacts - Remove obsolete tests for deleted persistence layer Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-05-23 16:35:59 +00:00 · 2026-04-22 11:24:53 +08:00
parent 38a6ec496f
commit 2fe0856e33
149 changed files with 3450 additions and 4664 deletions
@@ -0,0 +1,149 @@
+from langchain_core.messages import AIMessageChunk, HumanMessage
+
+from deerflow.models.patched_minimax import PatchedChatMiniMax
+
+
+def _make_model(**kwargs) -> PatchedChatMiniMax:
+    return PatchedChatMiniMax(
+        model="MiniMax-M2.5",
+        api_key="test-key",
+        base_url="https://example.com/v1",
+        **kwargs,
+    )
+
+
+def test_get_request_payload_preserves_thinking_and_forces_reasoning_split():
+    model = _make_model(extra_body={"thinking": {"type": "disabled"}})
+
+    payload = model._get_request_payload([HumanMessage(content="hello")])
+
+    assert payload["extra_body"]["thinking"]["type"] == "disabled"
+    assert payload["extra_body"]["reasoning_split"] is True
+
+
+def test_create_chat_result_maps_reasoning_details_to_reasoning_content():
+    model = _make_model()
+    response = {
+        "choices": [
+            {
+                "message": {
+                    "role": "assistant",
+                    "content": "最终答案",
+                    "reasoning_details": [
+                        {
+                            "type": "reasoning.text",
+                            "id": "reasoning-text-1",
+                            "format": "MiniMax-response-v1",
+                            "index": 0,
+                            "text": "先分析问题，再给出答案。",
+                        }
+                    ],
+                },
+                "finish_reason": "stop",
+            }
+        ],
+        "model": "MiniMax-M2.5",
+    }
+
+    result = model._create_chat_result(response)
+    message = result.generations[0].message
+
+    assert message.content == "最终答案"
+    assert message.additional_kwargs["reasoning_content"] == "先分析问题，再给出答案。"
+    assert result.generations[0].text == "最终答案"
+
+
+def test_create_chat_result_strips_inline_think_tags():
+    model = _make_model()
+    response = {
+        "choices": [
+            {
+                "message": {
+                    "role": "assistant",
+                    "content": "<think>\n这是思考过程。\n</think>\n\n真正回答。",
+                },
+                "finish_reason": "stop",
+            }
+        ],
+        "model": "MiniMax-M2.5",
+    }
+
+    result = model._create_chat_result(response)
+    message = result.generations[0].message
+
+    assert message.content == "真正回答。"
+    assert message.additional_kwargs["reasoning_content"] == "这是思考过程。"
+    assert result.generations[0].text == "真正回答。"
+
+
+def test_convert_chunk_to_generation_chunk_preserves_reasoning_deltas():
+    model = _make_model()
+    first = model._convert_chunk_to_generation_chunk(
+        {
+            "choices": [
+                {
+                    "delta": {
+                        "role": "assistant",
+                        "content": "",
+                        "reasoning_details": [
+                            {
+                                "type": "reasoning.text",
+                                "id": "reasoning-text-1",
+                                "format": "MiniMax-response-v1",
+                                "index": 0,
+                                "text": "The user",
+                            }
+                        ],
+                    }
+                }
+            ]
+        },
+        AIMessageChunk,
+        {},
+    )
+    second = model._convert_chunk_to_generation_chunk(
+        {
+            "choices": [
+                {
+                    "delta": {
+                        "content": "",
+                        "reasoning_details": [
+                            {
+                                "type": "reasoning.text",
+                                "id": "reasoning-text-1",
+                                "format": "MiniMax-response-v1",
+                                "index": 0,
+                                "text": " asks.",
+                            }
+                        ],
+                    }
+                }
+            ]
+        },
+        AIMessageChunk,
+        {},
+    )
+    answer = model._convert_chunk_to_generation_chunk(
+        {
+            "choices": [
+                {
+                    "delta": {
+                        "content": "最终答案",
+                    },
+                    "finish_reason": "stop",
+                }
+            ],
+            "model": "MiniMax-M2.5",
+        },
+        AIMessageChunk,
+        {},
+    )
+
+    assert first is not None
+    assert second is not None
+    assert answer is not None
+
+    combined = first.message + second.message + answer.message
+
+    assert combined.additional_kwargs["reasoning_content"] == "The user asks."
+    assert combined.content == "最终答案"