File size: 4,831 Bytes
7c5ed63 | 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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import { describe, it, expect, afterEach } from 'vitest';
import { buildCompactionContinuationText } from '@moonshot-ai/agent-core-v2/agent/contextMemory/compactionHandoff';
import { buildSessionFixture } from '../fixtures/build';
import { contextRoute } from '../../src/routes/context';
describe('context route', () => {
let cleanup: (() => Promise<void>) | null = null;
afterEach(async () => { if (cleanup) await cleanup(); cleanup = null; });
it('echoes the new projection fields (contextTokens, goal, swarm)', async () => {
const { home, cleanup: c } = await buildSessionFixture('sample-main');
cleanup = c;
const app = contextRoute(home);
const res = await app.request('/session_fixture/context?agent=main');
expect(res.status).toBe(200);
const body = (await res.json()) as Record<string, unknown>;
// The route must pass these projection fields straight through (it used to
// cherry-pick only messages/usage/config/permission/planMode).
expect(body).toHaveProperty('contextTokens');
expect(body).toHaveProperty('goal');
expect(body).toHaveProperty('swarm');
// The sample fixture's only step.end carries usage 10+5 → contextTokens=15,
// and has no goal / swarm records.
expect(body['contextTokens']).toBe(15);
expect(body['goal']).toBeNull();
expect(body['swarm']).toEqual({ active: false });
});
it('still echoes the existing fields', async () => {
const { home, cleanup: c } = await buildSessionFixture('sample-main');
cleanup = c;
const app = contextRoute(home);
const res = await app.request('/session_fixture/context?agent=main');
expect(res.status).toBe(200);
const body = (await res.json()) as Record<string, unknown>;
expect(body['sessionId']).toBe('session_fixture');
expect(body['agentId']).toBe('main');
expect(body).toHaveProperty('messages');
expect(body).toHaveProperty('usage');
expect(body).toHaveProperty('config');
expect(body).toHaveProperty('permission');
expect(body).toHaveProperty('planMode');
});
it('returns 404 for missing session', async () => {
const { home, cleanup: c } = await buildSessionFixture('sample-main');
cleanup = c;
const app = contextRoute(home);
const res = await app.request('/no-such-session/context?agent=main');
expect(res.status).toBe(404);
expect(await res.json()).toMatchObject({ code: 'NOT_FOUND' });
});
it('returns 400 for invalid agent id', async () => {
const { home, cleanup: c } = await buildSessionFixture('sample-main');
cleanup = c;
const app = contextRoute(home);
const res = await app.request('/session_fixture/context?agent=../escape');
expect(res.status).toBe(400);
expect(await res.json()).toMatchObject({ code: 'BAD_REQUEST' });
});
it('?history=full returns the pre-compaction messages (full reconstructed history)', async () => {
const { home, cleanup: c } = await buildSessionFixture('sample-compaction');
cleanup = c;
const app = contextRoute(home);
// Default (model view): the real user prompt before compaction is KEPT, the
// assistant reply is dropped, then the summary, then the post-compaction tail.
const modelRes = await app.request('/session_fixture/context?agent=main');
expect(modelRes.status).toBe(200);
const modelBody = (await modelRes.json()) as {
messages: { source: string; message: { content: { type: string; text?: string }[] } }[];
};
expect(modelBody.messages.map((m) => m.source)).toEqual([
'append_message', 'compaction_summary', 'append_message', 'append_message',
]);
expect(modelBody.messages[0]!.message.content[0]).toMatchObject({ text: 'before compaction' });
expect(modelBody.messages[2]!.message.content[0]).toMatchObject({ text: buildCompactionContinuationText() });
expect(modelBody.messages[3]!.message.content[0]).toMatchObject({ text: 'after compaction' });
// Full history: every pre-compaction message (user prompt + assistant reply)
// is KEPT, then the summary marker, then the post-compaction tail.
const fullRes = await app.request('/session_fixture/context?agent=main&history=full');
expect(fullRes.status).toBe(200);
const fullBody = (await fullRes.json()) as {
messages: { source: string; message: { content: { type: string; text?: string }[] } }[];
};
expect(fullBody.messages.map((m) => m.source)).toEqual([
'append_message', 'append_message', 'compaction_summary', 'append_message',
]);
expect(fullBody.messages[0]!.message.content[0]).toMatchObject({ text: 'before compaction' });
expect(fullBody.messages[1]!.message.content[0]).toMatchObject({ text: 'assistant reply' });
expect(fullBody.messages[3]!.message.content[0]).toMatchObject({ text: 'after compaction' });
});
});
|