feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/arch/bot-space-api/__tests__/index.test.ts

@@ -0,0 +1,94 @@
+/*
+ * 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 { globalVars } from '@coze-arch/web-context';
+import { useSpaceStore } from '@coze-arch/bot-studio-store';
+import { CustomError } from '@coze-arch/bot-error';
+import { DeveloperApi } from '@coze-arch/bot-api';
+
+import { SpaceApi } from '../src/index';
+
+vi.mock('@coze-arch/bot-api', () => ({
+  DeveloperApi: {
+    CreateWorkFlow: vi.fn(),
+    CreateWorkflowV2: vi.fn(),
+    WorkflowListV2: vi.fn(),
+    WorkFlowList: vi.fn(),
+    ExecuteDraftBot: vi.fn(),
+    QueryCardList: vi.fn(),
+  },
+}));
+
+vi.mock('@coze-arch/bot-studio-store', () => ({
+  useSpaceStore: {
+    getState: vi.fn().mockReturnValue({ getSpaceId: vi.fn() }),
+  },
+}));
+
+vi.mock('@coze-arch/report-events', () => ({ REPORT_EVENTS: { a: 'a' } }));
+vi.mock('@coze-arch/web-context', () => ({
+  globalVars: { LAST_EXECUTE_ID: undefined },
+}));
+vi.mock('@coze-arch/bot-error', () => ({ CustomError: vi.fn() }));
+vi.mock('axios', () => ({ default: { defaults: { transformResponse: [] } } }));
+vi.mock('../src/space-api-v2', () => ({ SpaceApiV2: vi.fn() }));
+
+describe('space api spec', () => {
+  vi.clearAllMocks();
+
+  it('should rewrite space id when calling ExecuteDraftBot', async () => {
+    useSpaceStore.getState().getSpaceId.mockReturnValue('test-id');
+
+    await SpaceApi.ExecuteDraftBot();
+    expect(globalVars.LAST_EXECUTE_ID).toBe(undefined);
+    expect(DeveloperApi.ExecuteDraftBot).toBeCalled();
+    expect(DeveloperApi.ExecuteDraftBot.mock.calls[0][0]).toMatchObject({
+      space_id: 'test-id',
+    });
+
+    const axiosConfig = DeveloperApi.ExecuteDraftBot.mock.calls[0][1];
+    expect(axiosConfig.transformResponse[0]).toBeTypeOf('function');
+    axiosConfig.transformResponse[0]({}, { 'x-tt-logid': 'test log id' });
+    expect(globalVars.LAST_EXECUTE_ID).toBe('test log id');
+  });
+
+  it('should call developer api directly', async () => {
+    useSpaceStore.getState().getSpaceId.mockReturnValue('test-id');
+
+    await SpaceApi.QueryCardList({});
+    expect(DeveloperApi.QueryCardList).toBeCalled();
+    expect(DeveloperApi.QueryCardList.mock.calls[0][0]).toMatchObject({
+      space_id: 'test-id',
+    });
+  });
+
+  it('should rename WorkflowList to WorkflowListV2', async () => {
+    useSpaceStore.getState().getSpaceId.mockReturnValue('test-id');
+
+    await SpaceApi.WorkFlowList({});
+    await SpaceApi.CreateWorkFlow({});
+    expect(DeveloperApi.WorkflowListV2).toBeCalled();
+    expect(DeveloperApi.WorkflowListV2.mock.calls[0][0]).toMatchObject({
+      space_id: 'test-id',
+    });
+    expect(DeveloperApi.CreateWorkflowV2).toBeCalled();
+  });
+
+  it('should throw custom error when calling not exits func', async () => {
+    await expect(() => SpaceApi['func not exits']()).toThrowError();
+    expect(CustomError).toBeCalled();
+  });
+});

frontend/packages/arch/bot-space-api/__tests__/space-api-v2.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 { useSpaceStore } from '@coze-arch/bot-studio-store';
+import { CustomError } from '@coze-arch/bot-error';
+import { PlaygroundApi } from '@coze-arch/bot-api';
+
+import { SpaceApiV2 } from '../src/space-api-v2';
+
+vi.mock('@coze-arch/bot-api', () => ({
+  DeveloperApi: {
+    InviteMemberLink: vi.fn(),
+    SearchMember: vi.fn(),
+  },
+  PlaygroundApi: {
+    InviteMemberLinkV2: vi.fn(),
+    SearchMemberV2: vi.fn(),
+  },
+}));
+
+vi.mock('@coze-arch/bot-studio-store', () => ({
+  useSpaceStore: {
+    getState: vi.fn().mockReturnValue({ getSpaceId: vi.fn() }),
+  },
+}));
+
+vi.mock('@coze-arch/report-events', () => ({ REPORT_EVENTS: { a: 'a' } }));
+vi.mock('@coze-arch/bot-error', () => ({ CustomError: vi.fn() }));
+vi.mock('axios', () => ({ default: { defaults: { transformResponse: [] } } }));
+
+describe('space api spec', () => {
+  vi.clearAllMocks();
+
+  it('should rewrite space id when calling SearchMemberV2 and fg true', async () => {
+    useSpaceStore.getState().getSpaceId.mockReturnValue('test-id');
+
+    await SpaceApiV2.SearchMemberV2({ search_list: ['name'] });
+
+    expect(PlaygroundApi.SearchMemberV2).toBeCalled();
+    expect(PlaygroundApi.SearchMemberV2.mock.calls[0][0]).toMatchObject({
+      space_id: 'test-id',
+    });
+  });
+
+  it('should throw custom error when calling not exits func', async () => {
+    await expect(() => SpaceApiV2['func not exits']()).toThrowError();
+    expect(CustomError).toBeCalled();
+  });
+});