|
| 1 | +import { describe, test, expect, beforeEach, afterEach } from 'bun:test'; |
| 2 | +import { mkdtemp, rm, mkdir, writeFile, appendFile } from 'node:fs/promises'; |
| 3 | +import { join } from 'node:path'; |
| 4 | +import { tmpdir } from 'node:os'; |
| 5 | +import { FileWatcher } from './file-watcher'; |
| 6 | + |
| 7 | +let claudeDir: string; |
| 8 | +let projectDir: string; |
| 9 | + |
| 10 | +beforeEach(async () => { |
| 11 | + claudeDir = await mkdtemp(join(tmpdir(), 'file-watcher-')); |
| 12 | + projectDir = join(claudeDir, 'projects', 'slug'); |
| 13 | + await mkdir(projectDir, { recursive: true }); |
| 14 | +}); |
| 15 | + |
| 16 | +afterEach(async () => { |
| 17 | + await rm(claudeDir, { recursive: true, force: true }); |
| 18 | +}); |
| 19 | + |
| 20 | +function jsonline(obj: unknown): string { |
| 21 | + return `${JSON.stringify(obj)}\n`; |
| 22 | +} |
| 23 | + |
| 24 | +interface ScanResult { |
| 25 | + newSessions: string[]; |
| 26 | + updates: string[]; |
| 27 | +} |
| 28 | + |
| 29 | +function makeWatcher(): { watcher: FileWatcher; result: ScanResult } { |
| 30 | + const result: ScanResult = { newSessions: [], updates: [] }; |
| 31 | + const watcher = new FileWatcher({ |
| 32 | + claudeDirs: [claudeDir], |
| 33 | + onNewSession: (sessionId) => { |
| 34 | + result.newSessions.push(sessionId); |
| 35 | + }, |
| 36 | + onSessionUpdate: (sessionId) => { |
| 37 | + result.updates.push(sessionId); |
| 38 | + }, |
| 39 | + }); |
| 40 | + return { watcher, result }; |
| 41 | +} |
| 42 | + |
| 43 | +describe('FileWatcher workflow (ultra mode) discovery', () => { |
| 44 | + test('discovers agent JSONLs nested under subagents/workflows/wf_*/', async () => { |
| 45 | + const sessionId = 'e20383a7-8361-4ea9-bfee-305ea79263bc'; |
| 46 | + await writeFile(join(projectDir, `${sessionId}.jsonl`), jsonline({ type: 'user' })); |
| 47 | + |
| 48 | + const subagentsDir = join(projectDir, sessionId, 'subagents'); |
| 49 | + await mkdir(subagentsDir, { recursive: true }); |
| 50 | + await writeFile(join(subagentsDir, 'agent-1111111111111111.jsonl'), jsonline({ type: 'user' })); |
| 51 | + |
| 52 | + const workflowDir = join(subagentsDir, 'workflows', 'wf_2ca7ddd8-324'); |
| 53 | + await mkdir(workflowDir, { recursive: true }); |
| 54 | + await writeFile(join(workflowDir, 'agent-a4f3fba033446cc4f.jsonl'), jsonline({ type: 'user' })); |
| 55 | + await writeFile(join(workflowDir, 'agent-a4f3fba033446cc4f.meta.json'), '{"agentType":"workflow-subagent"}'); |
| 56 | + await writeFile(join(workflowDir, 'journal.jsonl'), jsonline({ type: 'started' })); |
| 57 | + |
| 58 | + const { watcher, result } = makeWatcher(); |
| 59 | + await watcher.scan(); |
| 60 | + |
| 61 | + expect(result.newSessions).toContain(sessionId); |
| 62 | + expect(result.newSessions).toContain('agent-1111111111111111'); |
| 63 | + expect(result.newSessions).toContain('agent-a4f3fba033446cc4f'); |
| 64 | + // Orchestration journal and meta files must not become agents |
| 65 | + expect(result.newSessions).not.toContain('journal'); |
| 66 | + expect(result.newSessions).not.toContain('agent-a4f3fba033446cc4f.meta'); |
| 67 | + }); |
| 68 | + |
| 69 | + test('emits updates when a workflow agent JSONL grows', async () => { |
| 70 | + const sessionId = 'f207718a-066f-4609-82cb-eccf36a9e5a2'; |
| 71 | + const workflowDir = join(projectDir, sessionId, 'subagents', 'workflows', 'wf_9ca5aadf-8ab'); |
| 72 | + await mkdir(workflowDir, { recursive: true }); |
| 73 | + const agentPath = join(workflowDir, 'agent-afa8678eb14a03831.jsonl'); |
| 74 | + await writeFile(agentPath, jsonline({ type: 'user' })); |
| 75 | + |
| 76 | + const { watcher, result } = makeWatcher(); |
| 77 | + await watcher.scan(); |
| 78 | + expect(result.newSessions).toContain('agent-afa8678eb14a03831'); |
| 79 | + |
| 80 | + await appendFile(agentPath, jsonline({ type: 'assistant' })); |
| 81 | + await watcher.scan(); |
| 82 | + expect(result.updates).toContain('agent-afa8678eb14a03831'); |
| 83 | + }); |
| 84 | +}); |
0 commit comments