bdae539e9e
The graph now represents the agent's Brain: its documents and the content made from them. Work in progress, failed runs, research suggestions and schedules are status, so they are left out. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
38 lines
1.7 KiB
TypeScript
38 lines
1.7 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import type { Source } from '../api/engine'
|
|
import { buildGraph, HEIGHT, WIDTH } from './graph'
|
|
import type { Material } from './results'
|
|
|
|
const doc = (docId: number, title: string, status = 'completed') => ({ docId, title, originalName: `${title}.pdf`, status }) as Source
|
|
const material = (id: number, status: Material['status'], sources: string[]) =>
|
|
({ id, agentId: 7, kind: 'summary', status, planTitle: `Lesson ${id}`, sources }) as Material
|
|
|
|
describe('buildGraph', () => {
|
|
const data = {
|
|
docs: [doc(1, 'Onboarding'), doc(2, 'Safety'), doc(3, 'Broken', 'failed')],
|
|
materials: [
|
|
material(10, 'accepted', ['Onboarding.pdf', 'Safety']), material(11, 'generating', []),
|
|
material(12, 'failed', ['Safety']), material(13, 'ready', ['Safety']),
|
|
],
|
|
}
|
|
const graph = buildGraph(data, c => c)
|
|
|
|
it('holds only the knowledge base and the content made from it', () => {
|
|
expect(graph.nodes.filter(n => !n.hub).map(n => `${n.cluster}:${n.id}`).sort())
|
|
.toEqual(['made:mat-7-10', 'made:mat-7-13', 'sources:doc-1', 'sources:doc-2'])
|
|
})
|
|
|
|
it('links content to its sources', () => {
|
|
const cross = graph.links.filter(l => !l.hub).map(l => `${l.source}>${l.target}`)
|
|
expect(cross.sort()).toEqual(['mat-7-10>doc-1', 'mat-7-10>doc-2', 'mat-7-13>doc-2'])
|
|
})
|
|
|
|
it('lays out the same data the same way, inside the canvas', () => {
|
|
expect(buildGraph(data, c => c).nodes).toEqual(graph.nodes)
|
|
for (const n of graph.nodes) {
|
|
expect(n.x).toBeGreaterThanOrEqual(0); expect(n.x).toBeLessThanOrEqual(WIDTH)
|
|
expect(n.y).toBeGreaterThanOrEqual(0); expect(n.y).toBeLessThanOrEqual(HEIGHT)
|
|
}
|
|
})
|
|
})
|