dify/e2e/tests/cleanup.test.ts

import { describe, expect, it, vi } from 'vitest'
import { runCleanupTasks, shouldFailForCleanupErrors } from '../support/cleanup'

describe('runCleanupTasks', () => {
  it('runs every task in order', async () => {
    const order: string[] = []

    const errors = await runCleanupTasks([
      {
        label: 'first',
        run: () => {
          order.push('first')
        },
      },
      {
        label: 'second',
        run: async () => {
          order.push('second')
        },
      },
    ])

    expect(order).toEqual(['first', 'second'])
    expect(errors).toEqual([])
  })

  it('continues after failures and labels every error', async () => {
    const finalTask = vi.fn()

    const errors = await runCleanupTasks([
      {
        label: 'sync cleanup',
        run: () => {
          throw new Error('sync failure')
        },
      },
      {
        label: 'async cleanup',
        run: async () => {
          throw new Error('async failure')
        },
      },
      { label: 'final cleanup', run: finalTask },
    ])

    expect(finalTask).toHaveBeenCalledOnce()
    expect(errors).toEqual(['sync cleanup: sync failure', 'async cleanup: async failure'])
  })
})

describe('shouldFailForCleanupErrors', () => {
  it.each(['PASSED', 'SKIPPED'])('fails a %s scenario when cleanup fails', (status) => {
    expect(shouldFailForCleanupErrors(status)).toBe(true)
  })

  it.each(['FAILED', 'AMBIGUOUS', 'PENDING', 'UNDEFINED', 'UNKNOWN'])(
    'preserves an existing %s result when cleanup fails',
    (status) => {
      expect(shouldFailForCleanupErrors(status)).toBe(false)
    },
  )
})