Class ExecutionContextBuilder

packages/testing/src/builders/execution.builder.ts:20–224

 * ```
 */
export class ExecutionContextBuilder {
  private workflowId = 'test-workflow'
  private executionId = `exec-${generateRandomString(8)}`
  private blockStates = new Map<string, any>()
  private executedBlocks = new Set<string>()
  private blockLogs: any[] = []
  private metadata: { duration: number; startTime?: string; endTime?: string } = { duration: 0 }
  private environmentVariables: Record<string, string> = {}
  private workflowVariables: Record<string, any> = {}
  private routerDecisions = new Map<string, any>()
  private conditionDecisions = new Map<string, any>()
  private loopExecutions = new Map<string, any>()
  private completedLoops = new Set<string>()
  private activeExecutionPath = new Set<string>()
  private abortSignal?: AbortSignal

  /**
   * Sets the workflow ID.
   */
  forWorkflow(workflowId: string): this {
    this.workflowId = workflowId
    return this
  }

  /**
   * Sets a custom execution ID.
   */
  withExecutionId(executionId: string): this {
    this.executionId = executionId
    return this
  }

  /**
   * Adds a block state.
   */
  withBlockState(blockId: string, state: any): this {
    this.blockStates.set(blockId, state)
    return this
  }

  /**
   * Adds multiple block states at once.
   */
  withBlockStates(states: Record<string, any>): this {
    Object.entries(states).forEach(([id, state]) => {
      this.blockStates.set(id, state)
    })
    return this
  }

  /**
   * Marks a block as executed.
   */
  markExecuted(blockId: string): this {
    this.executedBlocks.add(blockId)
    return this
  }