Class MockAgent

packages/core/src/__tests__/test-utils.ts:30–130

}

export class MockAgent {
  public messages: Message[] = [];
  public state: Record<string, any> = {};
  public agentId?: string;
  public threadId?: string;
  public addMessages = vi.fn((messages: Message[]) => {
    this.messages.push(...messages);
  });
  public addMessage = vi.fn((message: Message) => {
    this.messages.push(message);
    // Also track on parent if this is a clone
    if (this._parentAgent) {
      this._parentAgent.addMessage(message);
    }
  });
  public abortRun = vi.fn();
  public clone = vi.fn(() => this._cloneImpl());

  private newMessages: Message[];
  private error?: Error | string;
  private runAgentDelay: number;
  public runAgentCallback?: (input: any) => void;
  public runAgentCalls: any[] = [];
  private _parentAgent?: MockAgent;

  constructor(options: MockAgentOptions = {}) {
    this.messages = options.messages || [];
    this.newMessages = options.newMessages || [];
    this.error = options.error;
    this.runAgentDelay = options.runAgentDelay || 0;
    this.runAgentCallback = options.runAgentCallback;
    this.agentId = options.agentId;
    this.threadId = options.threadId;
    this.state = options.state || {};
  }

  async runAgent(
    input: any,
    subscriber?: any,
  ): Promise<{ newMessages: Message[] }> {
    this.runAgentCalls.push(input);
    // Also track on parent if this is a clone
    if (this._parentAgent) {
      this._parentAgent.runAgentCalls.push(input);
    }

    if (this.runAgentCallback) {
      this.runAgentCallback(input);
    }

    if (this.runAgentDelay > 0) {
      await new Promise((resolve) => setTimeout(resolve, this.runAgentDelay));
    }

    if (this.error) {
      throw this.error;
    }