Add VnfInPlaceSoftwareUpdate for WFD

Added  CheckPserversLocked,SetClosedLoopDisabled,UnsetClosedLoopDisabled, CheckClosedLoopDisabled, CheckInMain

Change-Id: I6405b2c88d2109d952d452648bef2f99be6993df
Issue-ID: SO-1518
Signed-off-by: Bonkur, Venkat (vb8416) <vb8416@att.com>

2 files changed, 118 insertions, 0 deletions

diff --git a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/common/aai/tasks/AAIFlagTasksTest.java b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/common/aai/tasks/AAIFlagTasksTest.java
index cf28c114e8..cad97cad6e 100644
--- a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/common/aai/tasks/AAIFlagTasksTest.java
+++ b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/bpmn/common/aai/tasks/AAIFlagTasksTest.java
@@ -29,6 +29,8 @@ import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+import java.io.IOException;
+
 import org.camunda.bpm.engine.delegate.BpmnError;
 import org.junit.Before;
 import org.junit.Test;
@@ -119,4 +121,89 @@ public class AAIFlagTasksTest extends BaseTaskTest {
 			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
 		}
 	}
+	
+	@Test
+	public void checkVnfClosedLoopDisabledTestTrue() throws Exception {
+		doThrow(new BpmnError("VNF Closed Loop Disabled in A&AI")).when(exceptionUtil).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		doReturn(false).when(aaiVnfResources).checkVnfClosedLoopDisabledFlag(isA(String.class));
+		try {
+			aaiFlagTasks.checkVnfClosedLoopDisabledFlag(execution);
+		} catch (Exception e) {
+			verify(aaiVnfResources, times(1)).checkVnfClosedLoopDisabledFlag(any(String.class));
+			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), eq("VNF Closed Loop Disabled in A&AI"));		
+		}
+	}
+
+	@Test
+	public void checkVnfClosedLoopDisabledTestFalse() throws Exception {
+		doReturn(false).when(aaiVnfResources).checkVnfClosedLoopDisabledFlag(isA(String.class));
+		aaiFlagTasks.checkVnfClosedLoopDisabledFlag(execution);
+		verify(exceptionUtil, times(0)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), any(int.class), any(String.class));    
+	}
+
+	@Test
+	public void checkVnfClosedLoopDisabledFlagExceptionTest() {
+
+		doThrow(new BpmnError("Unknown Error")).when(exceptionUtil).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		doThrow(RuntimeException.class).when(aaiVnfResources).checkVnfClosedLoopDisabledFlag(isA(String.class));
+		try {
+			aaiFlagTasks.checkVnfClosedLoopDisabledFlag(execution);
+		} catch (Exception e) {
+			verify(aaiVnfResources, times(1)).checkVnfClosedLoopDisabledFlag(any(String.class));
+			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		}	
+	}
+
+	@Test
+	public void modifyVnfClosedLoopDisabledFlagTest() throws Exception {
+		doNothing().when(aaiVnfResources).updateObjectVnf(ArgumentMatchers.any(GenericVnf.class));
+		aaiFlagTasks.modifyVnfClosedLoopDisabledFlag(execution, true);
+		verify(aaiVnfResources, times(1)).updateObjectVnf(ArgumentMatchers.any(GenericVnf.class));
+	}
+
+	@Test
+	public void modifyVnfClosedLoopDisabledFlagExceptionTest() {
+		
+		doThrow(new BpmnError("Unknown Error")).when(exceptionUtil).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		doThrow(RuntimeException.class).when(aaiVnfResources).updateObjectVnf(isA(GenericVnf.class));
+		try {
+			aaiFlagTasks.modifyVnfClosedLoopDisabledFlag(execution, true);
+		} catch (Exception e) {
+			verify(aaiVnfResources, times(1)).checkVnfClosedLoopDisabledFlag(any(String.class));
+			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		}
+	}
+	
+
+	@Test
+	public void checkVnfPserversLockedFlagTestTrue() throws Exception {
+		doThrow(new BpmnError("VNF PServers in Locked in A&AI")).when(exceptionUtil).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		doReturn(true).when(aaiVnfResources).checkVnfPserversLockedFlag(isA(String.class));
+		try {
+			aaiFlagTasks.checkVnfClosedLoopDisabledFlag(execution);
+		} catch (Exception e) {
+			verify(aaiVnfResources, times(1)).checkVnfPserversLockedFlag(any(String.class));
+			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), eq("VNF PServers in Locked in A&AI"));		
+		}
+	}
+
+	@Test
+	public void checkVnfPserversLockedFlagTestFalse() throws Exception {
+		doReturn(false).when(aaiVnfResources).checkVnfPserversLockedFlag(isA(String.class));
+		aaiFlagTasks.checkVnfPserversLockedFlag(execution);
+		verify(exceptionUtil, times(0)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), any(int.class), any(String.class));    
+	}
+
+	@Test
+	public void checkVnfPserversLockedFlagExceptionTest() throws IOException {
+
+		doThrow(new BpmnError("Unknown Error")).when(exceptionUtil).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		doThrow(RuntimeException.class).when(aaiVnfResources).checkVnfClosedLoopDisabledFlag(isA(String.class));
+		try {
+			aaiFlagTasks.checkVnfPserversLockedFlag(execution);
+		} catch (Exception e) {
+			verify(aaiVnfResources, times(1)).checkVnfPserversLockedFlag(any(String.class));
+			verify(exceptionUtil, times(1)).buildAndThrowWorkflowException(any(BuildingBlockExecution.class), eq(7000), any(String.class));
+		}	
+	}
 }
diff --git a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/client/orchestration/AAIVnfResourcesTest.java b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/client/orchestration/AAIVnfResourcesTest.java
index b87b5e4166..d499bcd36e 100644
--- a/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/client/orchestration/AAIVnfResourcesTest.java
+++ b/bpmn/so-bpmn-tasks/src/test/java/org/onap/so/client/orchestration/AAIVnfResourcesTest.java
@@ -21,6 +21,7 @@
 package org.onap.so.client.orchestration;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.ArgumentMatchers.isA;
@@ -29,6 +30,7 @@ import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 
+import java.io.IOException;
 import java.util.Optional;
 
 import org.junit.Before;
@@ -47,6 +49,7 @@ import org.onap.so.bpmn.servicedecomposition.bbobjects.Platform;
 import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
 import org.onap.so.client.aai.AAIObjectType;
 import org.onap.so.client.aai.AAIResourcesClient;
+import org.onap.so.client.aai.AAIValidatorImpl;
 import org.onap.so.client.aai.entities.uri.AAIResourceUri;
 import org.onap.so.client.aai.entities.uri.AAIUriFactory;
 import org.onap.so.client.aai.mapper.AAIObjectMapper;
@@ -70,6 +73,9 @@ public class AAIVnfResourcesTest extends TestDataSetup {
 	@Mock
 	protected InjectionHelper MOCK_injectionHelper;
 	
+	@Mock
+	protected AAIValidatorImpl MOCK_aaiValidatorImpl;
+	
 	@InjectMocks
 	AAIVnfResources aaiVnfResources = new AAIVnfResources();
 
@@ -182,4 +188,29 @@ public class AAIVnfResourcesTest extends TestDataSetup {
 				eq(AAIUriFactory.createResourceUri(AAIObjectType.CLOUD_REGION, 
 						cloudRegion.getCloudOwner(), cloudRegion.getLcpCloudRegionId())));
 	}
+	
+
+	@Test
+	public void checkVnfClosedLoopDisabledFlagTest () {
+		Optional<org.onap.aai.domain.yang.GenericVnf> vnf = Optional.of(new org.onap.aai.domain.yang.GenericVnf());
+		vnf.get().setVnfId("vnfId");
+		vnf.get().setIsClosedLoopDisabled(true);
+		doReturn(vnf).when(MOCK_aaiResourcesClient).get(eq(org.onap.aai.domain.yang.GenericVnf.class),isA(AAIResourceUri.class));
+		boolean isCheckVnfClosedLoopDisabledFlag = aaiVnfResources.checkVnfClosedLoopDisabledFlag("vnfId");
+		verify(MOCK_aaiResourcesClient, times(1)).get(eq(org.onap.aai.domain.yang.GenericVnf.class),isA(AAIResourceUri.class));
+		assertEquals(isCheckVnfClosedLoopDisabledFlag, true);
+	}
+	
+	@Test
+	public void checkVnfPserversLockedFlagTest () throws IOException {
+		
+		 Optional<org.onap.aai.domain.yang.GenericVnf> vnf = Optional.of(new org.onap.aai.domain.yang.GenericVnf());
+         vnf.get().setVnfId("vnfId");      
+         doReturn(vnf).when(MOCK_aaiResourcesClient).get(eq(org.onap.aai.domain.yang.GenericVnf.class),isA(AAIResourceUri.class));
+         doReturn(true).when(MOCK_aaiValidatorImpl).isPhysicalServerLocked("vnfId");       
+         boolean isVnfPserversLockedFlag = aaiVnfResources.checkVnfPserversLockedFlag("vnfId");
+         verify(MOCK_aaiResourcesClient, times(1)).get(eq(org.onap.aai.domain.yang.GenericVnf.class),isA(AAIResourceUri.class));
+         verify(MOCK_aaiValidatorImpl, times(1)).isPhysicalServerLocked(isA(String.class));        
+         assertTrue(isVnfPserversLockedFlag);
+	}
 }