Class MockAgent

packages/sqlite-runner/src/__tests__/sqlite-runner.test.ts:28–78

};

class MockAgent extends AbstractAgent {
  constructor(
    private readonly events: BaseEvent[] = [],
    private readonly emitDefaultRunStarted = true,
  ) {
    super();
  }

  async runAgent(input: RunAgentInput, callbacks: RunCallbacks): Promise<void> {
    if (this.emitDefaultRunStarted) {
      const runStarted: RunStartedEvent = {
        type: EventType.RUN_STARTED,
        threadId: input.threadId,
        runId: input.runId,
      };
      await callbacks.onEvent({ event: runStarted });
      await callbacks.onRunStartedEvent?.();
    }

    for (const event of this.events) {
      await callbacks.onEvent({ event });
    }

    const hasTerminalEvent = this.events.some(
      (event) =>
        event.type === EventType.RUN_FINISHED ||
        event.type === EventType.RUN_ERROR,
    );

    if (!hasTerminalEvent) {
      const runFinished: RunFinishedEvent = {
        type: EventType.RUN_FINISHED,
        threadId: input.threadId,
        runId: input.runId,
      };
      await callbacks.onEvent({ event: runFinished });
    }
  }

  protected run(): ReturnType<AbstractAgent["run"]> {
    return EMPTY;
  }

  protected connect(): ReturnType<AbstractAgent["connect"]> {
    return EMPTY;
  }

  clone(): AbstractAgent {
    return new MockAgent(this.events, this.emitDefaultRunStarted);
  }
}

class StoppableAgent extends AbstractAgent {
  private shouldStop = false;