Java Code Examples for org.springframework.statemachine.action.Action#execute()

The following examples show how to use org.springframework.statemachine.action.Action#execute() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: EnvCreationActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
private void testNoEnvironment(Supplier<Action<?, ?>> creationAction, String selector) {
    Action<?, ?> action = configureAction(creationAction);

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.empty());

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), anyString());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyCreationActionFailureEvent(selector);
}
Example 2
Source File: ExternalDatabaseTerminationActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void externalDatabaseCreationFinishedAction() {
    TerminateExternalDatabaseResult terminateExternalDatabaseResultPayload =
            new TerminateExternalDatabaseResult(STACK_ID, EXTERNAL_DATABASE_WAIT_SUCCESS_EVENT.event(), STACK_NAME, STACK_CRN);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(terminateExternalDatabaseResultPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseTerminationFinishedAction);
    action.execute(stateContext);

    verifyNoMoreInteractions(stackUpdaterService);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    verify(metricService).incrementMetricCounter(MetricType.EXTERNAL_DATABASE_TERMINATION_SUCCESSFUL, STACK);
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("EXTERNAL_DATABASE_TERMINATION_FINISHED_EVENT");
    Object capturedPayload = payloadArgumentCaptor.getValue();
    assertThat(capturedPayload).isInstanceOf(StackEvent.class);
    StackEvent stackEvent = (StackEvent) capturedPayload;
    assertThat(stackEvent.getResourceId()).isEqualTo(STACK_ID);
}
Example 3
Source File: ExternalDatabaseTerminationActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@ParameterizedTest
@ValueSource(booleans = { false, true })
void externalDatabaseTermination(boolean forced) {
    TerminationEvent terminationEventPayload = new TerminationEvent(ACTION_PAYLOAD_SELECTOR, STACK_ID, forced);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(terminationEventPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseTermination);
    action.execute(stateContext);
    verifyNoMoreInteractions(stackUpdaterService);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("TerminateExternalDatabaseRequest");
    Object rawPayload = payloadArgumentCaptor.getValue();
    assertThat(rawPayload).isInstanceOf(TerminateExternalDatabaseRequest.class);
    TerminateExternalDatabaseRequest request = (TerminateExternalDatabaseRequest) rawPayload;
    assertThat(request.isForced()).isEqualTo(forced);
}
Example 4
Source File: ExternalDatabaseCreationActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void externalDatabaseCreationFinishedAction() {
    CreateExternalDatabaseResult createExternalDatabaseResultPayload =
            new CreateExternalDatabaseResult(STACK_ID, EXTERNAL_DATABASE_WAIT_SUCCESS_EVENT.event(), STACK_NAME, STACK_CRN);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(createExternalDatabaseResultPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseCreationFinishedAction);
    action.execute(stateContext);

    verifyNoMoreInteractions(stackUpdaterService);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    verify(metricService).incrementMetricCounter(MetricType.EXTERNAL_DATABASE_CREATION_SUCCESSFUL, STACK);
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("EXTERNAL_DATABASE_CREATION_FINISHED_EVENT");
    Object capturedPayload = payloadArgumentCaptor.getValue();
    assertThat(capturedPayload).isInstanceOf(StackEvent.class);
    StackEvent stackEvent = (StackEvent) capturedPayload;
    assertThat(stackEvent.getResourceId()).isEqualTo(STACK_ID);
}
Example 5
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datalakeClustersDeleteAction() {
    Action<?, ?> action = configureAction(() -> underTest.datalakeClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.of(environment));
    when(environmentService.save(environment)).thenReturn(savedEnvironment);
    when(environmentService.getEnvironmentDto(savedEnvironment)).thenReturn(environmentDto);
    when(environmentResponseConverter.dtoToSimpleResponse(environmentDto)).thenReturn(simpleEnvironmentResponse);

    action.execute(stateContext);

    verify(environment).setStatus(EnvironmentStatus.DATALAKE_CLUSTERS_DELETE_IN_PROGRESS);
    verify(notificationService).send(ResourceEvent.ENVIRONMENT_DATALAKE_CLUSTERS_DELETION_STARTED, simpleEnvironmentResponse, FLOW_TRIGGER_USER_CRN);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyDeleteActionSuccessEvent(DELETE_DATALAKE_CLUSTERS_EVENT);
}
Example 6
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datalakeClustersDeleteActionNoEnvironment() {
    Action<?, ?> action = configureAction(() -> underTest.datalakeClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.empty());

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(environment, never()).setStatus(any());
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), any());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyDeleteActionSuccessEvent(DELETE_DATALAKE_CLUSTERS_EVENT);
}
Example 7
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datalakeClustersDeleteActionFailure() {
    Action<?, ?> action = configureAction(() -> underTest.datalakeClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenThrow(new UnsupportedOperationException(MESSAGE));
    when(failureEvent.event()).thenReturn(FAILURE_EVENT);

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), anyString());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyFailureEvent();
}
Example 8
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datahubClustersDeleteAction() {
    Action<?, ?> action = configureAction(() -> underTest.datahubClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.of(environment));
    when(environmentService.save(environment)).thenReturn(savedEnvironment);
    when(environmentService.getEnvironmentDto(savedEnvironment)).thenReturn(environmentDto);
    when(environmentResponseConverter.dtoToSimpleResponse(environmentDto)).thenReturn(simpleEnvironmentResponse);

    action.execute(stateContext);

    verify(environment).setStatus(EnvironmentStatus.DATAHUB_CLUSTERS_DELETE_IN_PROGRESS);
    verify(notificationService).send(ResourceEvent.ENVIRONMENT_DATAHUB_CLUSTERS_DELETION_STARTED, simpleEnvironmentResponse, FLOW_TRIGGER_USER_CRN);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyDeleteActionSuccessEvent(DELETE_DATAHUB_CLUSTERS_EVENT);
}
Example 9
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datahubClustersDeleteActionNoEnvironment() {
    Action<?, ?> action = configureAction(() -> underTest.datahubClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.empty());

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(environment, never()).setStatus(any());
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), any());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyDeleteActionSuccessEvent(DELETE_DATAHUB_CLUSTERS_EVENT);
}
Example 10
Source File: EnvClustersDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
@Test
void datahubClustersDeleteActionFailure() {
    Action<?, ?> action = configureAction(() -> underTest.datahubClustersDeleteAction());

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenThrow(new UnsupportedOperationException(MESSAGE));
    when(failureEvent.event()).thenReturn(FAILURE_EVENT);

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), anyString());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyFailureEvent();
}
Example 11
Source File: EnvDeleteActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
private void testDeleteActionHappyPath(Supplier<Action<?, ?>> deleteAction,
        String selector,
        EnvironmentStatus environmentStatus,
        ResourceEvent eventStarted) {

    Action<?, ?> action = configureAction(deleteAction);

    EnvironmentDto environmentDto = mock(EnvironmentDto.class);
    when(environmentStatusUpdateService.updateEnvironmentStatusAndNotify(any(), any(), any(), any(), any())).thenReturn(environmentDto);

    action.execute(stateContext);

    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    verify(environmentStatusUpdateService).updateEnvironmentStatusAndNotify(any(), any(), eq(environmentStatus), eq(eventStarted), any());
}
Example 12
Source File: EnvCreationActionsTest.java    From cloudbreak with Apache License 2.0 6 votes vote down vote up 
private void testFailure(Supplier<Action<?, ?>> creationAction) {
    Action<?, ?> action = configureAction(creationAction);

    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenThrow(new UnsupportedOperationException(MESSAGE));
    when(failureEvent.event()).thenReturn(FAILURE_EVENT);

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), anyString());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyFailureEvent();
}
Example 13
Source File: ExternalDatabaseCreationActionsTest.java    From cloudbreak with Apache License 2.0 5 votes vote down vote up 
@Test
void externalDatabaseCreation() {
    StackEvent stackEventPayload = new StackEvent(ACTION_PAYLOAD_SELECTOR, STACK_ID);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(stackEventPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseCreation);
    action.execute(stateContext);
    verifyNoMoreInteractions(stackUpdaterService);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("CreateExternalDatabaseRequest");
}
Example 14
Source File: EnvDeleteActionsTest.java    From cloudbreak with Apache License 2.0 5 votes vote down vote up 
private void testNoEnvironment(Supplier<Action<?, ?>> deleteAction, String selector) {
    Action<?, ?> action = configureAction(deleteAction);

    action.execute(stateContext);

    verify(environmentService, never()).save(any(Environment.class));
    verify(environmentService, never()).getEnvironmentDto(any(Environment.class));
    verify(environmentResponseConverter, never()).dtoToSimpleResponse(any(EnvironmentDto.class));
    verify(notificationService, never()).send(any(ResourceEvent.class), any(SimpleEnvironmentResponse.class), anyString());
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
}
Example 15
Source File: ExternalDatabaseCreationActionsTest.java    From cloudbreak with Apache License 2.0 5 votes vote down vote up 
@Test
void externalDatabaseCreationFailureAction() {
    RuntimeException expectedException = new RuntimeException(MESSAGE);
    CreateExternalDatabaseFailed createExternalDatabaseFailedPayload =
            new CreateExternalDatabaseFailed(STACK_ID,  EXTERNAL_DATABASE_CREATION_FAILED_EVENT.event(),
                    STACK_NAME, null, expectedException);

    when(stackService.getByIdWithClusterInTransaction(any())).thenReturn(STACK);

    when(runningFlows.get(anyString())).thenReturn(flow);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(createExternalDatabaseFailedPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseCreationFailureAction);
    action.execute(stateContext);

    verify(stackUpdaterService).updateStatus(STACK_ID, DetailedStackStatus.EXTERNAL_DATABASE_CREATION_FAILED,
            ResourceEvent.CLUSTER_EXTERNAL_DATABASE_CREATION_FAILED, MESSAGE);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    verify(metricService).incrementMetricCounter(MetricType.EXTERNAL_DATABASE_CREATION_FAILED, STACK);
    verify(flow).setFlowFailed(exceptionCaptor.capture());
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("STACK_CREATION_FAILHANDLED");
    Object capturedPayload = payloadArgumentCaptor.getValue();
    assertThat(capturedPayload).isInstanceOf(StackEvent.class);
    StackEvent stackEvent = (StackEvent) capturedPayload;
    assertThat(stackEvent.getResourceId()).isEqualTo(STACK_ID);
    Exception exception = exceptionCaptor.getValue();
    assertThat(exception).isEqualTo(expectedException);
}
Example 16
Source File: EnvCreationActionsTest.java    From cloudbreak with Apache License 2.0 5 votes vote down vote up 
private void testCreationActionHappyPath(Supplier<Action<?, ?>> creationAction,
        String selector,
        EnvironmentStatus environmentStatus,
        ResourceEvent eventStarted) {

    Action<?, ?> action = configureAction(creationAction);

    Environment environment = mock(Environment.class);
    when(environmentService.findEnvironmentById(ENVIRONMENT_ID)).thenReturn(Optional.of(environment));
    Environment savedEnvironment = mock(Environment.class);
    when(environmentService.save(environment)).thenReturn(savedEnvironment);
    EnvironmentDto environmentDto = mock(EnvironmentDto.class);
    when(environmentDto.getResourceCrn()).thenReturn(ENVIRONMENT_CRN);
    when(environmentDto.getName()).thenReturn(ENVIRONMENT_NAME);
    when(environmentDto.getId()).thenReturn(ENVIRONMENT_ID);
    when(environmentService.getEnvironmentDto(savedEnvironment)).thenReturn(environmentDto);
    SimpleEnvironmentResponse response = mock(SimpleEnvironmentResponse.class);
    when(environmentResponseConverter.dtoToSimpleResponse(environmentDto)).thenReturn(response);

    action.execute(stateContext);

    verify(environment).setStatus(environmentStatus);
    verify(notificationService).send(eventStarted, response, FLOW_TRIGGER_USER_CRN);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());

    verifyCreationActionSuccessEvent(selector);
}
Example 17
Source File: ExternalDatabaseTerminationActionsTest.java    From cloudbreak with Apache License 2.0 5 votes vote down vote up 
@Test
void externalDatabaseCreationFailureAction() {
    RuntimeException expectedException = new RuntimeException(MESSAGE);
    TerminateExternalDatabaseFailed terminateExternalDatabaseFailedPayload =
            new TerminateExternalDatabaseFailed(STACK_ID,  EXTERNAL_DATABASE_CREATION_FAILED_EVENT.event(),
                    STACK_NAME, null, expectedException);

    when(stackService.getByIdWithClusterInTransaction(any())).thenReturn(STACK);

    when(runningFlows.get(anyString())).thenReturn(flow);
    when(stateContext.getMessageHeader(MessageFactory.HEADERS.DATA.name())).thenReturn(terminateExternalDatabaseFailedPayload);
    Action<?, ?> action = configureAction(underTest::externalDatabaseTerminationFailureAction);
    action.execute(stateContext);

    verify(stackUpdaterService).updateStatus(STACK_ID, DetailedStackStatus.DELETE_FAILED,
            ResourceEvent.CLUSTER_EXTERNAL_DATABASE_DELETION_FAILED, MESSAGE);
    verify(eventBus).notify(selectorArgumentCaptor.capture(), eventArgumentCaptor.capture());
    verify(reactorEventFactory).createEvent(headersArgumentCaptor.capture(), payloadArgumentCaptor.capture());
    verify(metricService).incrementMetricCounter(MetricType.EXTERNAL_DATABASE_TERMINATION_FAILED, STACK);
    verify(flow).setFlowFailed(exceptionCaptor.capture());
    assertThat(selectorArgumentCaptor.getValue()).isEqualTo("TERMINATIONFAILHANDLED");
    Object capturedPayload = payloadArgumentCaptor.getValue();
    assertThat(capturedPayload).isInstanceOf(StackEvent.class);
    StackEvent stackEvent = (StackEvent) capturedPayload;
    assertThat(stackEvent.getResourceId()).isEqualTo(STACK_ID);
    Exception exception = exceptionCaptor.getValue();
    assertThat(exception).isEqualTo(expectedException);
}