File size: 970 Bytes
f0634fb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import { describe, expect, it } from 'vitest';
import { IAgentStateService } from '#/agent/state/agentState';
import { createTestAgent } from '../../harness/agent';
import { BUILTIN_REPLAYABLE_STATE_KEYS } from '../../state/builtinReplayableKeys';
describe('agent state snapshot (full agent scope)', () => {
it('serializes every registered key and stays small', () => {
const ctx = createTestAgent();
const states = ctx.get(IAgentStateService);
const excluded = new Set(BUILTIN_REPLAYABLE_STATE_KEYS.map((key) => key.name));
const registered = states.entries().map(([name]) => name);
const snapshot = states.snapshot();
expect(Object.keys(snapshot).toSorted()).toEqual(
registered.filter((name) => !excluded.has(name)).toSorted(),
);
for (const name of excluded) {
expect(snapshot[name]).toBeUndefined();
}
const json = JSON.stringify(snapshot);
expect(json.length).toBeLessThan(5 * 1024 * 1024);
});
});
|