void executeRetryStepFunction(WorkflowExecution execution, String stepFunctionArn) { retryTemplate.execute { RetryContext context -> workflowExecutionInitiatorService.addExecuteStepFunction(execution, stepFunctionArn) } }

@Test
public void executeRetryStepFunction_whenException_retried() {
    // GIVEN
    WorkflowExecution execution = WorkflowExecutionFixture.someWorkflowExecution();
    String stepFunctionArn = "stepFunctionArn";

    // WHEN
    when(workflowExecutionInitiatorService.addExecuteStepFunction(execution, stepFunctionArn)).thenThrow(new RetryableException("some error"))
        .thenReturn(null);

    try {
        workflowInitiatorService.executeRetryStepFunction(execution, stepFunctionArn);
        fail("expected RetryableException");
    } catch (RetryableException ex) {
        // THEN
        verify(workflowExecutionInitiatorService, times(2)).addExecuteStepFunction(any(), any());
    }
}