d9bc093df1
### What problem does this PR solve? feat: test buildNodesAndEdgesFromDSLComponents #918 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
31 lines
796 B
TypeScript
31 lines
796 B
TypeScript
import { dsl } from './mock';
|
|
import { buildNodesAndEdgesFromDSLComponents } from './utils';
|
|
|
|
test('buildNodesAndEdgesFromDSLComponents', () => {
|
|
const { edges, nodes } = buildNodesAndEdgesFromDSLComponents(dsl.components);
|
|
|
|
expect(nodes.length).toEqual(4);
|
|
expect(edges.length).toEqual(4);
|
|
|
|
expect(edges).toEqual(
|
|
expect.arrayContaining([
|
|
expect.objectContaining({
|
|
source: 'begin',
|
|
target: 'Answer:China',
|
|
}),
|
|
expect.objectContaining({
|
|
source: 'Answer:China',
|
|
target: 'Retrieval:China',
|
|
}),
|
|
expect.objectContaining({
|
|
source: 'Retrieval:China',
|
|
target: 'Generate:China',
|
|
}),
|
|
expect.objectContaining({
|
|
source: 'Generate:China',
|
|
target: 'Answer:China',
|
|
}),
|
|
]),
|
|
);
|
|
});
|