@ai-devkit/agent-manager 0.31.0 → 0.33.0

package/dist/__tests__/adapters/PiAdapter.test.js

@@ -1,618 +0,0 @@
-/**
- * Tests for PiAdapter
- */ import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
-import { PiAdapter } from '../../adapters/PiAdapter.js';
-import { AgentStatus } from '../../adapters/AgentAdapter.js';
-import { AgentRegistry } from '../../utils/AgentRegistry.js';
-import { listAgentProcesses, enrichProcesses, captureProcessSnapshot } from '../../utils/process.js';
-import { matchProcessesToSessions, generateAgentName } from '../../utils/matching.js';
-vi.mock('../../utils/process.js', async (importOriginal)=>{
-    const actual = await importOriginal();
-    return {
-        ...actual,
-        listAgentProcesses: vi.fn(),
-        enrichProcesses: vi.fn(),
-        captureProcessSnapshot: vi.fn()
-    };
-});
-vi.mock('../../utils/matching.js', ()=>({
-        matchProcessesToSessions: vi.fn(),
-        generateAgentName: vi.fn()
-    }));
-const mockedListAgentProcesses = listAgentProcesses;
-const mockedEnrichProcesses = enrichProcesses;
-const mockedCaptureProcessSnapshot = captureProcessSnapshot;
-const mockedMatchProcessesToSessions = matchProcessesToSessions;
-const mockedGenerateAgentName = generateAgentName;
-describe('PiAdapter', ()=>{
-    let adapter;
-    let tmpHome;
-    let sessionsDir;
-    beforeEach(()=>{
-        tmpHome = fs.mkdtempSync(path.join(os.tmpdir(), 'pi-adapter-test-'));
-        process.env.HOME = tmpHome;
-        sessionsDir = path.join(tmpHome, '.pi', 'agent', 'sessions');
-        fs.mkdirSync(sessionsDir, {
-            recursive: true
-        });
-        adapter = new PiAdapter(new AgentRegistry(path.join(tmpHome, 'agents.json')));
-        mockedListAgentProcesses.mockReset();
-        mockedEnrichProcesses.mockReset();
-        mockedCaptureProcessSnapshot.mockReset();
-        mockedMatchProcessesToSessions.mockReset();
-        mockedGenerateAgentName.mockReset();
-        mockedEnrichProcesses.mockImplementation((procs)=>procs);
-        // Compatibility shim for standalone adapter discovery; the manager captures once and slices by name.
-        mockedCaptureProcessSnapshot.mockImplementation(async (names)=>enrichProcesses(names.flatMap((name)=>listAgentProcesses(name))));
-        mockedMatchProcessesToSessions.mockReturnValue([]);
-        mockedGenerateAgentName.mockImplementation((cwd, pid)=>{
-            const folder = path.basename(cwd) || 'unknown';
-            return `${folder} (${pid})`;
-        });
-    });
-    afterEach(()=>{
-        fs.rmSync(tmpHome, {
-            recursive: true,
-            force: true
-        });
-    });
-    it('exposes pi type', ()=>{
-        expect(adapter.type).toBe('pi');
-    });
-    it('identifies Pi commands without matching unrelated paths', ()=>{
-        expect(adapter.canHandle({
-            pid: 1,
-            command: 'pi',
-            cwd: '/repo',
-            tty: 'ttys001'
-        })).toBe(true);
-        expect(adapter.canHandle({
-            pid: 2,
-            command: '/usr/local/bin/PI --model x',
-            cwd: '/repo',
-            tty: 'ttys002'
-        })).toBe(true);
-        expect(adapter.canHandle({
-            pid: 3,
-            command: 'node /opt/pi/bin/pi.js',
-            cwd: '/repo',
-            tty: 'ttys003'
-        })).toBe(true);
-        expect(adapter.canHandle({
-            pid: 4,
-            command: 'node /repo/feature-pi-adapter/script.js',
-            cwd: '/repo',
-            tty: 'ttys004'
-        })).toBe(false);
-        expect(adapter.canHandle({
-            pid: 5,
-            command: 'C:\\tools\\node.exe C:\\lib\\pi.js',
-            cwd: '/repo',
-            tty: 'ttys005'
-        })).toBe(true);
-    });
-    it('rejects foreign executables from a hand-built snapshot context', async ()=>{
-        const foreign = {
-            pid: 6,
-            command: 'claude --resume /Users/x/repos/pi/session.jsonl',
-            cwd: '/repo',
-            tty: 'ttys006'
-        };
-        expect(await adapter.detectAgents({
-            processes: [
-                foreign
-            ]
-        })).toEqual([]);
-    });
-    it('maps a running Pi process to the tracker session for its PID', async ()=>{
-        const cwd = '/repo/project-a';
-        const proc = makeProcess({
-            pid: 101,
-            cwd
-        });
-        const sessionFile = writePiSession(cwd, [
-            {
-                type: 'session_meta',
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-101',
-                cwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'implement Pi adapter'
-            },
-            {
-                role: 'assistant',
-                timestamp: new Date().toISOString(),
-                content: 'working on it'
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            101: sessionFile
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0]).toMatchObject({
-            type: 'pi',
-            pid: 101,
-            projectPath: cwd,
-            sessionId: 'sess-101',
-            summary: 'implement Pi adapter',
-            status: AgentStatus.WAITING,
-            sessionFilePath: sessionFile
-        });
-        expect(mockedMatchProcessesToSessions).not.toHaveBeenCalled();
-    });
-    it('truncates long user prompts in detected agent summaries', async ()=>{
-        const cwd = '/repo/project-long-summary';
-        const proc = makeProcess({
-            pid: 112,
-            cwd
-        });
-        const longPrompt = 'x'.repeat(140);
-        const sessionFile = writePiSession(cwd, [
-            {
-                type: 'session',
-                timestamp: '2026-06-10T08:58:20.754Z',
-                id: 'sess-long',
-                cwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: longPrompt
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            112: sessionFile
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents[0].summary).toHaveLength(120);
-        expect(agents[0].summary.endsWith('...')).toBe(true);
-    });
-    it('uses the filename session id fallback and reports running when the latest message is from the user', async ()=>{
-        const cwd = '/repo/project-filename-fallback';
-        const proc = makeProcess({
-            pid: 113,
-            cwd
-        });
-        const sessionFile = writePiSessionWithFileName(cwd, 'plain-session.jsonl', [
-            {
-                role: 'user',
-                timestamp: new Date().toISOString(),
-                content: 'still working'
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            113: sessionFile
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents[0]).toMatchObject({
-            sessionId: 'plain-session',
-            summary: 'still working',
-            status: AgentStatus.RUNNING
-        });
-    });
-    it('falls back to legacy matching when sessions.json is missing', async ()=>{
-        const cwd = '/repo/project-b';
-        const proc = makeProcess({
-            pid: 202,
-            cwd
-        });
-        const sessionFile = writePiSession(cwd, [
-            {
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-202'
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'fallback matching please'
-            }
-        ]);
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        mockedMatchProcessesToSessions.mockReturnValue([
-            {
-                process: proc,
-                session: {
-                    sessionId: 'sess-202',
-                    filePath: sessionFile,
-                    projectDir: path.dirname(sessionFile),
-                    birthtimeMs: Date.now(),
-                    resolvedCwd: cwd
-                },
-                deltaMs: 0
-            }
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0]).toMatchObject({
-            type: 'pi',
-            pid: 202,
-            projectPath: cwd,
-            sessionId: 'sess-202',
-            summary: 'fallback matching please'
-        });
-        expect(mockedMatchProcessesToSessions).toHaveBeenCalledWith([
-            proc
-        ], expect.arrayContaining([
-            expect.objectContaining({
-                filePath: sessionFile,
-                resolvedCwd: cwd
-            })
-        ]));
-    });
-    it('ignores malformed tracker metadata and still falls back to legacy matching', async ()=>{
-        const cwd = '/repo/project-c';
-        const proc = makeProcess({
-            pid: 303,
-            cwd
-        });
-        const sessionFile = writePiSession(cwd, [
-            {
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-303',
-                cwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'recover from bad tracker'
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), '{bad json');
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        mockedMatchProcessesToSessions.mockReturnValue([
-            {
-                process: proc,
-                session: {
-                    sessionId: 'sess-303',
-                    filePath: sessionFile,
-                    projectDir: path.dirname(sessionFile),
-                    birthtimeMs: Date.now(),
-                    resolvedCwd: cwd
-                },
-                deltaMs: 0
-            }
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0].sessionId).toBe('sess-303');
-    });
-    it('falls back to legacy matching when a trusted tracker session is unparseable', async ()=>{
-        const cwd = '/repo/project-bad-tracker-session';
-        const proc = makeProcess({
-            pid: 304,
-            cwd
-        });
-        const badSessionFile = writePiSessionWithFileName(cwd, 'bad.jsonl', [
-            '{not json'
-        ]);
-        const fallbackSessionFile = writePiSession(cwd, [
-            {
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-304',
-                cwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'fallback after bad tracker session'
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            304: badSessionFile
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        mockedMatchProcessesToSessions.mockReturnValue([
-            {
-                process: proc,
-                session: {
-                    sessionId: 'sess-304',
-                    filePath: fallbackSessionFile,
-                    projectDir: path.dirname(fallbackSessionFile),
-                    birthtimeMs: Date.now(),
-                    resolvedCwd: cwd
-                },
-                deltaMs: 0
-            }
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0]).toMatchObject({
-            sessionId: 'sess-304',
-            summary: 'fallback after bad tracker session',
-            sessionFilePath: fallbackSessionFile
-        });
-        expect(mockedMatchProcessesToSessions).toHaveBeenCalled();
-    });
-    it('does not trust tracker paths outside the Pi sessions directory', async ()=>{
-        const cwd = '/repo/project-d';
-        const proc = makeProcess({
-            pid: 404,
-            cwd
-        });
-        const outside = path.join(tmpHome, 'outside.jsonl');
-        fs.writeFileSync(outside, JSON.stringify({
-            role: 'user',
-            content: 'nope'
-        }));
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            404: outside
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0]).toMatchObject({
-            type: 'pi',
-            pid: 404,
-            sessionId: 'pid-404',
-            summary: 'Pi process running'
-        });
-    });
-    it('returns a process-only agent when no session can be matched', async ()=>{
-        const proc = makeProcess({
-            pid: 505,
-            cwd: '/repo/project-e'
-        });
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents).toHaveLength(1);
-        expect(agents[0]).toMatchObject({
-            type: 'pi',
-            status: AgentStatus.RUNNING,
-            pid: 505,
-            projectPath: '/repo/project-e',
-            sessionId: 'pid-505',
-            summary: 'Pi process running'
-        });
-    });
-    it('reads user and assistant conversation messages from JSONL', ()=>{
-        const cwd = '/repo/project-f';
-        const sessionFile = writePiSession(cwd, [
-            {
-                role: 'system',
-                timestamp: '2026-06-10T08:58:20.000Z',
-                content: 'hidden'
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'hello pi'
-            },
-            {
-                type: 'assistant',
-                timestamp: '2026-06-10T08:58:22.000Z',
-                message: {
-                    content: 'hello human'
-                }
-            },
-            '{not json'
-        ]);
-        expect(adapter.getConversation(sessionFile)).toEqual([
-            {
-                role: 'user',
-                content: 'hello pi',
-                timestamp: '2026-06-10T08:58:21.000Z'
-            },
-            {
-                role: 'assistant',
-                content: 'hello human',
-                timestamp: '2026-06-10T08:58:22.000Z'
-            }
-        ]);
-    });
-    it('includes system entries only in verbose conversation mode', ()=>{
-        const cwd = '/repo/project-verbose';
-        const sessionFile = writePiSession(cwd, [
-            {
-                role: 'system',
-                timestamp: '2026-06-10T08:58:20.000Z',
-                content: 'model changed'
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'visible'
-            }
-        ]);
-        expect(adapter.getConversation(sessionFile)).toEqual([
-            {
-                role: 'user',
-                content: 'visible',
-                timestamp: '2026-06-10T08:58:21.000Z'
-            }
-        ]);
-        expect(adapter.getConversation(sessionFile, {
-            verbose: true
-        })).toEqual([
-            {
-                role: 'system',
-                content: 'model changed',
-                timestamp: '2026-06-10T08:58:20.000Z'
-            },
-            {
-                role: 'user',
-                content: 'visible',
-                timestamp: '2026-06-10T08:58:21.000Z'
-            }
-        ]);
-    });
-    it('reads real Pi message entries with nested role and text parts', async ()=>{
-        const cwd = '/repo/project-real';
-        const proc = makeProcess({
-            pid: 606,
-            cwd
-        });
-        const sessionFile = writePiSession(cwd, [
-            {
-                type: 'session',
-                version: 3,
-                id: 'sess-real',
-                timestamp: '2026-06-10T13:27:17.581Z',
-                cwd
-            },
-            {
-                type: 'model_change',
-                id: 'model-1',
-                timestamp: '2026-06-10T13:27:17.655Z',
-                modelId: 'claude-sonnet-4-6'
-            },
-            {
-                type: 'message',
-                id: 'msg-user',
-                timestamp: '2026-06-10T13:27:37.975Z',
-                message: {
-                    role: 'user',
-                    content: [
-                        {
-                            type: 'text',
-                            text: 'hello'
-                        }
-                    ],
-                    timestamp: 1781098057974
-                }
-            },
-            {
-                type: 'message',
-                id: 'msg-assistant',
-                timestamp: '2026-06-10T13:27:40.161Z',
-                message: {
-                    role: 'assistant',
-                    content: [
-                        {
-                            type: 'text',
-                            text: 'Hello! How can I help you today?'
-                        }
-                    ],
-                    provider: 'anthropic',
-                    model: 'claude-sonnet-4-6',
-                    timestamp: 1781098058012
-                }
-            }
-        ]);
-        fs.writeFileSync(path.join(tmpHome, '.pi', 'agent', 'sessions.json'), JSON.stringify({
-            606: sessionFile
-        }));
-        mockedListAgentProcesses.mockReturnValue([
-            proc
-        ]);
-        expect(adapter.getConversation(sessionFile)).toEqual([
-            {
-                role: 'user',
-                content: 'hello',
-                timestamp: '2026-06-10T13:27:37.975Z'
-            },
-            {
-                role: 'assistant',
-                content: 'Hello! How can I help you today?',
-                timestamp: '2026-06-10T13:27:40.161Z'
-            }
-        ]);
-        const agents = await adapter.detectAgents();
-        expect(agents[0]).toMatchObject({
-            sessionId: 'sess-real',
-            summary: 'hello',
-            lastActive: new Date('2026-06-10T13:27:40.161Z')
-        });
-    });
-    it('lists historical sessions and applies cwd filtering', async ()=>{
-        const matchingCwd = '/repo/project-g';
-        const otherCwd = '/repo/project-h';
-        const matchingSession = writePiSession(matchingCwd, [
-            {
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-g',
-                cwd: matchingCwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'first matching message'
-            }
-        ]);
-        writePiSession(otherCwd, [
-            {
-                timestamp: '2026-06-10T08:58:20.754Z',
-                sessionId: 'sess-h',
-                cwd: otherCwd
-            },
-            {
-                role: 'user',
-                timestamp: '2026-06-10T08:58:21.000Z',
-                content: 'other message'
-            }
-        ]);
-        const sessions = await adapter.listSessions({
-            cwd: matchingCwd
-        });
-        expect(sessions).toEqual([
-            expect.objectContaining({
-                type: 'pi',
-                sessionId: 'sess-g',
-                cwd: matchingCwd,
-                firstUserMessage: 'first matching message',
-                sessionFilePath: matchingSession
-            })
-        ]);
-    });
-    function makeProcess(overrides) {
-        return {
-            pid: 1,
-            command: 'pi',
-            cwd: '/repo',
-            tty: 'ttys001',
-            startTime: new Date('2026-06-10T08:58:20.000Z'),
-            ...overrides
-        };
-    }
-    function writePiSession(cwd, entries) {
-        const projectDir = path.join(sessionsDir, encodeProjectDir(cwd));
-        fs.mkdirSync(projectDir, {
-            recursive: true
-        });
-        const sessionId = entries.map((entry)=>typeof entry === 'string' ? undefined : entry.sessionId).find((value)=>typeof value === 'string') ?? cryptoRandomSessionId();
-        const filePath = path.join(projectDir, `2026-06-10T08-58-20-754Z_${sessionId}.jsonl`);
-        fs.writeFileSync(filePath, entries.map((entry)=>typeof entry === 'string' ? entry : JSON.stringify(entry)).join('\n'));
-        return filePath;
-    }
-    function writePiSessionWithFileName(cwd, fileName, entries) {
-        const projectDir = path.join(sessionsDir, encodeProjectDir(cwd));
-        fs.mkdirSync(projectDir, {
-            recursive: true
-        });
-        const filePath = path.join(projectDir, fileName);
-        fs.writeFileSync(filePath, entries.map((entry)=>typeof entry === 'string' ? entry : JSON.stringify(entry)).join('\n'));
-        return filePath;
-    }
-    function encodeProjectDir(cwd) {
-        return cwd.replace(/\//g, '-').replace(/^-?/, '--') + '--';
-    }
-    function cryptoRandomSessionId() {
-        return `019eb0c1-06d2-71ed-90ee-${Math.random().toString(16).slice(2, 14).padEnd(12, '0')}`;
-    }
-});
-
-//# sourceMappingURL=PiAdapter.test.js.map