feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/devops/mockset-manage/__tests__/hooks.test.ts

@@ -0,0 +1,130 @@
+/*
+ * 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 { renderHook } from '@testing-library/react-hooks';
+import {
+  type BizCtx,
+  ComponentType,
+  TrafficScene,
+} from '@coze-arch/bot-api/debugger_api';
+
+import { type BasicMockSetInfo } from '@/interface';
+import { useInitialGetEnabledMockSet } from '@/hooks/use-get-mockset';
+
+vi.mock('@/const', () => ({
+  IS_OVERSEA: true,
+  MockTrafficEnabled: {
+    DISABLE: 0,
+    ENABLE: 1,
+  },
+}));
+
+vi.mock('@coze-arch/bot-utils', () => ({
+  safeJSONParse: JSON.parse,
+}));
+
+vi.mock('@coze-arch/logger', () => ({
+  logger: {
+    createLoggerWith: vi.fn(),
+    error: vi.fn(),
+    info: vi.fn(),
+  },
+  reporter: {
+    createReporterWithPreset: vi.fn(),
+  },
+}));
+
+vi.mock('@coze-arch/bot-api', () => ({
+  debuggerApi: {
+    MGetMockSetBinding: vi.fn().mockResolvedValue({
+      code: 0,
+      msg: '',
+      mockSetBindings: [
+        {
+          mockSetID: '1',
+          mockSubject: {
+            componentID: 'tool1',
+            componentType: ComponentType.CozeTool,
+            parentComponentID: 'plugin1',
+            parentComponentType: ComponentType.CozePlugin,
+          },
+          bizCtx: {
+            bizSpaceID: '1',
+            trafficCallerID: '1',
+            trafficScene: TrafficScene.CozeSingleAgentDebug,
+            connectorID: '1',
+            connectorUID: '2',
+          },
+        },
+      ],
+      mockSetDetails: {
+        '1': {
+          id: '1',
+          name: 'test-detail',
+          mockSubject: {
+            componentID: 'tool1',
+            componentType: ComponentType.CozeTool,
+            parentComponentID: 'plugin1',
+            parentComponentType: ComponentType.CozePlugin,
+          },
+        },
+      },
+    }),
+  },
+}));
+
+describe('mock-set-hooks', () => {
+  it('fetch-mock-list', async () => {
+    const TEST_COMMON_BIZ = {
+      connectorID: '1',
+      connectorUID: '2',
+    };
+
+    const TEST_BIZ_CTX1: BizCtx = {
+      bizSpaceID: '1',
+      trafficCallerID: '1',
+      trafficScene: TrafficScene.CozeSingleAgentDebug,
+    };
+
+    const TEST_MOCK_SUBJECT = {
+      componentID: 'tool1',
+      componentType: ComponentType.CozeTool,
+      parentComponentID: 'plugin1',
+      parentComponentType: ComponentType.CozePlugin,
+    };
+
+    const singleAgentToolItem: BasicMockSetInfo = {
+      bindSubjectInfo: TEST_MOCK_SUBJECT,
+      bizCtx: {
+        ...TEST_COMMON_BIZ,
+        ...TEST_BIZ_CTX1,
+      },
+    };
+    const { result } = renderHook(() =>
+      useInitialGetEnabledMockSet({
+        bizCtx: singleAgentToolItem.bizCtx,
+        pollingInterval: 0,
+      }),
+    );
+
+    const { start } = result.current;
+
+    await start();
+    const { data } = result.current;
+    expect(data[0]?.mockSetBinding.mockSetID).toEqual('1');
+    expect(data[0]?.mockSetDetail?.id).toEqual('1');
+  });
+});

frontend/packages/devops/mockset-manage/__tests__/utils.test.ts

@@ -0,0 +1,330 @@
+/*
+ * 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 {
+  type BizCtx,
+  ComponentType,
+  TrafficScene,
+} from '@coze-arch/bot-api/debugger_api';
+
+import {
+  isRealData,
+  isCurrent,
+  isSameWorkflowTool,
+  isSameScene,
+  getPluginInfo,
+  getMockSubjectInfo,
+  getUsedScene,
+} from '@/utils';
+import { type BasicMockSetInfo } from '@/interface';
+
+vi.mock('@coze-arch/bot-utils', () => ({
+  safeJSONParse: JSON.parse,
+}));
+
+vi.mock('@/const', () => ({
+  IS_OVERSEA: true,
+  REAL_DATA_MOCKSET: {
+    id: '0',
+  },
+}));
+
+vi.mock('@coze-arch/logger', () => ({
+  logger: {
+    createLoggerWith: vi.fn(),
+  },
+}));
+
+const TEST_COMMON_BIZ = {
+  connectorID: '1',
+  connectorUID: '2',
+};
+
+const TEST_BIZ_CTX1: BizCtx = {
+  bizSpaceID: '1',
+  trafficCallerID: '1',
+  trafficScene: TrafficScene.CozeSingleAgentDebug,
+};
+
+const TEST_BIZ_CTX2: BizCtx = {
+  bizSpaceID: '1',
+  trafficCallerID: '2',
+  trafficScene: TrafficScene.CozeSingleAgentDebug,
+};
+
+const TEST_BIZ_CTX3: BizCtx = {
+  bizSpaceID: '1',
+  trafficCallerID: '1',
+  trafficScene: TrafficScene.CozeWorkflowDebug,
+};
+
+const TEST_BIZ_CTX4: BizCtx = {
+  bizSpaceID: '2',
+  trafficCallerID: '1',
+  trafficScene: TrafficScene.CozeWorkflowDebug,
+};
+
+const TEST_MOCK_SUBJECT1 = {
+  componentID: 'tool1',
+  componentType: ComponentType.CozeTool,
+  parentComponentID: 'plugin1',
+  parentComponentType: ComponentType.CozePlugin,
+};
+
+const TEST_MOCK_SUBJECT2 = {
+  componentID: 'tool2',
+  componentType: ComponentType.CozeTool,
+  parentComponentID: 'plugin1',
+  parentComponentType: ComponentType.CozePlugin,
+};
+
+const TEST_MOCK_WORKFLOW_NODE1 = {
+  componentID: 'node1',
+  componentType: ComponentType.CozeToolNode,
+  parentComponentID: 'workflow1',
+  parentComponentType: ComponentType.CozeWorkflow,
+};
+
+const TEST_MOCK_WORKFLOW_NODE2 = {
+  componentID: 'node2',
+  componentType: ComponentType.CozeToolNode,
+  parentComponentID: 'workflow1',
+  parentComponentType: ComponentType.CozeWorkflow,
+};
+
+const singleAgentToolItem1: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_SUBJECT1,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX1,
+  },
+};
+const singleAgentToolItem2: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_SUBJECT2,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX2,
+  },
+};
+
+const multiAgentToolItem: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_SUBJECT1,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX1,
+    trafficScene: TrafficScene.CozeMultiAgentDebug,
+  },
+};
+
+const workflowToolItem1: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_WORKFLOW_NODE1,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX3,
+    ext: { mockSubjectInfo: JSON.stringify(TEST_MOCK_SUBJECT1) },
+  },
+};
+const workflowToolItem2: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_WORKFLOW_NODE1,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX3,
+    ext: { mockSubjectInfo: JSON.stringify(TEST_MOCK_SUBJECT2) },
+  },
+};
+const workflowToolItem3: BasicMockSetInfo = {
+  bindSubjectInfo: TEST_MOCK_WORKFLOW_NODE2,
+  bizCtx: {
+    ...TEST_COMMON_BIZ,
+    ...TEST_BIZ_CTX3,
+    ext: { mockSubjectInfo: JSON.stringify(TEST_MOCK_SUBJECT1) },
+  },
+};
+
+describe('mock-set-utils-isRealData', () => {
+  it('real data', () => {
+    const res = isRealData({ id: '0' });
+
+    expect(res).toEqual(true);
+  });
+  it('not real data', () => {
+    const res = isRealData({ id: '123' });
+
+    expect(res).toEqual(false);
+  });
+});
+
+describe('mock-set-utils-isCurrent', () => {
+  it('same single agent mock tool', () => {
+    const res = isCurrent(singleAgentToolItem1, singleAgentToolItem1);
+
+    expect(res).toEqual(true);
+  });
+
+  it('same agent different mock tool', () => {
+    const res = isCurrent(singleAgentToolItem1, singleAgentToolItem2);
+
+    expect(res).toEqual(false);
+  });
+
+  it('different single multi agent mock tool', () => {
+    const res = isCurrent(singleAgentToolItem1, multiAgentToolItem);
+
+    expect(res).toEqual(false);
+  });
+
+  it('different single agent workflow mock tool', () => {
+    const res = isCurrent(singleAgentToolItem1, workflowToolItem1);
+
+    expect(res).toEqual(false);
+  });
+
+  it('same workflow mock tool', () => {
+    const res = isCurrent(workflowToolItem1, workflowToolItem1);
+
+    expect(res).toEqual(true);
+  });
+
+  it('different workflow mock tool', () => {
+    const res = isCurrent(workflowToolItem1, workflowToolItem2);
+
+    expect(res).toEqual(false);
+  });
+  it('different workflow mock tool', () => {
+    const res = isCurrent(workflowToolItem1, workflowToolItem2);
+
+    expect(res).toEqual(false);
+  });
+  it('different workflow mock node', () => {
+    const res = isCurrent(workflowToolItem1, workflowToolItem3);
+
+    expect(res).toEqual(false);
+  });
+});
+
+describe('mock-set-utils-isSameWorkflowTool', () => {
+  it('workflow same mock subject', () => {
+    const res = isSameWorkflowTool(
+      workflowToolItem1.bizCtx.ext?.mockSubjectInfo || '',
+      workflowToolItem1.bizCtx.ext?.mockSubjectInfo || '',
+    );
+
+    expect(res).toEqual(true);
+  });
+
+  it('different mock subject', () => {
+    const res = isSameWorkflowTool(
+      workflowToolItem1.bizCtx.ext?.mockSubjectInfo || '',
+      workflowToolItem2.bizCtx.ext?.mockSubjectInfo || '',
+    );
+
+    expect(res).toEqual(false);
+  });
+});
+
+describe('mock-set-utils-isSameScene', () => {
+  it('isSameScene', () => {
+    const res = isSameScene(TEST_BIZ_CTX1, TEST_BIZ_CTX1);
+
+    expect(res).toEqual(true);
+  });
+  it('different caller id', () => {
+    const res = isSameScene(TEST_BIZ_CTX1, TEST_BIZ_CTX2);
+
+    expect(res).toEqual(false);
+  });
+  it('different used scene', () => {
+    const res = isSameScene(TEST_BIZ_CTX1, TEST_BIZ_CTX3);
+
+    expect(res).toEqual(false);
+  });
+  it('different space id', () => {
+    const res = isSameScene(TEST_BIZ_CTX1, TEST_BIZ_CTX4);
+
+    expect(res).toEqual(false);
+  });
+});
+
+describe('mock-set-utils-getPluginInfo', () => {
+  it('get agent tool info', () => {
+    const res = getPluginInfo(TEST_BIZ_CTX1, TEST_MOCK_SUBJECT1);
+
+    expect(res.pluginID).toEqual(TEST_MOCK_SUBJECT1.parentComponentID);
+    expect(res.toolID).toEqual(TEST_MOCK_SUBJECT1.componentID);
+    expect(res.spaceID).toEqual(TEST_BIZ_CTX1.bizSpaceID);
+  });
+
+  it('get workflow tool info', () => {
+    const res = getPluginInfo(
+      workflowToolItem1.bizCtx,
+      workflowToolItem1.bindSubjectInfo,
+    );
+
+    expect(res.pluginID).toEqual(TEST_MOCK_SUBJECT1.parentComponentID);
+    expect(res.toolID).toEqual(TEST_MOCK_SUBJECT1.componentID);
+    expect(res.spaceID).toEqual(TEST_BIZ_CTX3.bizSpaceID);
+  });
+});
+
+describe('mock-set-utils-getMockSubjectInfo', () => {
+  it('get workflow mock subject info', () => {
+    const res = getMockSubjectInfo(
+      workflowToolItem1.bizCtx,
+      workflowToolItem1.bindSubjectInfo,
+    );
+
+    expect(res.componentID).toEqual(TEST_MOCK_SUBJECT1.componentID);
+    expect(res.componentType).toEqual(TEST_MOCK_SUBJECT1.componentType);
+    expect(res.parentComponentID).toEqual(TEST_MOCK_SUBJECT1.parentComponentID);
+    expect(res.parentComponentType).toEqual(
+      TEST_MOCK_SUBJECT1.parentComponentType,
+    );
+  });
+
+  it('get agent subject info', () => {
+    const res = getMockSubjectInfo(
+      singleAgentToolItem1.bizCtx,
+      singleAgentToolItem1.bindSubjectInfo,
+    );
+
+    expect(res.componentID).toEqual(TEST_MOCK_SUBJECT1.componentID);
+    expect(res.componentType).toEqual(TEST_MOCK_SUBJECT1.componentType);
+    expect(res.parentComponentID).toEqual(TEST_MOCK_SUBJECT1.parentComponentID);
+    expect(res.parentComponentType).toEqual(
+      TEST_MOCK_SUBJECT1.parentComponentType,
+    );
+  });
+});
+
+describe('mock-set-utils-getMockSubjectInfo', () => {
+  it('get single agent', () => {
+    const res = getUsedScene(TrafficScene.CozeSingleAgentDebug);
+
+    expect(res).toEqual('bot');
+  });
+
+  it('get multi agent', () => {
+    const res = getUsedScene(TrafficScene.CozeMultiAgentDebug);
+
+    expect(res).toEqual('agent');
+  });
+
+  it('get workflow', () => {
+    const res = getUsedScene(TrafficScene.CozeWorkflowDebug);
+
+    expect(res).toEqual('flow');
+  });
+});