Merge "Workflow Designer Activities Implementation"

2 files changed, 139 insertions, 0 deletions

diff --git a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksITTest.java b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksITTest.java
new file mode 100644
index 0000000000..9e1dac69e0
--- /dev/null
+++ b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksITTest.java
@@ -0,0 +1,103 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed 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.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.so.bpmn.infrastructure.appc.tasks;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+
+import java.util.HashMap;
+import java.util.Optional;
+import java.util.UUID;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.onap.appc.client.lcm.model.Action;
+import org.onap.so.bpmn.BaseTaskTest;
+import org.onap.so.bpmn.common.BuildingBlockExecution;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.GenericVnf;
+import org.onap.so.bpmn.servicedecomposition.generalobjects.RequestContext;
+import org.onap.so.bpmn.servicedecomposition.generalobjects.RequestParameters;
+import org.onap.so.db.catalog.beans.ControllerSelectionReference;
+import org.springframework.beans.factory.annotation.Autowired;
+
+public class AppcRunTasksITTest extends BaseTaskTest {
+	
+	@Autowired
+	private AppcRunTasks appcRunTasks;
+	
+	private GenericVnf genericVnf;
+	private RequestContext requestContext;
+	private String msoRequestId;
+
+	@Before
+	public void before() {
+		genericVnf = setGenericVnf();
+		msoRequestId = UUID.randomUUID().toString();
+		requestContext = setRequestContext();
+		requestContext.setMsoRequestId(msoRequestId);
+		gBBInput.setRequestContext(requestContext);
+	}
+	
+	@Test 
+	public void preProcessActivityTest() throws Exception {
+		appcRunTasks.preProcessActivity(execution);
+		assertEquals(execution.getVariable("actionQuiesceTraffic"), Action.QuiesceTraffic);
+		assertEquals(execution.getVariable("rollbackQuiesceTraffic"), false);		
+	}
+	
+	@Test
+	public void runAppcCommandTest() throws Exception {
+		Action action = Action.QuiesceTraffic;
+		ControllerSelectionReference controllerSelectionReference = new ControllerSelectionReference();
+		controllerSelectionReference.setControllerName("testName");
+		controllerSelectionReference.setActionCategory(action.toString());
+		controllerSelectionReference.setVnfType("testVnfType");
+		
+		doReturn(controllerSelectionReference).when(catalogDbClient).getControllerSelectionReferenceByVnfTypeAndActionCategory(genericVnf.getVnfType(), Action.QuiesceTraffic.toString());
+		
+		execution.setVariable("aicIdentity", "testAicIdentity");		
+		
+		String vnfId = genericVnf.getVnfId();
+		genericVnf.setIpv4OamAddress("testOamIpAddress");
+		String payload = "{\"testName\":\"testValue\",}";
+		RequestParameters requestParameters = new RequestParameters();
+		requestParameters.setPayload(payload);
+		gBBInput.getRequestContext().setRequestParameters(requestParameters);
+		
+		String controllerType = "testName";
+		HashMap<String, String> payloadInfo = new HashMap<String, String>();
+		payloadInfo.put("vnfName", "testVnfName1");		
+		payloadInfo.put("aicIdentity", "testAicIdentity");
+		payloadInfo.put("vnfHostIpAddress", "testOamIpAddress");
+		payloadInfo.put("vserverIdList", null);
+		payloadInfo.put("vfModuleId", null);
+		payloadInfo.put("identityUrl", null);
+		payloadInfo.put("vmIdList", null);
+		
+		doNothing().when(appCClient).runAppCCommand(action, msoRequestId, vnfId, Optional.of(payload), payloadInfo, controllerType);
+		
+		appcRunTasks.runAppcCommand(execution, action);
+		verify(appCClient, times(1)).runAppCCommand(action, msoRequestId, vnfId, Optional.of(payload), payloadInfo, controllerType);
+	}
+}
diff --git a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksTest.java b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksTest.java
new file mode 100644
index 0000000000..7cade7703a
--- /dev/null
+++ b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/infrastructure/appc/tasks/AppcRunTasksTest.java
@@ -0,0 +1,36 @@
+package org.onap.so.bpmn.infrastructure.appc.tasks;
+
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.mock;
+
+import org.junit.Test;
+import org.onap.appc.client.lcm.model.Action;
+import org.onap.so.bpmn.common.BuildingBlockExecution;
+
+public class AppcRunTasksTest {
+
+	
+	private AppcRunTasks appcRunTasks = new AppcRunTasks();
+	@Test
+	public void mapRollbackVariablesTest() {
+		
+		BuildingBlockExecution mock = mock(BuildingBlockExecution.class);
+		
+		appcRunTasks.mapRollbackVariables(mock, Action.Lock, "1");
+		verify(mock, times(0)).setVariable(any(String.class), any());
+		appcRunTasks.mapRollbackVariables(mock, Action.Lock, "0");
+		verify(mock, times(1)).setVariable("rollbackVnfLock", true);
+		appcRunTasks.mapRollbackVariables(mock, Action.Unlock, "0");
+		verify(mock, times(1)).setVariable("rollbackVnfLock", false);
+		appcRunTasks.mapRollbackVariables(mock, Action.Start, "0");
+		verify(mock, times(1)).setVariable("rollbackVnfStop", false);
+		appcRunTasks.mapRollbackVariables(mock, Action.Stop, "0");
+		verify(mock, times(1)).setVariable("rollbackVnfStop", true);
+		appcRunTasks.mapRollbackVariables(mock, Action.QuiesceTraffic, "0");
+		verify(mock, times(1)).setVariable("rollbackQuiesceTraffic", true);
+		appcRunTasks.mapRollbackVariables(mock, Action.ResumeTraffic, "0");
+		verify(mock, times(1)).setVariable("rollbackQuiesceTraffic", false);
+	}
+}