@ai-devkit/agent-manager 0.26.2 → 0.26.3

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

@@ -5,14 +5,19 @@ import * as path from 'path';
 import { CodexAdapter } from '../../adapters/CodexAdapter.js';
 import { AgentStatus } from '../../adapters/AgentAdapter.js';
 import { AgentRegistry } from '../../utils/AgentRegistry.js';
-import { listAgentProcesses, enrichProcesses } from '../../utils/process.js';
+import { listAgentProcesses, enrichProcesses, captureProcessSnapshot } from '../../utils/process.js';
 import { batchGetSessionFileBirthtimes } from '../../utils/session.js';
 import { matchProcessesToSessions, generateAgentName } from '../../utils/matching.js';
 import * as os from 'os';
-vi.mock('../../utils/process.js', ()=>({
+vi.mock('../../utils/process.js', async (importOriginal)=>{
+    const actual = await importOriginal();
+    return {
+        ...actual,
         listAgentProcesses: vi.fn(),
-        enrichProcesses: vi.fn()
-    }));
+        enrichProcesses: vi.fn(),
+        captureProcessSnapshot: vi.fn()
+    };
+});
 vi.mock('../../utils/session.js', async ()=>{
     const actual = await vi.importActual('../../utils/session');
     return {
@@ -26,6 +31,7 @@ vi.mock('../../utils/matching.js', ()=>({
     }));
 const mockedListAgentProcesses = listAgentProcesses;
 const mockedEnrichProcesses = enrichProcesses;
+const mockedCaptureProcessSnapshot = captureProcessSnapshot;
 const mockedBatchGetSessionFileBirthtimes = batchGetSessionFileBirthtimes;
 const mockedMatchProcessesToSessions = matchProcessesToSessions;
 const mockedGenerateAgentName = generateAgentName;
@@ -35,11 +41,14 @@ describe('CodexAdapter', ()=>{
         adapter = new CodexAdapter();
         mockedListAgentProcesses.mockReset();
         mockedEnrichProcesses.mockReset();
+        mockedCaptureProcessSnapshot.mockReset();
         mockedBatchGetSessionFileBirthtimes.mockReset();
         mockedMatchProcessesToSessions.mockReset();
         mockedGenerateAgentName.mockReset();
         // Default: enrichProcesses returns what it receives
         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))));
         // Default: generateAgentName returns "folder (pid)"
         mockedGenerateAgentName.mockImplementation((cwd, pid)=>{
             const folder = path.basename(cwd) || 'unknown';