78 lines
2.4 KiB
TypeScript
78 lines
2.4 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import {
|
|
WorkflowCheckpointStore,
|
|
type WorkflowCheckpointSettingsAdapter,
|
|
} from '../../../../src/main/agentic/workflow/checkpointStore';
|
|
|
|
class InMemorySettingsAdapter implements WorkflowCheckpointSettingsAdapter {
|
|
private readonly store = new Map<string, string>();
|
|
|
|
async getSetting(key: string): Promise<string | null> {
|
|
return this.store.get(key) ?? null;
|
|
}
|
|
|
|
async setSetting(key: string, value: string): Promise<void> {
|
|
this.store.set(key, value);
|
|
}
|
|
}
|
|
|
|
describe('WorkflowCheckpointStore', () => {
|
|
it('persists and reloads workflow checkpoints by conversation id', async () => {
|
|
const adapter = new InMemorySettingsAdapter();
|
|
const store = new WorkflowCheckpointStore(adapter);
|
|
|
|
await store.save({
|
|
conversationId: 'conversation-1',
|
|
state: 'awaiting_input',
|
|
pendingFields: ['date'],
|
|
lastTraceId: 'trace-1',
|
|
updatedAt: new Date('2026-02-25T10:00:00.000Z').toISOString(),
|
|
});
|
|
|
|
const loaded = await store.load('conversation-1');
|
|
|
|
expect(loaded).not.toBeNull();
|
|
expect(loaded?.state).toBe('awaiting_input');
|
|
expect(loaded?.pendingFields).toEqual(['date']);
|
|
expect(loaded?.lastTraceId).toBe('trace-1');
|
|
});
|
|
|
|
it('returns null when no checkpoint exists', async () => {
|
|
const adapter = new InMemorySettingsAdapter();
|
|
const store = new WorkflowCheckpointStore(adapter);
|
|
|
|
const loaded = await store.load('missing-conversation');
|
|
|
|
expect(loaded).toBeNull();
|
|
});
|
|
|
|
it('returns null for malformed checkpoint JSON', async () => {
|
|
const adapter = new InMemorySettingsAdapter();
|
|
await adapter.setSetting('agui.workflow.conversation-2', '{not-valid-json');
|
|
const store = new WorkflowCheckpointStore(adapter);
|
|
|
|
const loaded = await store.load('conversation-2');
|
|
|
|
expect(loaded).toBeNull();
|
|
});
|
|
|
|
it('returns null when stored checkpoint conversation id does not match', async () => {
|
|
const adapter = new InMemorySettingsAdapter();
|
|
await adapter.setSetting(
|
|
'agui.workflow.conversation-3',
|
|
JSON.stringify({
|
|
conversationId: 'other-conversation',
|
|
state: 'planning',
|
|
pendingFields: [],
|
|
lastTraceId: 'trace-mismatch',
|
|
updatedAt: new Date('2026-02-25T10:00:00.000Z').toISOString(),
|
|
}),
|
|
);
|
|
|
|
const store = new WorkflowCheckpointStore(adapter);
|
|
const loaded = await store.load('conversation-3');
|
|
|
|
expect(loaded).toBeNull();
|
|
});
|
|
});
|