mirror of
https://github.com/labring/FastGPT.git
synced 2025-10-15 07:31:19 +00:00
Add unit tests for utility functions in web/core/app/utils (#4681)
Co-authored-by: gru-agent[bot] <185149714+gru-agent[bot]@users.noreply.github.com>
This commit is contained in:
105
test/cases/web/core/app/utils.test.ts
Normal file
105
test/cases/web/core/app/utils.test.ts
Normal file
@@ -0,0 +1,105 @@
|
||||
import { describe, it, expect } from 'vitest';
|
||||
import {
|
||||
form2AppWorkflow,
|
||||
filterSensitiveFormData,
|
||||
getAppQGuideCustomURL
|
||||
} from '@/web/core/app/utils';
|
||||
import { getDefaultAppForm } from '@fastgpt/global/core/app/utils';
|
||||
import { FlowNodeTypeEnum } from '@fastgpt/global/core/workflow/node/constant';
|
||||
import { NodeInputKeyEnum } from '@fastgpt/global/core/workflow/constants';
|
||||
import type { AppSchema } from '@fastgpt/global/core/app/type';
|
||||
|
||||
describe('web/core/app/utils', () => {
|
||||
const mockT = (text: string) => text;
|
||||
|
||||
describe('form2AppWorkflow', () => {
|
||||
it('should generate simple chat workflow', () => {
|
||||
const form = getDefaultAppForm();
|
||||
const result = form2AppWorkflow(form, mockT);
|
||||
|
||||
expect(result.nodes).toHaveLength(3);
|
||||
expect(result.edges).toHaveLength(1);
|
||||
expect(result.chatConfig).toBeDefined();
|
||||
});
|
||||
|
||||
it('should generate dataset workflow', () => {
|
||||
const form = getDefaultAppForm();
|
||||
form.dataset.datasets = ['dataset1'];
|
||||
const result = form2AppWorkflow(form, mockT);
|
||||
|
||||
expect(result.nodes).toHaveLength(4);
|
||||
expect(result.edges).toHaveLength(2);
|
||||
});
|
||||
|
||||
it('should generate tools workflow', () => {
|
||||
const form = getDefaultAppForm();
|
||||
form.selectedTools = [
|
||||
{
|
||||
id: 'tool1',
|
||||
name: 'Tool 1',
|
||||
flowNodeType: FlowNodeTypeEnum.tools,
|
||||
inputs: [],
|
||||
outputs: []
|
||||
}
|
||||
];
|
||||
const result = form2AppWorkflow(form, mockT);
|
||||
|
||||
expect(result.nodes.length).toBeGreaterThan(3);
|
||||
expect(result.edges.length).toBeGreaterThan(1);
|
||||
});
|
||||
});
|
||||
|
||||
describe('filterSensitiveFormData', () => {
|
||||
it('should filter sensitive data', () => {
|
||||
const form = getDefaultAppForm();
|
||||
form.dataset.datasets = ['sensitive'];
|
||||
|
||||
const result = filterSensitiveFormData(form);
|
||||
|
||||
expect(result.dataset).toEqual(getDefaultAppForm().dataset);
|
||||
expect(result).not.toEqual(form);
|
||||
});
|
||||
});
|
||||
|
||||
describe('getAppQGuideCustomURL', () => {
|
||||
it('should get custom URL from app detail', () => {
|
||||
const appDetail = {
|
||||
modules: [
|
||||
{
|
||||
flowNodeType: FlowNodeTypeEnum.systemConfig,
|
||||
inputs: [
|
||||
{
|
||||
key: NodeInputKeyEnum.chatInputGuide,
|
||||
value: {
|
||||
customUrl: 'https://example.com'
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
} as AppSchema;
|
||||
|
||||
const result = getAppQGuideCustomURL(appDetail);
|
||||
expect(result).toBe('https://example.com');
|
||||
});
|
||||
|
||||
it('should return empty string if no custom URL', () => {
|
||||
const appDetail = {
|
||||
modules: [
|
||||
{
|
||||
flowNodeType: FlowNodeTypeEnum.systemConfig,
|
||||
inputs: [
|
||||
{
|
||||
key: NodeInputKeyEnum.chatInputGuide,
|
||||
value: {}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
} as AppSchema;
|
||||
|
||||
const result = getAppQGuideCustomURL(appDetail);
|
||||
expect(result).toBe('');
|
||||
});
|
||||
});
|
||||
});
|
Reference in New Issue
Block a user