feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/agent-ide/bot-input-length-limit/__tests__/services.test.ts

@@ -0,0 +1,208 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+
+import {
+  type BotInputLengthConfig,
+  type WorkInfoOnboardingContent,
+} from '../src/services/type';
+import { BotInputLengthService, botInputLengthService } from '../src/services';
+
+// 模拟 SuggestedQuestionsShowMode 枚举
+enum SuggestedQuestionsShowMode {
+  Random = 0,
+  All = 1,
+}
+
+// 模拟配置
+const mockConfig: BotInputLengthConfig = {
+  botName: 10,
+  botDescription: 100,
+  onboarding: 50,
+  onboardingSuggestion: 20,
+  suggestionPrompt: 200,
+  projectName: 10,
+  projectDescription: 100,
+};
+
+// 模拟获取配置的函数
+const mockGetConfig = vi.fn().mockReturnValue(mockConfig);
+
+describe('BotInputLengthService', () => {
+  let service: BotInputLengthService;
+
+  beforeEach(() => {
+    // 重置模拟
+    vi.clearAllMocks();
+    // 创建服务实例
+    service = new BotInputLengthService(mockGetConfig);
+  });
+
+  describe('getInputLengthLimit', () => {
+    it('应该返回指定字段的长度限制', () => {
+      expect(service.getInputLengthLimit('botName')).toBe(10);
+      expect(service.getInputLengthLimit('botDescription')).toBe(100);
+      expect(service.getInputLengthLimit('onboarding')).toBe(50);
+      expect(service.getInputLengthLimit('onboardingSuggestion')).toBe(20);
+      expect(service.getInputLengthLimit('suggestionPrompt')).toBe(200);
+      expect(service.getInputLengthLimit('projectName')).toBe(10);
+      expect(service.getInputLengthLimit('projectDescription')).toBe(100);
+
+      // 验证配置获取函数被调用
+      expect(mockGetConfig).toHaveBeenCalledTimes(7);
+    });
+  });
+
+  describe('getValueLength', () => {
+    it('应该返回字符串的字形簇数量', () => {
+      // 普通字符串
+      expect(service.getValueLength('hello')).toBe(5);
+
+      // 包含表情符号的字符串（表情符号算作一个字形簇）
+      expect(service.getValueLength('hi😊')).toBe(3);
+
+      // 包含组合字符的字符串
+      expect(service.getValueLength('café')).toBe(4);
+
+      // 空字符串
+      expect(service.getValueLength('')).toBe(0);
+
+      // undefined
+      expect(service.getValueLength(undefined)).toBe(0);
+    });
+  });
+
+  describe('sliceStringByMaxLength', () => {
+    it('应该根据字段限制截取字符串', () => {
+      // 字符串长度小于限制
+      expect(
+        service.sliceStringByMaxLength({ value: 'hello', field: 'botName' }),
+      ).toBe('hello');
+
+      // 字符串长度等于限制
+      expect(
+        service.sliceStringByMaxLength({
+          value: '1234567890',
+          field: 'botName',
+        }),
+      ).toBe('1234567890');
+
+      // 字符串长度大于限制
+      expect(
+        service.sliceStringByMaxLength({
+          value: '12345678901234567890',
+          field: 'botName',
+        }),
+      ).toBe('1234567890');
+
+      // 包含表情符号的字符串
+      expect(
+        service.sliceStringByMaxLength({
+          value: 'hello😊world',
+          field: 'botName',
+        }),
+      ).toBe('hello😊worl');
+
+      // 验证配置获取函数被调用
+      expect(mockGetConfig).toHaveBeenCalledTimes(4);
+    });
+  });
+
+  describe('sliceWorkInfoOnboardingByMaxLength', () => {
+    it('应该截取工作信息的开场白和建议问题', () => {
+      const workInfo: WorkInfoOnboardingContent = {
+        prologue:
+          'This is a very long prologue that exceeds the limit of 50 characters and should be truncated',
+        suggested_questions: [
+          {
+            id: '1',
+            content:
+              'This is a very long suggested question that exceeds the limit',
+            highlight: true,
+          },
+          { id: '2', content: 'Short question' },
+          {
+            id: '3',
+            content:
+              'Another very long suggested question that should be truncated',
+            highlight: false,
+          },
+        ],
+        suggested_questions_show_mode: SuggestedQuestionsShowMode.All,
+      };
+
+      const result = service.sliceWorkInfoOnboardingByMaxLength(workInfo);
+
+      // 验证开场白被截取
+      expect(result.prologue).toBe(
+        'This is a very long prologue that exceeds the limi',
+      );
+      expect(result.prologue.length).toBeLessThanOrEqual(50);
+
+      // 验证建议问题被截取
+      expect(result.suggested_questions[0]?.content).toBe(
+        'This is a very long ',
+      );
+      expect(result.suggested_questions[0]?.content.length).toBeLessThanOrEqual(
+        20,
+      );
+      expect(result.suggested_questions[0]?.id).toBe('1');
+      expect(result.suggested_questions[0]?.highlight).toBe(true);
+
+      expect(result.suggested_questions[1]?.content).toBe('Short question');
+      expect(result.suggested_questions[1]?.id).toBe('2');
+
+      expect(result.suggested_questions[2]?.content).toBe(
+        'Another very long su',
+      );
+      expect(result.suggested_questions[2]?.content.length).toBeLessThanOrEqual(
+        20,
+      );
+      expect(result.suggested_questions[2]?.id).toBe('3');
+      expect(result.suggested_questions[2]?.highlight).toBe(false);
+
+      // 验证显示模式保持不变
+      expect(result.suggested_questions_show_mode).toBe(
+        SuggestedQuestionsShowMode.All,
+      );
+    });
+
+    it('应该处理空的工作信息', () => {
+      const workInfo: WorkInfoOnboardingContent = {
+        prologue: '',
+        suggested_questions: [],
+        suggested_questions_show_mode: SuggestedQuestionsShowMode.Random,
+      };
+
+      const result = service.sliceWorkInfoOnboardingByMaxLength(workInfo);
+
+      expect(result.prologue).toBe('');
+      expect(result.suggested_questions).toEqual([]);
+      expect(result.suggested_questions_show_mode).toBe(
+        SuggestedQuestionsShowMode.Random,
+      );
+    });
+  });
+});
+
+// 测试导出的单例
+describe('botInputLengthService', () => {
+  it('应该导出一个 BotInputLengthService 的实例', () => {
+    // 验证导出的单例是 BotInputLengthService 的实例
+    expect(botInputLengthService).toBeInstanceOf(BotInputLengthService);
+  });
+});