[Init] Initial commit - NetMesh terminal manager

infrastructure/ai/harness/sessionState.test.ts

@@ -0,0 +1,135 @@
+import assert from 'node:assert/strict';
+import test from 'node:test';
+import { SessionStateStore } from './sessionState.ts';
+
+test('SessionStateStore tracks terminal commands and reinjection text', () => {
+  const store = new SessionStateStore();
+  store.mergeFromUserGoal('chat-1', 'Fix nginx upstream timeout');
+  store.updateFromToolResult(
+    'chat-1',
+    'terminal_execute',
+    { sessionId: 'sess-1', command: 'tail -n 100 /var/log/nginx/error.log' },
+    'upstream timed out',
+    false,
+  );
+
+  const text = store.toReinjectionText('chat-1');
+  assert.ok(text?.includes('Fix nginx upstream timeout'));
+  assert.ok(text?.includes('sess-1'));
+  assert.ok(text?.includes('tail -n 100'));
+});
+
+test('SessionStateStore records tool errors as blockers', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1',
+    'terminal_execute',
+    { sessionId: 'sess-1', command: 'systemctl restart nginx' },
+    '{ "error": "Job failed" }',
+    true,
+  );
+  const text = store.toReinjectionText('chat-1');
+  assert.ok(text?.includes('Open blockers'));
+});
+
+test('SessionStateStore restores active background jobs and poll cursors', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1',
+    'terminal_start',
+    { sessionId: 'sess-1', command: 'npm run dev' },
+    JSON.stringify({ ok: true, jobId: 'job-1', status: 'running', nextOffset: 0 }),
+  );
+  store.updateFromToolResult(
+    'chat-1',
+    'terminal_poll',
+    { jobId: 'job-1', offset: 0 },
+    JSON.stringify({ ok: true, jobId: 'job-1', status: 'running', nextOffset: 420 }),
+  );
+
+  const state = store.get('chat-1');
+  assert.equal(state.version, 1);
+  assert.equal(state.activeJobs['job-1'].nextOffset, 420);
+  const text = store.toReinjectionText('chat-1') ?? '';
+  assert.match(text, /job-1/);
+  assert.match(text, /offset=420/);
+  assert.match(text, /poll the existing job/i);
+  assert.match(text, /do not restart/i);
+  assert.match(text, /unverified after compaction/i);
+});
+
+test('SessionStateStore drops a remembered job after poll reports it missing', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1', 'terminal_start', { sessionId: 'sess-1', command: 'npm run dev' },
+    JSON.stringify({ jobId: 'job-lost', status: 'running' }), false,
+  );
+  store.updateFromToolResult(
+    'chat-1', 'terminal_poll', { jobId: 'job-lost', offset: 0 },
+    JSON.stringify({ error: 'Job not found' }), true,
+  );
+  assert.equal(store.get('chat-1').activeJobs['job-lost'], undefined);
+  assert.doesNotMatch(store.toReinjectionText('chat-1') ?? '', /Remembered terminal jobs/);
+});
+
+test('SessionStateStore preserves a running job after a transient poll error', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1', 'terminal_start', { sessionId: 'sess-1', command: 'npm run dev' },
+    JSON.stringify({ jobId: 'job-running', status: 'running', nextOffset: 420 }), false,
+  );
+  store.updateFromToolResult(
+    'chat-1', 'terminal_poll', { jobId: 'job-running', offset: 420 },
+    JSON.stringify({ error: 'temporary IPC timeout' }), true,
+  );
+
+  const job = store.get('chat-1').activeJobs['job-running'];
+  assert.equal(job.status, 'unverified');
+  assert.equal(job.nextOffset, 420);
+  assert.match(store.toReinjectionText('chat-1') ?? '', /job-running/);
+  assert.match(store.toReinjectionText('chat-1') ?? '', /do not restart/i);
+});
+
+test('SessionStateStore drops a job after poll confirms cancellation', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1', 'terminal_start', { sessionId: 'sess-1', command: 'npm run dev' },
+    JSON.stringify({ jobId: 'job-cancelled', status: 'running' }), false,
+  );
+  store.updateFromToolResult(
+    'chat-1', 'terminal_poll', { jobId: 'job-cancelled', offset: 0 },
+    JSON.stringify({ jobId: 'job-cancelled', status: 'cancelled', error: 'Cancelled' }), true,
+  );
+
+  assert.equal(store.get('chat-1').activeJobs['job-cancelled'], undefined);
+});
+
+test('SessionStateStore records the last terminal screen range read', () => {
+  const store = new SessionStateStore();
+  store.updateFromToolResult(
+    'chat-1',
+    'terminal_read_context',
+    { sessionId: 'sess-1', range: 'tail', startLine: 80, maxLines: 20 },
+    JSON.stringify({ ok: true, sessionId: 'sess-1', startLine: 80, endLine: 99 }),
+  );
+
+  assert.deepEqual(store.get('chat-1').terminalReadCursors['sess-1'], {
+    range: 'tail',
+    startLine: 80,
+    endLine: 99,
+  });
+});
+
+test('SessionStateStore reinjects edited files and unfinished plan items', () => {
+  const store = new SessionStateStore();
+  store.mergeFileChanges('chat-1', ['/repo/src/a.ts', '/repo/src/b.ts']);
+  store.mergePlan('chat-1', [
+    { text: 'inspect failure', completed: true },
+    { text: 'run regression tests', completed: false },
+  ]);
+
+  const text = store.toReinjectionText('chat-1') ?? '';
+  assert.match(text, /\/repo\/src\/a\.ts/);
+  assert.match(text, /\[done\] inspect failure/);
+  assert.match(text, /\[todo\] run regression tests/);
+});