82 lines
3.1 KiB
TypeScript
82 lines
3.1 KiB
TypeScript
import { ExecutionVerdict, FlowExecutorContext } from '../../src/lib/handler/context/flow-execution-context'
|
|
import { runWithExponentialBackoff } from '../../src/lib/helper/error-handling'
|
|
import { buildCodeAction, generateMockEngineConstants } from '../handler/test-helper'
|
|
|
|
describe('runWithExponentialBackoff', () => {
|
|
const executionState = FlowExecutorContext.empty()
|
|
const action = buildCodeAction({
|
|
name: 'runtime',
|
|
input: {},
|
|
errorHandlingOptions: {
|
|
continueOnFailure: {
|
|
value: false,
|
|
},
|
|
retryOnFailure: {
|
|
value: true,
|
|
},
|
|
},
|
|
})
|
|
const constants = generateMockEngineConstants()
|
|
const requestFunction = jest.fn()
|
|
|
|
beforeEach(() => {
|
|
jest.clearAllMocks()
|
|
})
|
|
|
|
afterAll(() => {
|
|
jest.clearAllMocks()
|
|
})
|
|
|
|
it('should return resultExecutionState when verdict is not FAILED', async () => {
|
|
const resultExecutionState = FlowExecutorContext.empty().setVerdict(ExecutionVerdict.SUCCEEDED, undefined)
|
|
requestFunction.mockResolvedValue(resultExecutionState)
|
|
|
|
const output = await runWithExponentialBackoff(executionState, action, constants, requestFunction)
|
|
|
|
expect(output).toEqual(resultExecutionState)
|
|
expect(requestFunction).toHaveBeenCalledWith({ action, executionState, constants })
|
|
})
|
|
|
|
|
|
it('should retry and return resultExecutionState when verdict is FAILED and retry is enabled', async () => {
|
|
const resultExecutionState = FlowExecutorContext.empty().setVerdict(ExecutionVerdict.FAILED, undefined)
|
|
|
|
requestFunction.mockResolvedValue(resultExecutionState)
|
|
|
|
const output = await runWithExponentialBackoff(executionState, action, constants, requestFunction)
|
|
|
|
expect(output).toEqual(resultExecutionState)
|
|
// Mock applies for the first attempt and second attempt is a real call which return success
|
|
expect(requestFunction).toHaveBeenCalledTimes(2)
|
|
expect(requestFunction).toHaveBeenCalledWith({ action, executionState, constants })
|
|
expect(requestFunction).toHaveBeenCalledWith({ action, executionState, constants })
|
|
})
|
|
|
|
it('should not retry and return resultExecutionState when verdict is FAILED but retry is disabled', async () => {
|
|
const resultExecutionState = FlowExecutorContext.empty().setVerdict(ExecutionVerdict.FAILED, undefined)
|
|
|
|
requestFunction.mockResolvedValue(resultExecutionState)
|
|
|
|
|
|
const actionWithDisabledRetry = buildCodeAction({
|
|
name: 'runtime',
|
|
input: {},
|
|
errorHandlingOptions: {
|
|
continueOnFailure: {
|
|
value: false,
|
|
},
|
|
retryOnFailure: {
|
|
value: false,
|
|
},
|
|
},
|
|
})
|
|
|
|
const output = await runWithExponentialBackoff(executionState, actionWithDisabledRetry, constants, requestFunction)
|
|
|
|
expect(output).toEqual(resultExecutionState)
|
|
expect(requestFunction).toHaveBeenCalledTimes(1)
|
|
expect(requestFunction).toHaveBeenCalledWith({ action: actionWithDisabledRetry, executionState, constants })
|
|
|
|
})
|
|
|
|
}) |