feat(tests): 添加流工具调用参数的测试用例，覆盖完整、分片和并发调用场景

Author: cuipengfei

tests/generate-content/stream-tool-call-accumulator.test.ts

@@ -0,0 +1,240 @@
+import { afterEach, expect, test, mock } from "bun:test"
+
+function asyncIterableFrom(events: Array<{ data?: string }>) {
+  return {
+    [Symbol.asyncIterator]() {
+      let i = 0
+      return {
+        next() {
+          if (i < events.length)
+            return Promise.resolve({ value: events[i++], done: false })
+          return Promise.resolve({ value: undefined, done: true })
+        },
+      }
+    },
+  }
+}
+
+afterEach(() => {
+  mock.restore()
+})
+
+test("[Stream] handles complete tool call parameters in single chunk", async () => {
+  await mock.module("~/services/copilot/create-chat-completions", () => ({
+    createChatCompletions: () =>
+      asyncIterableFrom([
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [
+              {
+                index: 0,
+                delta: {
+                  tool_calls: [
+                    {
+                      index: 0,
+                      type: "function",
+                      function: {
+                        name: "ReadFile",
+                        arguments: '{"absolute_path": "/path/to/file.txt"}',
+                      },
+                    },
+                  ],
+                },
+                finish_reason: null,
+              },
+            ],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        { data: "[DONE]" },
+      ]),
+  }))
+
+  await mock.module("~/lib/rate-limit", () => ({
+    checkRateLimit: (_: unknown) => {},
+  }))
+  const { server } = await import("~/server?stream-complete-params")
+  const res = await server.request(
+    "/v1beta/models/gemini-pro:streamGenerateContent",
+    {
+      method: "POST",
+      headers: { "content-type": "application/json" },
+      body: JSON.stringify({
+        contents: [{ role: "user", parts: [{ text: "Read the file" }] }],
+      }),
+    },
+  )
+
+  expect(res.status).toBe(200)
+  const body = await res.text()
+  expect(
+    body.includes(
+      '"functionCall":{"name":"ReadFile","args":{"absolute_path":"/path/to/file.txt"}}',
+    ),
+  ).toBe(true)
+})
+
+test("[Stream] handles fragmented tool call parameters across multiple chunks", async () => {
+  await mock.module("~/services/copilot/create-chat-completions", () => ({
+    createChatCompletions: () =>
+      asyncIterableFrom([
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [
+              {
+                index: 0,
+                delta: {
+                  tool_calls: [
+                    {
+                      index: 0,
+                      type: "function",
+                      function: { name: "ReadFile", arguments: '{"absolu' },
+                    },
+                  ],
+                },
+                finish_reason: null,
+              },
+            ],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [
+              {
+                index: 0,
+                delta: {
+                  tool_calls: [
+                    {
+                      index: 0,
+                      type: "function",
+                      function: { arguments: 'te_path": "/file.txt"}' },
+                    },
+                  ],
+                },
+                finish_reason: null,
+              },
+            ],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        { data: "[DONE]" },
+      ]),
+  }))
+
+  await mock.module("~/lib/rate-limit", () => ({
+    checkRateLimit: (_: unknown) => {},
+  }))
+  const { server } = await import("~/server?stream-fragmented-params")
+  const res = await server.request(
+    "/v1beta/models/gemini-pro:streamGenerateContent",
+    {
+      method: "POST",
+      headers: { "content-type": "application/json" },
+      body: JSON.stringify({
+        contents: [{ role: "user", parts: [{ text: "Read the file" }] }],
+      }),
+    },
+  )
+
+  expect(res.status).toBe(200)
+  const body = await res.text()
+  expect(
+    body.includes(
+      '"functionCall":{"name":"ReadFile","args":{"absolute_path":"/file.txt"}}',
+    ),
+  ).toBe(true)
+})
+
+test("[Stream] correctly processes multiple concurrent tool calls", async () => {
+  await mock.module("~/services/copilot/create-chat-completions", () => ({
+    createChatCompletions: () =>
+      asyncIterableFrom([
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [
+              {
+                index: 0,
+                delta: {
+                  tool_calls: [
+                    {
+                      index: 0,
+                      type: "function",
+                      function: {
+                        name: "ReadFile",
+                        arguments: '{"path": "/read.txt"}',
+                      },
+                    },
+                    {
+                      index: 1,
+                      type: "function",
+                      function: {
+                        name: "WriteFile",
+                        arguments: '{"path": "/write.txt", "content": "data"}',
+                      },
+                    },
+                  ],
+                },
+                finish_reason: null,
+              },
+            ],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        {
+          data: JSON.stringify({
+            id: "c1",
+            choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
+            usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          }),
+        },
+        { data: "[DONE]" },
+      ]),
+  }))
+
+  await mock.module("~/lib/rate-limit", () => ({
+    checkRateLimit: (_: unknown) => {},
+  }))
+  const { server } = await import("~/server?stream-multiple-tools")
+  const res = await server.request(
+    "/v1beta/models/gemini-pro:streamGenerateContent",
+    {
+      method: "POST",
+      headers: { "content-type": "application/json" },
+      body: JSON.stringify({
+        contents: [{ role: "user", parts: [{ text: "Read and write files" }] }],
+      }),
+    },
+  )
+
+  expect(res.status).toBe(200)
+  const body = await res.text()
+  expect(
+    body.includes(
+      '"functionCall":{"name":"ReadFile","args":{"path":"/read.txt"}}',
+    ),
+  ).toBe(true)
+  expect(
+    body.includes(
+      '"functionCall":{"name":"WriteFile","args":{"path":"/write.txt","content":"data"}}',
+    ),
+  ).toBe(true)
+})