feat: test buildNodesAndEdgesFromDSLComponents (#940)

### What problem does this PR solve?
 feat: test buildNodesAndEdgesFromDSLComponents #918

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2024-05-27 19:35:14 +08:00
committed by GitHub
parent 571aaaff22
commit d9bc093df1
9 changed files with 4844 additions and 188 deletions

View File

@@ -0,0 +1,30 @@
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',
}),
]),
);
});