1 files changed, 312 insertions, 0 deletions
diff --git a/appc-dispatcher/appc-command-executor/appc-command-executor-core/src/test/java/org/openecomp/appc/executor/TestCommandExecutionTask.java b/appc-dispatcher/appc-command-executor/appc-command-executor-core/src/test/java/org/openecomp/appc/executor/TestCommandExecutionTask.java
new file mode 100644
index 000000000..0504c3998
--- /dev/null
+++ b/appc-dispatcher/appc-command-executor/appc-command-executor-core/src/test/java/org/openecomp/appc/executor/TestCommandExecutionTask.java
@@ -0,0 +1,312 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * openECOMP : APP-C
+ * ================================================================================
+ * 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.openecomp.appc.executor;
+/**
+ *
+ */
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mockito;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+import org.openecomp.appc.domainmodel.lcm.*;
+import org.openecomp.appc.executor.impl.CommandTask;
+import org.openecomp.appc.executor.impl.CommandTaskFactory;
+import org.openecomp.appc.executor.impl.LCMCommandTask;
+import org.openecomp.appc.executor.impl.LCMReadonlyCommandTask;
+import org.openecomp.appc.executor.impl.objects.CommandRequest;
+import org.openecomp.appc.executor.impl.objects.LCMCommandRequest;
+import org.openecomp.appc.executor.impl.objects.LCMReadOnlyCommandRequest;
+import org.openecomp.appc.executor.objects.*;
+import org.openecomp.appc.lifecyclemanager.LifecycleManager;
+import org.openecomp.appc.requesthandler.RequestHandler;
+import org.openecomp.appc.workflow.WorkFlowManager;
+import org.openecomp.appc.workflow.objects.WorkflowRequest;
+import org.openecomp.appc.workflow.objects.WorkflowResponse;
+import org.openecomp.sdnc.sli.SvcLogicContext;
+import org.openecomp.sdnc.sli.SvcLogicException;
+import org.openecomp.sdnc.sli.SvcLogicResource;
+import org.openecomp.sdnc.sli.aai.AAIService;
+import org.osgi.framework.Bundle;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.FrameworkUtil;
+import org.osgi.framework.ServiceReference;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Properties;
+
+import static junit.framework.Assert.assertEquals;
+import static org.mockito.Matchers.*;
+
+
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest( {FrameworkUtil.class, CommandTask.class, LCMCommandTask.class})
+public class TestCommandExecutionTask {
+
+	private static final String TTL_FLAG= "TTL";
+	private static final String API_VERSION= "2.0.0";
+	private static final String ORIGINATOR_ID= "1";
+	private LCMCommandTask executionTask;
+	private LCMReadonlyCommandTask LCMReadonlyCommandTask;
+	private CommandTaskFactory factory ;
+
+	private RequestHandler requestHandler;
+	private WorkFlowManager workflowManager;
+	private AAIService aaiService;
+	private LifecycleManager lifecyclemanager;
+
+	private final BundleContext bundleContext=Mockito.mock(BundleContext.class);
+	private final Bundle bundleService=Mockito.mock(Bundle.class);
+	private final ServiceReference sref=Mockito.mock(ServiceReference.class);
+
+	@Before
+	public void init() throws SvcLogicException {
+
+		// ***
+		AAIService aaiService = Mockito.mock(AAIService.class);;
+		PowerMockito.mockStatic(FrameworkUtil.class);
+		PowerMockito.when(FrameworkUtil.getBundle(AAIService.class)).thenReturn(bundleService);
+		PowerMockito.when(bundleService.getBundleContext()).thenReturn(bundleContext);
+		PowerMockito.when(bundleContext.getServiceReference(AAIService.class.getName())).thenReturn(sref);
+		PowerMockito.when(bundleContext.getService(sref)).thenReturn(aaiService);
+		PowerMockito.when(aaiService.query(anyString(),anyBoolean(),anyString(),anyString(),anyString(),anyString(),(SvcLogicContext)anyObject())).thenAnswer(new Answer<SvcLogicResource.QueryStatus>() {
+			@Override
+			public SvcLogicResource.QueryStatus answer(InvocationOnMock invocation) throws Throwable {
+				Object[] args = invocation.getArguments();
+				SvcLogicContext ctx =(SvcLogicContext)args[6];
+				String prefix = (String)args[4];
+				String key = (String)args[3];
+				if(key.contains("'28'")){
+					return  SvcLogicResource.QueryStatus.FAILURE ;
+				}else if ( key.contains("'8'")) {
+					return  SvcLogicResource.QueryStatus.NOT_FOUND ;
+				}else {
+					ctx.setAttribute(prefix + ".vnf-type", "FIREWALL");
+					ctx.setAttribute(prefix + ".orchestration-status", "INSTANTIATED");
+				}
+				return  SvcLogicResource.QueryStatus.SUCCESS ;
+			}
+		});
+		PowerMockito.when(aaiService.update(anyString(),anyString(),(Map)anyObject(),anyString(),(SvcLogicContext)anyObject())).thenReturn(SvcLogicResource.QueryStatus.SUCCESS);
+
+		requestHandler =  Mockito.mock(RequestHandler.class);
+		workflowManager = Mockito.mock(WorkFlowManager.class);
+		lifecyclemanager = Mockito.mock(LifecycleManager.class );
+
+		executionTask = new LCMCommandTask(requestHandler,workflowManager,lifecyclemanager);
+		LCMReadonlyCommandTask = new LCMReadonlyCommandTask(requestHandler,workflowManager);
+		factory = new CommandTaskFactory();
+		factory.setLifecyclemanager(lifecyclemanager);
+		factory.setWorkflowManager(workflowManager);
+		factory.setRequestHandler(requestHandler);
+		Mockito.when(workflowManager.executeWorkflow((WorkflowRequest)anyObject())).thenReturn(getWorkflowResponse () );
+	}
+
+
+	@Test
+	public void testFactory(){
+		CommandTask task;
+		task = factory.getExecutionTask("Configure");
+		assertEquals(LCMCommandTask.class,task.getClass() );
+		task = factory.getExecutionTask("Sync");
+		assertEquals(LCMReadonlyCommandTask.class,task.getClass() );
+
+	}
+
+
+
+	@Test
+	public void testOnRequestCompletion(){
+		Mockito.doNothing().when(requestHandler).onRequestTTLEnd((RuntimeContext) anyObject(),anyBoolean());
+		LCMCommandRequest request = getLCMCommandRequest("FIREWALL",30,new Date(), "11" ,setTTLInFlags("30"), VNFOperation.Configure, "1", "1.0");
+		CommandResponse response = getCommandResponse(VNFOperation.Configure, true, "11", "","1");
+		executionTask.onRequestCompletion(request, response);
+	}
+
+	@Test
+	public void testRunGetConfig(){
+		LCMReadOnlyCommandRequest request = getConfigCommandRequest("FIREWALL",30,new Date(), "11" ,setTTLInFlags("30"),VNFOperation.Sync, "1", "1.0");
+		LCMReadonlyCommandTask.setCommandRequest(request);
+		LCMReadonlyCommandTask.run();
+	}
+
+	@Test
+	public void testRun(){
+		LCMCommandRequest request = getLCMCommandRequest("FIREWALL",30,new Date(), "11" ,setTTLInFlags("30"),VNFOperation.Sync, "1", "1.0");
+		executionTask.setCommandRequest(request);
+		executionTask.run();
+	}
+
+	@Test
+	public void testRunNegative(){
+		LCMCommandRequest request = getLCMCommandRequest("FIREWALL",30,new Date(), "11" ,setTTLInFlags("30"),VNFOperation.Sync, "1", "1.0");
+		executionTask.setCommandRequest(request);
+		executionTask.run();
+	}
+
+
+	CommandResponse getCommandResponse(VNFOperation action , boolean success, String responseId, String payload, String vnfId){
+		CommandResponse commandResponse = new CommandResponse();
+		RuntimeContext runtimeContext = new RuntimeContext();
+		commandResponse.setRuntimeContext(runtimeContext);
+		ResponseContext responseContext = new ResponseContext();
+		runtimeContext.setResponseContext(responseContext);
+		RequestContext requestContext = new RequestContext();
+		runtimeContext.setRequestContext(requestContext);
+		CommonHeader commonHeader = new CommonHeader();
+		requestContext.setCommonHeader(commonHeader);
+		responseContext.setCommonHeader(commonHeader);
+		Flags flags = new Flags();
+		commonHeader.setFlags(flags);
+		ActionIdentifiers actionIdentifiers = new ActionIdentifiers();
+		requestContext.setActionIdentifiers(actionIdentifiers);
+		VNFContext vnfContext = new VNFContext();
+		runtimeContext.setVnfContext(vnfContext);
+		requestContext.setAction(action);
+		runtimeContext.setRpcName(action.name().toLowerCase());
+		commonHeader.setApiVer(API_VERSION);
+		Status status = new Status();
+		status.setCode(100);
+		responseContext.setStatus(status);
+		commonHeader.setRequestId(responseId);
+		responseContext.setPayload(payload);
+		commonHeader.setTimestamp(new Date());
+		vnfContext.setId(vnfId);
+		return commandResponse;
+	}
+
+
+
+	@Test
+	public void testPositiveFlow_configure()  {
+
+		Date timeStamp = new Date();
+		String requestId = "1";
+
+		CommandExecutorInput commandExecutorInput = pouplateCommandExecutorInput("FIREWALL",30, "1.0", timeStamp, API_VERSION, requestId, ORIGINATOR_ID, "", VNFOperation.Configure, "33", "");
+		CommandRequest request = new CommandRequest(commandExecutorInput);
+	}
+
+
+	private Map<String,Object> setTTLInFlags( String value){
+		Map<String,Object> flags = new HashMap<String,Object>();
+		if( value != null || !("".equalsIgnoreCase(value))){
+			flags.put(TTL_FLAG, value);
+		}
+		return flags;
+	}
+
+
+	private LCMReadOnlyCommandRequest getConfigCommandRequest(String vnfType , Integer ttl , Date timeStamp, String requestId,
+															  Map<String,Object> flags, VNFOperation command , String vnfId, String vnfVersion ){
+
+		CommandExecutorInput commandExecutorInput = pouplateCommandExecutorInput(vnfType, ttl, vnfVersion, timeStamp, API_VERSION, requestId, ORIGINATOR_ID, "", command, vnfId, "");
+		LCMReadOnlyCommandRequest request = new LCMReadOnlyCommandRequest(commandExecutorInput);
+
+		return request;
+	}
+
+	private LCMCommandRequest getLCMCommandRequest(String vnfType , Integer ttl ,Date timeStamp, String requestId,
+											 Map<String,Object> flags, VNFOperation command , String vnfId, String vnfVersion ){
+
+		CommandExecutorInput commandExecutorInput = pouplateCommandExecutorInput(vnfType, ttl, vnfVersion, timeStamp, API_VERSION, requestId, ORIGINATOR_ID, "", command, vnfId, "");
+		LCMCommandRequest request = new LCMCommandRequest(commandExecutorInput);
+
+		return request;
+	}
+
+	public WorkflowResponse getWorkflowResponse (){
+		WorkflowResponse wfResponse = new WorkflowResponse();
+		ResponseContext responseContext = createResponseContextWithSuObjects();
+		wfResponse.setResponseContext(responseContext);
+		responseContext.setPayload("");
+		wfResponse.getResponseContext().getStatus().setCode(100);
+		return wfResponse;
+	}
+
+	private CommandExecutorInput pouplateCommandExecutorInput(String vnfType, int ttl, String vnfVersion, Date timeStamp, String apiVersion, String requestId, String originatorID, String subRequestID, VNFOperation action, String vnfId , String payload){
+		CommandExecutorInput commandExecutorInput = createCommandExecutorInputWithSubObjects();
+		RuntimeContext runtimeContext = commandExecutorInput.getRuntimeContext();
+		RequestContext requestContext = runtimeContext.getRequestContext();
+		ResponseContext responseContext = createResponseContextWithSuObjects();
+		runtimeContext.setResponseContext(responseContext);
+
+		requestContext.getCommonHeader().getFlags().setTtl(ttl);
+		requestContext.getCommonHeader().setApiVer(apiVersion);
+		requestContext.getCommonHeader().setTimestamp(timeStamp);
+		requestContext.getCommonHeader().setRequestId(requestId);
+		requestContext.getCommonHeader().setSubRequestId(subRequestID);
+		requestContext.getCommonHeader().setOriginatorId(originatorID);
+		requestContext.setAction(action);
+		requestContext.setPayload(payload);
+		requestContext.getActionIdentifiers().setVnfId(vnfId);
+		VNFContext vnfContext = runtimeContext.getVnfContext();
+		vnfContext.setType(vnfType);
+		vnfContext.setId(vnfId);
+		vnfContext.setVersion(vnfVersion);
+		return commandExecutorInput;
+	}
+
+	private CommandExecutorInput createCommandExecutorInputWithSubObjects() {
+		CommandExecutorInput commandExecutorInput = new CommandExecutorInput();
+		RuntimeContext runtimeContext = createRuntimeContextWithSubObjects();
+		commandExecutorInput.setRuntimeContext(runtimeContext);
+		return commandExecutorInput;
+	}
+
+	private RuntimeContext createRuntimeContextWithSubObjects() {
+		RuntimeContext runtimeContext = new RuntimeContext();
+		RequestContext requestContext = new RequestContext();
+		runtimeContext.setRequestContext(requestContext);
+		CommonHeader commonHeader = new CommonHeader();
+		requestContext.setCommonHeader(commonHeader);
+		Flags flags = new Flags();
+		commonHeader.setFlags(flags);
+		ActionIdentifiers actionIdentifiers = new ActionIdentifiers();
+		requestContext.setActionIdentifiers(actionIdentifiers);
+		VNFContext vnfContext = new VNFContext();
+		runtimeContext.setVnfContext(vnfContext);
+		return runtimeContext;
+
+	}
+
+	private ResponseContext createResponseContextWithSuObjects(){
+		ResponseContext responseContext = new ResponseContext();
+		CommonHeader commonHeader = new CommonHeader();
+		Flags flags = new Flags();
+		Status status = new Status();
+		responseContext.setCommonHeader(commonHeader);
+		responseContext.setStatus(status);
+		commonHeader.setFlags(flags);
+		return responseContext;
+	}
+
+}
+