feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/arch/bot-http/__tests__/global-interceptor.test.ts

@@ -0,0 +1,122 @@
+/*
+ * 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 MockAdapter from 'axios-mock-adapter';
+
+vi.mock('@coze-arch/logger', () => ({
+  logger: {
+    info: vi.fn(),
+    persist: {
+      error: vi.fn(),
+    },
+  },
+}));
+
+vi.mock('../src/eventbus', () => ({
+  emitAPIErrorEvent: vi.fn(),
+  APIErrorEvent: {
+    UNAUTHORIZED: 'unauthorized',
+    COUNTRY_RESTRICTED: 'countryRestricted',
+    COZE_TOKEN_INSUFFICIENT: 'cozeTokenInsufficient',
+  },
+}));
+vi.mock('../src/api-error', async () => {
+  const actual = (await vi.importActual('../src/api-error')) as any;
+
+  return {
+    ...actual,
+    reportHttpError: vi.fn(),
+  };
+});
+
+vi.mock('@coze-arch/web-context', () => ({
+  redirect: vi.fn(),
+}));
+
+describe('globalRequestInterceptor', () => {
+  beforeEach(() => {
+    vi.resetModules();
+    vi.clearAllMocks();
+  });
+
+  it('should run intercept logic', async () => {
+    const { addGlobalRequestInterceptor, axiosInstance } =
+      await vi.importActual('../src/axios');
+    // This sets the mock adapter on the default instance
+    const mock = new MockAdapter(axiosInstance);
+
+    addGlobalRequestInterceptor(config => {
+      config.headers.set('x-tt-foo', 'bar');
+      return config;
+    });
+
+    mock.onGet('/users').reply(200, {
+      code: 0,
+      data: { users: [{ id: 1, name: 'John Smith' }] },
+    });
+
+    const response = await axiosInstance.get('/users');
+
+    expect(response.config.headers['x-tt-foo']).toEqual('bar');
+  });
+
+  it('run extra interceptor logic', async () => {
+    const { addGlobalResponseInterceptor, axiosInstance } =
+      await vi.importActual('../src/axios');
+    const mock = new MockAdapter(axiosInstance);
+    const removeInterceptor = addGlobalResponseInterceptor(obj => {
+      obj.data.data.oh = 2;
+      return obj;
+    });
+
+    mock.onGet('/oh').reply(200, {
+      code: 0,
+      data: { oh: 1 },
+    });
+    const response = await axiosInstance.get('/oh');
+    expect(response.data.data.oh).toBe(2);
+
+    removeInterceptor();
+    const response2 = await axiosInstance.get('/oh');
+    expect(response2.data.data.oh).toBe(1);
+  });
+
+  it('should support remove interceptors', async () => {
+    const {
+      addGlobalRequestInterceptor,
+      removeGlobalRequestInterceptor,
+      axiosInstance,
+    } = await vi.importActual('../src/axios');
+    // This sets the mock adapter on the default instance
+
+    const mock = new MockAdapter(axiosInstance);
+    const id = addGlobalRequestInterceptor(config => {
+      config.headers.set('x-tt-foo', 'bar');
+      console.log('wfe', 'wefe');
+      return config;
+    });
+    removeGlobalRequestInterceptor(id);
+
+    mock.onGet('/users').reply(200, {
+      code: 0,
+      data: { users: [{ id: 1, name: 'John Smith' }] },
+    });
+
+    const response = await axiosInstance.get('/users');
+
+    expect(response.config.headers['x-tt-foo']).toBeUndefined();
+  });
+});