Class FakeRpc

test/bindingRuntime.prompt.test.ts:21–176

import type { JsonRpcMessage, JsonRpcRequest, JsonRpcResponse } from '../src/acp/jsonrpc.js';

class FakeRpc implements StdioProcess {
  private messageHandlers: Array<(m: JsonRpcMessage) => void> = [];
  private stderrHandlers: Array<(line: string) => void> = [];
  written: JsonRpcMessage[] = [];

  private promptRequestId: number | null = null;
  private sessionId = 'sess-1';
  private workspaceFile: string;

  constructor(params: { workspaceFile: string }) {
    this.workspaceFile = params.workspaceFile;
  }

  write(message: JsonRpcMessage): void {
    this.written.push(message);

    // Client -> Agent requests
    if ('method' in message) {
      const req = message as JsonRpcRequest;

      if (req.method === 'initialize') {
        queueMicrotask(() => {
          this.emit({
            jsonrpc: '2.0',
            id: req.id,
            result: {
              protocolVersion: 1,
              agentCapabilities: { loadSession: false },
            },
          } as JsonRpcResponse);
        });
        return;
      }

      if (req.method === 'session/new') {
        queueMicrotask(() => {
          this.emit({
            jsonrpc: '2.0',
            id: req.id,
            result: { sessionId: this.sessionId },
          } as JsonRpcResponse);
        });
        return;
      }

      if (req.method === 'session/prompt') {
        this.promptRequestId = Number(req.id);

        queueMicrotask(() => {
          // emit a tool call update (UI)
          this.emit({
            jsonrpc: '2.0',
            method: 'session/update',
            params: {
              sessionId: this.sessionId,
              update: {
                sessionUpdate: 'tool_call_update',
                title: 'terminal/create',