feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/common/chat-area/utils/__tests__/rate-limit.test.ts

@@ -0,0 +1,62 @@
+/*
+ * 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 { expect, it, vi } from 'vitest';
+
+import { RateLimit } from '../src/rate-limit';
+
+it('limit rate', async () => {
+  vi.useFakeTimers();
+  const request = vi.fn();
+  const limiter = new RateLimit(request, {
+    limit: 3,
+    onLimitDelay: 1000,
+    timeWindow: 5000,
+  });
+  for (const i of [1, 2, 3, 4, 5]) {
+    limiter.invoke(i);
+  }
+  expect(request.mock.calls.length).toBe(3);
+  // 1000
+  await vi.advanceTimersByTimeAsync(1000);
+  expect(request.mock.calls.length).toBe(4);
+  // 2000
+  await vi.advanceTimersByTimeAsync(1000);
+  expect(request.mock.calls.length).toBe(5);
+  // 3000
+  await vi.advanceTimersByTimeAsync(1000);
+  limiter.invoke();
+  limiter.invoke();
+  // 3010
+  await vi.advanceTimersByTimeAsync(10);
+  expect(request.mock.calls.length).toBe(6);
+  // 4010
+  await vi.advanceTimersByTimeAsync(1000);
+  expect(request.mock.calls.length).toBe(7);
+
+  // 离开窗口
+  await vi.advanceTimersByTimeAsync(5000);
+  limiter.invoke();
+  limiter.invoke();
+  limiter.invoke();
+  expect(request.mock.calls.length).toBe(10);
+  // 进入限流
+  limiter.invoke();
+  await vi.advanceTimersByTimeAsync(100);
+  expect(request.mock.calls.length).toBe(10);
+  expect((limiter as unknown as { records: number[] }).records.length).toBe(4);
+  vi.useRealTimers();
+});