更新

wechat_rpa/test_dsh_agent.py

@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+from __future__ import annotations
+
+import tempfile
+import unittest
+from pathlib import Path
+from unittest import mock
+
+import dsh_agent
+
+
+class DeepSeekDeskTest(unittest.TestCase):
+    def setUp(self):
+        self.temp = tempfile.TemporaryDirectory()
+        self.desk = dsh_agent.DeepSeekDesk(Path(self.temp.name))
+
+    def tearDown(self):
+        self.desk.close()
+        self.temp.cleanup()
+
+    def test_create_session_and_reuse_session_id(self):
+        first = self.desk.create_session()
+        self.assertTrue(first.id.startswith("desk-"))
+        self.assertEqual(first.title, "新对话")
+        with mock.patch.object(
+            self.desk,
+            "_run_model",
+            return_value=dsh_agent.RunOutcome(
+                session_id=first.id,
+                text="已经帮您拟好回复。",
+                backend=dsh_agent.BACKEND_COMPAT,
+                finish_reason="completed",
+            ),
+        ):
+            result = self.desk.run("帮我回客户说下午再联系", session_id=first.id)
+        self.assertEqual(result.session_id, first.id)
+        stored = self.desk.get_session(first.id)
+        self.assertEqual(stored.title, "帮我回客户说下午再联系")
+        self.assertEqual(len(stored.messages), 2)
+        self.assertEqual(stored.messages[0]["role"], "user")
+        self.assertEqual(stored.messages[1]["content"], "已经帮您拟好回复。")
+
+    def test_independent_tasks_use_new_session_ids(self):
+        one = self.desk.create_session()
+        two = self.desk.create_session()
+        self.assertNotEqual(one.id, two.id)
+
+    def test_compat_backend_sends_system_prompt_and_history(self):
+        session = self.desk.create_session()
+        captured = {}
+
+        def fake_completion(messages, tools=None):
+            captured["messages"] = messages
+            return {"content": "草稿已写好"}
+
+        with (
+            mock.patch.object(self.desk, "_start_harness", side_effect=RuntimeError("no runtime")),
+            mock.patch("ai_chat._chat_completion", side_effect=fake_completion),
+        ):
+            first = self.desk.run("客户问血糖高怎么办", session_id=session.id)
+            second = self.desk.run("那饮食呢", session_id=session.id)
+        self.assertEqual(first.backend, dsh_agent.BACKEND_COMPAT)
+        self.assertEqual(second.text, "草稿已写好")
+        roles = [item["role"] for item in captured["messages"]]
+        self.assertEqual(roles[0], "system")
+        self.assertIn("DeepSeek Agent", captured["messages"][0]["content"])
+        self.assertEqual(roles[-1], "user")
+        self.assertEqual(captured["messages"][-1]["content"], "那饮食呢")
+        self.assertTrue(any(item.get("content") == "客户问血糖高怎么办" for item in captured["messages"]))
+
+    def test_sdk_backend_reuses_harness_run(self):
+        session = self.desk.create_session()
+        harness = mock.Mock()
+        harness.run.return_value = mock.Mock(
+            final_response="SDK 回复",
+            finish_reason="completed",
+        )
+        with mock.patch.object(self.desk, "_start_harness", return_value=harness):
+            result = self.desk.run("检查这段话术", session_id=session.id)
+        self.assertEqual(result.backend, dsh_agent.BACKEND_SDK)
+        self.assertEqual(result.text, "SDK 回复")
+        harness.run.assert_called_once()
+        self.assertEqual(harness.run.call_args.kwargs["session_id"], session.id)
+
+    def test_empty_prompt_is_rejected(self):
+        with self.assertRaises(ValueError):
+            self.desk.run("   ")
+
+
+if __name__ == "__main__":
+    unittest.main()