@zjex/git-workflow 0.2.24 → 0.3.2

package/tests/ai-service.test.ts

@@ -121,7 +121,7 @@ describe("AI Service 模块测试", () => {
         provider: "github",
         apiKey: "test-key",
         language: "zh-CN",
-        maxTokens: 200,
+        // 不设置 maxTokens，让它自动计算
       },
     };
 
@@ -310,7 +310,7 @@ describe("AI Service 模块测试", () => {
     });
 
     it("应该截断过长的 diff", async () => {
-      const longDiff = "a".repeat(5000);
+      const longDiff = "a".repeat(7000); // 超过详细模式的 6000 限制
       vi.mocked(execOutput).mockReturnValue(longDiff);
 
       const mockFetch = vi.mocked(fetch);
@@ -343,6 +343,7 @@ describe("AI Service 模块测试", () => {
         aiCommit: {
           ...mockConfig.aiCommit!,
           language: "en-US" as const,
+          detailedDescription: false, // 使用简洁模式测试语言
         },
       };
 
@@ -437,6 +438,240 @@ describe("AI Service 模块测试", () => {
 
       expect(body.max_tokens).toBe(500);
     });
+
+    it("默认应该启用详细描述模式", async () => {
+      const defaultConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          // 不设置 detailedDescription 和 maxTokens，测试默认行为
+        },
+      };
+      // 移除 maxTokens 设置
+      delete defaultConfig.aiCommit!.maxTokens;
+
+      vi.mocked(execOutput).mockReturnValue("diff --git a/file.ts b/file.ts");
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat(auth): 添加用户登录功能\n\n- 实现用户名密码登录接口\n- 添加登录状态验证中间件",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      const result = await generateAICommitMessage(defaultConfig);
+
+      expect(result).toContain("feat(auth): 添加用户登录功能");
+      expect(result).toContain("- 实现用户名密码登录接口");
+
+      // 验证使用了详细描述的 prompt
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+      const prompt = body.messages[0].content;
+
+      expect(prompt).toContain("详细描述：列出主要修改点");
+      expect(body.max_tokens).toBe(400); // 详细模式的 token 数
+    });
+
+    it("详细描述模式应该生成包含修改点的 commit message", async () => {
+      const detailedConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          detailedDescription: true,
+        },
+      };
+
+      vi.mocked(execOutput).mockReturnValue("diff --git a/file.ts b/file.ts");
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat(auth): 添加用户登录功能\n\n- 实现用户名密码登录接口\n- 添加登录状态验证中间件\n- 完善登录错误处理逻辑",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      const result = await generateAICommitMessage(detailedConfig);
+
+      expect(result).toContain("feat(auth): 添加用户登录功能");
+      expect(result).toContain("- 实现用户名密码登录接口");
+      expect(result).toContain("- 添加登录状态验证中间件");
+      expect(result).toContain("- 完善登录错误处理逻辑");
+
+      // 验证 prompt 包含详细描述的指令
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+      const prompt = body.messages[0].content;
+
+      expect(prompt).toContain("详细描述：列出主要修改点");
+      expect(prompt).toContain("以 \"- \" 开头");
+    });
+
+    it("详细描述模式应该使用更大的 maxTokens", async () => {
+      const detailedConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          // 不设置 detailedDescription 和 maxTokens，使用默认值
+        },
+      };
+      // 移除 maxTokens 设置
+      delete detailedConfig.aiCommit!.maxTokens;
+
+      vi.mocked(execOutput).mockReturnValue("diff --git a/file.ts b/file.ts");
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat: add feature\n\n- change 1\n- change 2",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      await generateAICommitMessage(detailedConfig);
+
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+
+      // 默认启用详细描述模式，应该使用 400 tokens
+      expect(body.max_tokens).toBe(400);
+    });
+
+    it("详细描述模式应该允许更长的 diff", async () => {
+      const detailedConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          // 不设置 detailedDescription，使用默认值 true
+        },
+      };
+
+      const longDiff = "a".repeat(5500); // 超过简洁模式的 4000 限制，但在详细模式的 6000 限制内
+      vi.mocked(execOutput).mockReturnValue(longDiff);
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat: update",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      await generateAICommitMessage(detailedConfig);
+
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+      const prompt = body.messages[0].content;
+
+      // 在详细模式下，5500 字符的 diff 不应该被截断
+      expect(prompt).toContain(longDiff);
+      expect(prompt).not.toContain("...");
+    });
+
+    it("简洁模式应该生成简短的 commit message", async () => {
+      const simpleConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          detailedDescription: false,
+        },
+      };
+
+      vi.mocked(execOutput).mockReturnValue("diff --git a/file.ts b/file.ts");
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat(auth): 添加用户登录功能",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      const result = await generateAICommitMessage(simpleConfig);
+
+      expect(result).toBe("feat(auth): 添加用户登录功能");
+
+      // 验证 prompt 不包含详细描述的指令
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+      const prompt = body.messages[0].content;
+
+      expect(prompt).not.toContain("详细描述：列出主要修改点");
+      expect(prompt).toContain("只返回一条 commit message");
+    });
+
+    it("英文详细描述模式应该使用英文指令", async () => {
+      const enDetailedConfig = {
+        ...mockConfig,
+        aiCommit: {
+          ...mockConfig.aiCommit!,
+          language: "en-US" as const,
+          detailedDescription: true,
+        },
+      };
+
+      vi.mocked(execOutput).mockReturnValue("diff --git a/file.ts b/file.ts");
+
+      const mockFetch = vi.mocked(fetch);
+      mockFetch.mockResolvedValue({
+        ok: true,
+        json: async () => ({
+          choices: [
+            {
+              message: {
+                content: "feat(auth): add user login functionality\n\n- Implement username/password login API\n- Add login status validation middleware",
+              },
+            },
+          ],
+        }),
+      } as Response);
+
+      const result = await generateAICommitMessage(enDetailedConfig);
+
+      expect(result).toContain("feat(auth): add user login functionality");
+      expect(result).toContain("- Implement username/password login API");
+
+      // 验证 prompt 使用英文指令
+      const callArgs = mockFetch.mock.calls[0][1] as RequestInit;
+      const body = JSON.parse(callArgs.body as string);
+      const prompt = body.messages[0].content;
+
+      expect(prompt).toContain("Detailed description: List main changes");
+      expect(prompt).toContain("starting with \"- \"");
+      expect(prompt).not.toContain("详细描述");
+    });
   });
 
   describe("Ollama 特殊情况", () => {