feat(cli): Add agent view PTY workers

packages/cli/src/agent-view/attach-lease.test.ts

@@ -0,0 +1,257 @@
+/**
+ * @license
+ * Copyright 2026 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, expect, it } from 'vitest';
+import {
+  AgentViewAttachLeaseManager,
+  MAX_AGENT_VIEW_ATTACH_LEASE_TTL_MS,
+} from './attach-lease.js';
+
+describe('AgentViewAttachLeaseManager', () => {
+  it('acquires a lease for an unattached session', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => 'lease-1',
+      defaultTtlMs: 1000,
+    });
+
+    expect(manager.acquire('session-1', { clientId: 'terminal-1' })).toEqual({
+      ok: true,
+      lease: {
+        sessionId: 'session-1',
+        leaseId: 'lease-1',
+        clientId: 'terminal-1',
+        acquiredAt: '2026-07-17T00:00:00.000Z',
+        lastHeartbeatAt: '2026-07-17T00:00:00.000Z',
+        expiresAt: '2026-07-17T00:00:01.000Z',
+      },
+    });
+  });
+
+  it('uses a random lease id when no id factory is provided', () => {
+    const manager = new AgentViewAttachLeaseManager();
+    const result = manager.acquire('session-1');
+
+    expect(result).toMatchObject({
+      ok: true,
+      lease: {
+        sessionId: 'session-1',
+      },
+    });
+    if (result.ok) {
+      expect(result.lease.leaseId).toMatch(
+        /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/,
+      );
+    }
+  });
+
+  it('isolates leases across sessions', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    let idCounter = 0;
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => `lease-${++idCounter}`,
+    });
+
+    const first = manager.acquire('session-a', { clientId: 'terminal-a' });
+    const second = manager.acquire('session-b', { clientId: 'terminal-b' });
+
+    expect(first.ok && first.lease.leaseId).toBe('lease-1');
+    expect(second.ok && second.lease.leaseId).toBe('lease-2');
+
+    // Session B's lease must not disturb session A's.
+    expect(manager.get('session-a')?.leaseId).toBe('lease-1');
+    expect(manager.heartbeat('session-a', 'lease-1')?.leaseId).toBe('lease-1');
+
+    // Releasing session A must not affect session B.
+    expect(manager.release('session-a', 'lease-1')).toBe(true);
+    expect(manager.get('session-a')).toBeUndefined();
+    expect(manager.get('session-b')?.leaseId).toBe('lease-2');
+  });
+
+  it('rejects a second acquire while a lease is active', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => 'lease-1',
+    });
+    const first = manager.acquire('session-1');
+
+    expect(
+      manager.acquire('session-1', {
+        leaseId: 'lease-2',
+      }),
+    ).toEqual({
+      ok: false,
+      reason: 'already_attached',
+      lease: {
+        sessionId: first.lease.sessionId,
+        acquiredAt: first.lease.acquiredAt,
+        lastHeartbeatAt: first.lease.lastHeartbeatAt,
+        expiresAt: first.lease.expiresAt,
+      },
+    });
+  });
+
+  it('does not disclose the active lease id on contested acquire', () => {
+    const manager = new AgentViewAttachLeaseManager({
+      createLeaseId: () => 'lease-1',
+    });
+    manager.acquire('session-1');
+
+    const result = manager.acquire('session-1', { leaseId: 'lease-2' });
+
+    expect(result.ok).toBe(false);
+    if (!result.ok) {
+      expect('leaseId' in result.lease).toBe(false);
+    }
+  });
+
+  it('generates a lease id when the provided id is empty', () => {
+    const manager = new AgentViewAttachLeaseManager({
+      createLeaseId: () => 'generated-lease',
+    });
+
+    expect(manager.acquire('session-1', { leaseId: '' })).toMatchObject({
+      ok: true,
+      lease: {
+        leaseId: 'generated-lease',
+      },
+    });
+  });
+
+  it('releases only the matching lease', () => {
+    const manager = new AgentViewAttachLeaseManager({
+      createLeaseId: () => 'lease-1',
+    });
+    manager.acquire('session-1');
+
+    expect(manager.release('session-1', 'other-lease')).toBe(false);
+    expect(manager.get('session-1')).toMatchObject({ leaseId: 'lease-1' });
+    expect(manager.release('session-1', 'lease-1')).toBe(true);
+    expect(manager.get('session-1')).toBeUndefined();
+  });
+
+  it('rejects non-positive, oversized, and non-finite ttls', () => {
+    const manager = new AgentViewAttachLeaseManager();
+
+    expect(() => manager.acquire('session-1', { ttlMs: 0 })).toThrow(
+      'Attach lease ttlMs must be positive.',
+    );
+    expect(() => manager.acquire('session-1', { ttlMs: Number.NaN })).toThrow(
+      RangeError,
+    );
+    expect(() =>
+      manager.acquire('session-1', {
+        ttlMs: MAX_AGENT_VIEW_ATTACH_LEASE_TTL_MS + 1,
+      }),
+    ).toThrow('Attach lease ttlMs must not exceed');
+    expect(() => manager.acquire('')).toThrow(
+      'Agent View session id is required.',
+    );
+    expect(manager.get('session-1')).toBeUndefined();
+  });
+
+  it('honors per-call ttls and expires lazily through get and release', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => 'lease-1',
+      defaultTtlMs: 1000,
+    });
+    manager.acquire('session-1', { ttlMs: 2000 });
+
+    clock.advance(1500);
+    expect(manager.get('session-1')).toMatchObject({ leaseId: 'lease-1' });
+    expect(
+      manager.heartbeat('session-1', 'lease-1', { ttlMs: 3000 }),
+    ).toMatchObject({ expiresAt: '2026-07-17T00:00:04.500Z' });
+
+    clock.advance(3000);
+    expect(manager.get('session-1')).toBeUndefined();
+    expect(manager.release('session-1', 'lease-1')).toBe(false);
+  });
+
+  it('expires stale leases and allows reacquire', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    const leaseIds = ['lease-1', 'lease-2'];
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => leaseIds.shift() ?? 'missing-lease',
+      defaultTtlMs: 1000,
+    });
+    const first = manager.acquire('session-1');
+
+    clock.advance(1000);
+
+    expect(manager.expire()).toEqual([first.lease]);
+    expect(manager.acquire('session-1')).toMatchObject({
+      ok: true,
+      lease: {
+        leaseId: 'lease-2',
+        acquiredAt: '2026-07-17T00:00:01.000Z',
+      },
+    });
+  });
+
+  it('expires corrupted leases with invalid timestamps', () => {
+    const manager = new AgentViewAttachLeaseManager({
+      createLeaseId: () => 'lease-1',
+    });
+    manager.acquire('session-1');
+    (
+      manager as unknown as {
+        leases: Map<string, { expiresAt: string }>;
+      }
+    ).leases.get('session-1')!.expiresAt = 'not-a-date';
+
+    expect(manager.expire()).toEqual([
+      expect.objectContaining({
+        sessionId: 'session-1',
+        leaseId: 'lease-1',
+        expiresAt: 'not-a-date',
+      }),
+    ]);
+    expect(manager.get('session-1')).toBeUndefined();
+  });
+
+  it('heartbeat extends the matching lease', () => {
+    const clock = fakeClock('2026-07-17T00:00:00.000Z');
+    const manager = new AgentViewAttachLeaseManager({
+      now: clock.now,
+      createLeaseId: () => 'lease-1',
+      defaultTtlMs: 1000,
+    });
+    manager.acquire('session-1');
+    clock.advance(500);
+
+    expect(manager.heartbeat('session-1', 'wrong-lease')).toBeUndefined();
+    expect(manager.heartbeat('session-1', 'lease-1')).toMatchObject({
+      sessionId: 'session-1',
+      leaseId: 'lease-1',
+      acquiredAt: '2026-07-17T00:00:00.000Z',
+      lastHeartbeatAt: '2026-07-17T00:00:00.500Z',
+      expiresAt: '2026-07-17T00:00:01.500Z',
+    });
+
+    clock.advance(999);
+    expect(manager.get('session-1')).toMatchObject({ leaseId: 'lease-1' });
+  });
+});
+
+function fakeClock(start: string): {
+  now: () => Date;
+  advance: (ms: number) => void;
+} {
+  let nowMs = Date.parse(start);
+  return {
+    now: () => new Date(nowMs),
+    advance: (ms: number) => {
+      nowMs += ms;
+    },
+  };
+}