/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.camel.component.aws.swf;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.Map;
import com.amazonaws.services.simpleworkflow.AmazonSimpleWorkflowClient;
import com.amazonaws.services.simpleworkflow.flow.DynamicWorkflowClientExternal;
import com.amazonaws.services.simpleworkflow.flow.DynamicWorkflowClientExternalImpl;
import com.amazonaws.services.simpleworkflow.model.DescribeWorkflowExecutionRequest;
import com.amazonaws.services.simpleworkflow.model.WorkflowExecution;
import com.amazonaws.services.simpleworkflow.model.WorkflowExecutionDetail;
import com.amazonaws.services.simpleworkflow.model.WorkflowExecutionInfo;
import org.junit.Before;
import org.junit.Test;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
public class CamelSWFWorkflowClientTest {
private SWFConfiguration configuration;
private AmazonSimpleWorkflowClient swClient;
private SWFEndpoint endpoint;
private CamelSWFWorkflowClient camelSWFWorkflowClient;
private DynamicWorkflowClientExternal clientExternal;
@Before
public void setUp() throws Exception {
configuration = new SWFConfiguration();
configuration.setDomainName("testDomain");
swClient = mock(AmazonSimpleWorkflowClient.class);
configuration.setAmazonSWClient(swClient);
configuration.setStartWorkflowOptionsParameters(Collections.<String, Object>emptyMap());
endpoint = new SWFEndpoint();
endpoint.setConfiguration(configuration);
clientExternal = mock(DynamicWorkflowClientExternalImpl.class);
camelSWFWorkflowClient = new CamelSWFWorkflowClient(endpoint, configuration) {
@Override
DynamicWorkflowClientExternal getDynamicWorkflowClient(String workflowId, String runId) {
return clientExternal;
}
};
}
@Test
public void testDescribeWorkflowInstance() throws Exception {
WorkflowExecutionInfo executionInfo = new WorkflowExecutionInfo();
executionInfo.setCloseStatus("COMPLETED");
Date closeTimestamp = new Date();
executionInfo.setCloseTimestamp(closeTimestamp);
executionInfo.setExecutionStatus("CLOSED");
executionInfo.setTagList(Collections.EMPTY_LIST);
WorkflowExecutionDetail workflowExecutionDetail = new WorkflowExecutionDetail();
workflowExecutionDetail.setExecutionInfo(executionInfo);
when(swClient.describeWorkflowExecution(any(DescribeWorkflowExecutionRequest.class))).thenReturn(workflowExecutionDetail);
Map<String, Object> description = camelSWFWorkflowClient.describeWorkflowInstance("123", "run1");
DescribeWorkflowExecutionRequest describeRequest = new DescribeWorkflowExecutionRequest();
describeRequest.setDomain(configuration.getDomainName());
describeRequest.setExecution(new WorkflowExecution().withWorkflowId("123").withRunId("run1"));
verify(swClient).describeWorkflowExecution(describeRequest);
assertThat((String) description.get("closeStatus"), is("COMPLETED"));
assertThat((Date) description.get("closeTimestamp"), is(closeTimestamp));
assertThat((String) description.get("executionStatus"), is("CLOSED"));
assertThat((List) description.get("tagList"), is(Collections.EMPTY_LIST));
assertThat((WorkflowExecutionDetail) description.get("executionDetail"), is(workflowExecutionDetail));
}
@Test
public void testSignalWorkflowExecution() throws Exception {
camelSWFWorkflowClient.signalWorkflowExecution("123", "run1", "signalMethod", "Hi");
verify(clientExternal).signalWorkflowExecution("signalMethod", new Object[] {"Hi"});
}
@Test
public void testGetWorkflowExecutionState() throws Throwable {
Class<String> stateType = String.class;
when(clientExternal.getWorkflowExecutionState(stateType)).thenReturn("some state");
String state = (String) camelSWFWorkflowClient.getWorkflowExecutionState("123", "run1", stateType);
verify(clientExternal).getWorkflowExecutionState(stateType);
assertThat(state, is("some state"));
}
@Test
public void testRequestCancelWorkflowExecution() throws Throwable {
camelSWFWorkflowClient.requestCancelWorkflowExecution("123", "run1");
verify(clientExternal).requestCancelWorkflowExecution();
}
@Test
public void testTerminateWorkflowExecution() throws Throwable {
camelSWFWorkflowClient.terminateWorkflowExecution("123", "run1", "reason", "details", null);
verify(clientExternal).terminateWorkflowExecution("reason", "details", null);
}
@Test
public void testStartWorkflowExecution() throws Throwable {
WorkflowExecution workflowExecution = new WorkflowExecution();
workflowExecution.setWorkflowId("123");
workflowExecution.setRunId("run1");
when(clientExternal.getWorkflowExecution()).thenReturn(workflowExecution);
String[] ids = camelSWFWorkflowClient.startWorkflowExecution(null, null, "eventName", "version", null);
verify(clientExternal).startWorkflowExecution(new Object[]{null});
assertThat("123", is(ids[0]));
assertThat("run1", is(ids[1]));
}
}