| 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); | |
| }); | |
| }); | |