simstudioai--sim/apps/sim/background/async-execution-correlation.test.ts

/**
 * @vitest-environment node
 */

import { describe, expect, it } from 'vitest'
import {
  describeRetryableInfrastructureError,
  isRetryableInfrastructureError,
} from '@/lib/core/errors/retryable-infrastructure'
import {
  buildScheduleCorrelation,
  scheduleExecutionTaskOptions,
} from '@/background/schedule-execution'
import { buildWebhookCorrelation } from '@/background/webhook-execution'
import { buildWorkflowCorrelation } from '@/background/workflow-execution'

describe('async execution correlation fallbacks', () => {
  it('falls back for legacy workflow payloads missing correlation fields', () => {
    const correlation = buildWorkflowCorrelation({
      workflowId: 'workflow-1',
      userId: 'user-1',
      triggerType: 'api',
      executionId: 'execution-legacy',
    })

    expect(correlation).toEqual({
      executionId: 'execution-legacy',
      requestId: 'executio',
      source: 'workflow',
      workflowId: 'workflow-1',
      triggerType: 'api',
    })
  })

  it('falls back for legacy schedule payloads missing preassigned request id', () => {
    const correlation = buildScheduleCorrelation({
      scheduleId: 'schedule-1',
      workflowId: 'workflow-1',
      executionId: 'schedule-exec-1',
      now: '2025-01-01T00:00:00.000Z',
      scheduledFor: '2025-01-01T00:00:00.000Z',
    })

    expect(correlation).toEqual({
      executionId: 'schedule-exec-1',
      requestId: 'schedule',
      source: 'schedule',
      workflowId: 'workflow-1',
      scheduleId: 'schedule-1',
      triggerType: 'schedule',
      scheduledFor: '2025-01-01T00:00:00.000Z',
    })
  })

  it('caps schedule execution concurrency at the task queue', () => {
    expect(scheduleExecutionTaskOptions).toMatchObject({
      queue: {
        name: 'schedule-execution',
        concurrencyLimit: 30,
      },
    })
  })

  it('classifies retryable driver causes without treating every failed query as retryable', () => {
    const driverError = Object.assign(new Error('remaining connection slots are reserved'), {
      code: '53300',
    })
    const drizzleError = new Error('Failed query: select * from "environment"', {
      cause: driverError,
    })

    expect(isRetryableInfrastructureError(drizzleError)).toBe(true)
    expect(describeRetryableInfrastructureError(drizzleError)).toEqual(
      expect.objectContaining({
        code: '53300',
        message: 'remaining connection slots are reserved',
      })
    )
    expect(
      isRetryableInfrastructureError(new Error('remaining connection slots are reserved'))
    ).toBe(false)
    expect(
      isRetryableInfrastructureError(
        Object.assign(new Error('connect failed'), { code: 'ETIMEDOUT' })
      )
    ).toBe(true)
    expect(isRetryableInfrastructureError(new Error('Failed query: syntax error'))).toBe(false)
  })

  it('falls back for legacy webhook payloads missing preassigned fields', () => {
    const correlation = buildWebhookCorrelation({
      webhookId: 'webhook-1',
      workflowId: 'workflow-1',
      userId: 'user-1',
      executionId: 'webhook-exec-1',
      provider: 'slack',
      body: {},
      headers: {},
      path: 'incoming/slack',
    })

    expect(correlation).toEqual({
      executionId: 'webhook-exec-1',
      requestId: 'webhook-',
      source: 'webhook',
      workflowId: 'workflow-1',
      webhookId: 'webhook-1',
      path: 'incoming/slack',
      provider: 'slack',
      triggerType: 'webhook',
    })
  })
})