Function loadExecutionModule

apps/sim/lib/execution/isolated-vm.test.ts:213–275

(options: {
  envOverrides?: Record<string, string>
  spawns: SpawnFactory[]
  redisEvalImpl?: RedisEval
  secureFetchImpl?: SecureFetchImpl
})

}))

async function loadExecutionModule(options: {
  envOverrides?: Record<string, string>
  spawns: SpawnFactory[]
  redisEvalImpl?: RedisEval
  secureFetchImpl?: SecureFetchImpl
}) {
  const spawnQueue = [...options.spawns]
  mockSpawn.mockImplementation(() => {
    const next = spawnQueue.shift()
    if (!next) {
      throw new Error('No mock spawn factory configured')
    }
    return next() as ReturnType<typeof mockSpawn>
  })

  if (options.secureFetchImpl) {
    mockSecureFetch.mockImplementation(options.secureFetchImpl)
  } else {
    mockSecureFetch.mockImplementation(async () => ({
      ok: true,
      status: 200,
      statusText: 'OK',
      headers: new Map<string, string>(),
      text: async () => '',
      json: async () => ({}),
      arrayBuffer: async () => new ArrayBuffer(0),
    }))
  }

  Object.assign(mockEnv, {
    IVM_POOL_SIZE: '1',
    IVM_MAX_CONCURRENT: '100',
    IVM_MAX_PER_WORKER: '100',
    IVM_WORKER_IDLE_TIMEOUT_MS: '60000',
    IVM_MAX_QUEUE_SIZE: '10',
    IVM_MAX_ACTIVE_PER_OWNER: '100',
    IVM_MAX_QUEUED_PER_OWNER: '10',
    IVM_MAX_OWNER_WEIGHT: '5',
    IVM_DISTRIBUTED_MAX_INFLIGHT_PER_OWNER: '100',
    IVM_DISTRIBUTED_LEASE_MIN_TTL_MS: '1000',
    IVM_QUEUE_TIMEOUT_MS: '1000',
    IVM_MAX_FETCH_RESPONSE_BYTES: '',
    IVM_MAX_FETCH_RESPONSE_CHARS: '',
    IVM_MAX_FETCH_URL_LENGTH: '',
    IVM_MAX_FETCH_OPTIONS_JSON_CHARS: '',
    REDIS_URL: '',
    ...(options.envOverrides ?? {}),
  })

  const redisEval = options.redisEvalImpl ? vi.fn(options.redisEvalImpl) : undefined
  mockGetRedisClient.mockImplementation(() =>
    redisEval
      ? ({
          eval: redisEval,
        } as unknown)
      : null
  )