File size: 4,552 Bytes
4e23b01 | 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 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 | import { rm } from 'node:fs/promises';
import { describe, expect, it } from 'vitest';
import { defineKlientConformance } from './helpers/conformance.js';
import { createKlient } from '../src/transports/memory/index.js';
import { createMemoryDispatcher } from '../src/transports/memory/dispatcher.js';
import { RPCError } from '../src/core/errors.js';
import { makeEngine } from './helpers/engine.js';
defineKlientConformance('memory', async () => {
const { homeDir, app } = await makeEngine();
const klient = createKlient({ scope: app });
return {
klient,
app,
cleanup: async () => {
await klient.close();
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
},
};
});
describe('memory dispatcher specifics', () => {
it('rejects unknown services and methods with RPCError(40001)', async () => {
const { homeDir, app } = await makeEngine();
const dispatcher = createMemoryDispatcher(app);
await expect(dispatcher.call({}, 'noSuchService', 'get', [])).rejects.toMatchObject({
name: 'RPCError',
code: 40001,
});
await expect(dispatcher.call({}, 'sessionIndex', 'noSuchMethod', [])).rejects.toMatchObject({
name: 'RPCError',
code: 40001,
});
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
});
it('reads non-function members as properties', async () => {
const { homeDir, app } = await makeEngine();
const dispatcher = createMemoryDispatcher(app);
await expect(dispatcher.call({}, 'bootstrapService', 'platform', [])).resolves.toBe(
process.platform,
);
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
});
it('rejects session/agent scopes for now', async () => {
const { homeDir, app } = await makeEngine();
const dispatcher = createMemoryDispatcher(app);
await expect(
dispatcher.call({ sessionId: 's1' }, 'sessionIndex', 'list', [{}]),
).rejects.toBeInstanceOf(RPCError);
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
});
it('delivers wire-cloned payloads (no live object identity)', async () => {
const { homeDir, app } = await makeEngine();
const klient = createKlient({ scope: app });
const list = await klient.global.workspaces.list();
// Mutating the result must not affect what a second call returns.
(list as unknown[]).push({ id: 'polluted' });
const again = await klient.global.workspaces.list();
expect(again.some((w) => w.id === 'polluted')).toBe(false);
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
});
it('does not respond to interactions owned by another session', async () => {
const { homeDir, app } = await makeEngine();
const klient = createKlient({ scope: app });
const dispatcher = createMemoryDispatcher(app);
const first = await klient.global.sessions.create({ workDir: process.cwd() });
const second = await klient.global.sessions.create({ workDir: process.cwd() });
try {
const parked = (await dispatcher.call({ sessionId: first.id }, 'sessionInteractionService', 'enqueue', [
{
kind: 'approval',
payload: {
toolName: 'Bash',
action: 'run',
display: { kind: 'command', command: 'ls' },
},
},
])) as { id: string };
await dispatcher.call({ sessionId: second.id }, 'sessionInteractionService', 'respond', [
parked.id,
{ decision: 'approved' },
]);
const pendingAfterCross = (await dispatcher.call(
{ sessionId: first.id },
'sessionInteractionService',
'listPending',
['approval'],
)) as readonly { id: string }[];
expect(pendingAfterCross.map((i) => i.id)).toEqual([parked.id]);
await dispatcher.call({ sessionId: first.id }, 'sessionInteractionService', 'respond', [
parked.id,
{ decision: 'approved' },
]);
const pendingAfterOwn = (await dispatcher.call(
{ sessionId: first.id },
'sessionInteractionService',
'listPending',
['approval'],
)) as readonly { id: string }[];
expect(pendingAfterOwn).toEqual([]);
} finally {
await klient.close();
app.dispose();
await rm(homeDir, { recursive: true, force: true, maxRetries: 3, retryDelay: 25 });
}
});
});
|