feat: block follow-up IPC after closeStdin with closingStdin flag

Prevent piping new messages to an agent that is shutting down.
Reset flag on new process start and cleanup.
This commit is contained in:
Eyejoker
2026-03-16 05:39:35 +09:00
parent ea03956d0a
commit 385f3a0349
2 changed files with 46 additions and 0 deletions

View File

@@ -413,6 +413,34 @@ describe('GroupQueue', () => {
await vi.advanceTimersByTimeAsync(10);
});
it('does not pipe follow-up messages to an agent after closeStdin', async () => {
let resolveProcess: () => void;
const processMessages = vi.fn(async () => {
await new Promise<void>((resolve) => {
resolveProcess = resolve;
});
return true;
});
queue.setProcessMessagesFn(processMessages);
queue.enqueueMessageCheck('group1@g.us', 'test-group');
await vi.advanceTimersByTimeAsync(10);
queue.registerProcess('group1@g.us', {} as any, 'agent-1', 'test-group');
queue.notifyIdle('group1@g.us');
queue.closeStdin('group1@g.us');
expect(queue.sendMessage('group1@g.us', 'hello after clear')).toBe(false);
queue.enqueueMessageCheck('group1@g.us', 'test-group');
resolveProcess!();
await vi.advanceTimersByTimeAsync(10);
expect(processMessages).toHaveBeenCalledTimes(2);
});
it('preempts when idle arrives with pending tasks', async () => {
const fs = await import('fs');
let resolveProcess: () => void;