creating relationships vnf and tenant

Change-Id: I515fc40c7924c2fdb3f0bc4dd3a620eb0a3bf948
Issue-ID: SO-1284
Signed-off-by: Kalkere Ramesh, Sharan (sk720x) <sk720x@att.com>

4 files changed, 250 insertions, 3 deletions

diff --git a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
index 7edf39df14..c0f1c48b76 100644
--- a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
+++ b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
@@ -90,6 +90,7 @@ import org.onap.so.serviceinstancebeans.Vnfs;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
+import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.SerializationFeature;
 
@@ -112,6 +113,9 @@ public class BBInputSetup implements JavaDelegate {
 
 	@Autowired
 	private BBInputSetupMapperLayer mapperLayer;
+	
+	@Autowired
+	private CloudInfoFromAAI bbInputSetupHelper;
 
 	@Autowired
 	private ExceptionBuilder exceptionUtil;
@@ -981,11 +985,18 @@ public class BBInputSetup implements JavaDelegate {
 			Map<ResourceKey, String> lookupKeyMap, String bbName, GeneralBuildingBlock gBB, Service service,
 			String requestAction, CloudConfiguration cloudConfiguration) throws Exception {
 		ServiceInstance serviceInstance = gBB.getServiceInstance();
+		CloudRegion cloudRegion = null;
+		if(cloudConfiguration == null) {
+			Optional<CloudRegion> cloudRegionOp = bbInputSetupHelper.getCloudInfoFromAAI(serviceInstance);
+			if(cloudRegionOp.isPresent()) {
+				cloudRegion = cloudRegionOp.get();
+			}
+		}
 		if (cloudConfiguration != null && requestAction.equalsIgnoreCase("deleteInstance")) {
 			org.onap.aai.domain.yang.CloudRegion aaiCloudRegion = bbInputSetupUtils.getCloudRegion(cloudConfiguration);
-			CloudRegion cloudRegion = mapperLayer.mapCloudRegion(cloudConfiguration, aaiCloudRegion);
-			gBB.setCloudRegion(cloudRegion);
+			cloudRegion = mapperLayer.mapCloudRegion(cloudConfiguration, aaiCloudRegion);
 		}
+		gBB.setCloudRegion(cloudRegion);
 		if (bbName.contains(VNF)) {
 			for (GenericVnf genericVnf : serviceInstance.getVnfs()) {
 				if (lookupKeyMap.get(ResourceKey.GENERIC_VNF_ID) != null
diff --git a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAI.java b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAI.java
new file mode 100644
index 0000000000..37088022bc
--- /dev/null
+++ b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAI.java
@@ -0,0 +1,93 @@
+package org.onap.so.bpmn.servicedecomposition.tasks;
+
+import java.util.List;
+import java.util.Optional;
+
+import org.onap.so.bpmn.servicedecomposition.bbobjects.CloudRegion;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.GenericVnf;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.L3Network;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
+import org.onap.so.client.aai.AAICommonObjectMapperProvider;
+import org.onap.so.client.aai.AAIObjectType;
+import org.onap.so.client.aai.entities.AAIResultWrapper;
+import org.onap.so.client.aai.entities.Relationships;
+import org.onap.so.logger.MsoLogger;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+
+@Component
+public class CloudInfoFromAAI {
+
+	private static final MsoLogger msoLogger = MsoLogger.getMsoLogger(MsoLogger.Catalog.BPEL, CloudInfoFromAAI.class);
+	@Autowired
+	private BBInputSetupUtils bbInputSetupUtils;
+	
+	public void setBbInputSetupUtils(BBInputSetupUtils bbInputSetupUtils) {
+		this.bbInputSetupUtils = bbInputSetupUtils;
+	}
+	
+	protected Optional<CloudRegion> getCloudInfoFromAAI(ServiceInstance serviceInstance) throws JsonProcessingException {
+		Optional<Relationships> relationshipsOp = Optional.empty();
+		if(!serviceInstance.getVnfs().isEmpty()) {
+			GenericVnf vnf = serviceInstance.getVnfs().get(0);
+			org.onap.aai.domain.yang.GenericVnf aaiVnf = bbInputSetupUtils.getAAIGenericVnf(vnf.getVnfId());
+			AAIResultWrapper vnfWrapper = new AAIResultWrapper(
+					new AAICommonObjectMapperProvider().getMapper().writeValueAsString(aaiVnf));
+			relationshipsOp = getRelationshipsFromWrapper(vnfWrapper);
+		} else if(!serviceInstance.getNetworks().isEmpty()) {
+			L3Network network = serviceInstance.getNetworks().get(0);
+			org.onap.aai.domain.yang.L3Network aaiL3Network = bbInputSetupUtils.getAAIL3Network(network.getNetworkId());
+			AAIResultWrapper networkWrapper = new AAIResultWrapper(
+					new AAICommonObjectMapperProvider().getMapper().writeValueAsString(aaiL3Network));
+			relationshipsOp = getRelationshipsFromWrapper(networkWrapper);
+		} else {
+			msoLogger.debug("BBInputSetup could not find a cloud region or tenant, since there are no resources under the SI.");
+			return Optional.empty();
+		}
+		if (relationshipsOp.isPresent()) {
+			return getRelatedCloudRegionAndTenant(relationshipsOp.get());
+		} else {
+			msoLogger.debug("BBInputSetup could not find a cloud region or tenant");
+			return Optional.empty();
+		}
+	}
+
+	protected Optional<Relationships> getRelationshipsFromWrapper(AAIResultWrapper wrapper) {
+		Optional<Relationships> relationshipsOp;
+		relationshipsOp = wrapper.getRelationships();
+		if(relationshipsOp.isPresent()) {
+			return relationshipsOp;
+		}
+		return Optional.empty();
+	}
+
+	protected Optional<CloudRegion> getRelatedCloudRegionAndTenant(Relationships relationships) {
+		CloudRegion cloudRegion = new CloudRegion();
+		List<AAIResultWrapper> cloudRegions = relationships.getByType(AAIObjectType.CLOUD_REGION);
+		List<AAIResultWrapper> tenants = relationships.getByType(AAIObjectType.TENANT);
+		if(!cloudRegions.isEmpty()) {
+			AAIResultWrapper cloudRegionWrapper = cloudRegions.get(0);
+			Optional<org.onap.aai.domain.yang.CloudRegion> aaiCloudRegionOp = cloudRegionWrapper
+					.asBean(org.onap.aai.domain.yang.CloudRegion.class);
+			if(aaiCloudRegionOp.isPresent()) {
+				org.onap.aai.domain.yang.CloudRegion aaiCloudRegion = aaiCloudRegionOp.get();
+				cloudRegion.setCloudOwner(aaiCloudRegion.getCloudOwner());
+				cloudRegion.setCloudRegionVersion(aaiCloudRegion.getCloudRegionVersion());
+				cloudRegion.setLcpCloudRegionId(aaiCloudRegion.getCloudRegionId());
+				cloudRegion.setComplex(aaiCloudRegion.getComplexName());
+			}
+		}
+		if(!tenants.isEmpty()) {
+			AAIResultWrapper tenantWrapper = tenants.get(0);
+			Optional<org.onap.aai.domain.yang.Tenant> aaiTenantOp = tenantWrapper
+					.asBean(org.onap.aai.domain.yang.Tenant.class);
+			if(aaiTenantOp.isPresent()) {
+				org.onap.aai.domain.yang.Tenant aaiTenant = aaiTenantOp.get();
+				cloudRegion.setTenantId(aaiTenant.getTenantId());
+			}
+		}
+		return Optional.of(cloudRegion);
+	}
+}
diff --git a/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetupTest.java b/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetupTest.java
index b36d6d43f8..8399fe15a3 100644
--- a/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetupTest.java
+++ b/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetupTest.java
@@ -84,6 +84,7 @@ import org.onap.so.bpmn.servicedecomposition.modelinfo.ModelInfoVfModule;
 import org.onap.so.client.aai.AAICommonObjectMapperProvider;
 import org.onap.so.client.aai.AAIObjectType;
 import org.onap.so.client.aai.entities.AAIResultWrapper;
+import org.onap.so.client.aai.entities.Relationships;
 import org.onap.so.client.aai.entities.uri.AAIResourceUri;
 import org.onap.so.client.aai.entities.uri.AAIUriFactory;
 import org.onap.so.constants.Defaults;
@@ -137,7 +138,7 @@ public class BBInputSetupTest {
 		SPY_bbInputSetup.setBbInputSetupUtils(SPY_bbInputSetupUtils);
 		SPY_bbInputSetup.setMapperLayer(bbInputSetupMapperLayer);
 	}
-
+	
 	@Test
 	public void testGetAlaCarteServiceInstance() throws Exception {
 		ServiceInstance expected = mapper.readValue(
diff --git a/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAITest.java b/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAITest.java
new file mode 100644
index 0000000000..ccaa97b735
--- /dev/null
+++ b/bpmn/MSOCommonBPMN/src/test/java/org/onap/so/bpmn/servicedecomposition/tasks/CloudInfoFromAAITest.java
@@ -0,0 +1,142 @@
+package org.onap.so.bpmn.servicedecomposition.tasks;
+
+import static com.shazam.shazamcrest.MatcherAssert.assertThat;
+import static com.shazam.shazamcrest.matcher.Matchers.sameBeanAs;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.mockito.Matchers.isA;
+import static org.mockito.Mockito.doReturn;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Optional;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.Spy;
+import org.mockito.runners.MockitoJUnitRunner;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.CloudRegion;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.GenericVnf;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.L3Network;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
+import org.onap.so.client.aai.AAICommonObjectMapperProvider;
+import org.onap.so.client.aai.AAIObjectType;
+import org.onap.so.client.aai.entities.AAIResultWrapper;
+import org.onap.so.client.aai.entities.Relationships;
+
+import com.fasterxml.jackson.core.JsonParseException;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonMappingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+@RunWith(MockitoJUnitRunner.class)
+public class CloudInfoFromAAITest {
+
+	private static final String RESOURCE_PATH = "src/test/resources/__files/ExecuteBuildingBlock/";
+
+	@Spy
+	private CloudInfoFromAAI SPY_CloudInfoFromAAI = new CloudInfoFromAAI();
+
+	protected ObjectMapper mapper = new ObjectMapper();
+	
+	@Mock
+	private BBInputSetupUtils SPY_bbInputSetupUtils;
+
+	@Before
+	public void setup(){
+		SPY_CloudInfoFromAAI.setBbInputSetupUtils(SPY_bbInputSetupUtils);
+	}
+	
+	@Test
+	public void testGetCloudInfoFromAAI() throws JsonParseException, JsonMappingException, IOException {
+		//Test vnfs
+		ServiceInstance serviceInstance = mapper.readValue(
+				new File(RESOURCE_PATH + "ServiceInstance_getServiceInstanceNOAAIExpected.json"),
+				ServiceInstance.class);
+		CloudRegion expected = new CloudRegion();
+		GenericVnf vnf = new GenericVnf();
+		String vnfId = "vnfId";
+		vnf.setVnfId(vnfId);
+		serviceInstance.getVnfs().add(vnf);
+		org.onap.aai.domain.yang.GenericVnf aaiVnf = new org.onap.aai.domain.yang.GenericVnf();
+		aaiVnf.setVnfId(vnfId);
+		Relationships relationships = Mockito.mock(Relationships.class);
+		Optional<Relationships> relationshipsOp= Optional.of(relationships);
+		doReturn(aaiVnf).when(SPY_bbInputSetupUtils).getAAIGenericVnf(vnf.getVnfId());
+		doReturn(relationshipsOp).when(SPY_CloudInfoFromAAI).getRelationshipsFromWrapper(isA(AAIResultWrapper.class));
+		doReturn(Optional.of(expected)).when(SPY_CloudInfoFromAAI).getRelatedCloudRegionAndTenant(relationships);
+		Optional<CloudRegion> actual = SPY_CloudInfoFromAAI.getCloudInfoFromAAI(serviceInstance);
+		assertThat(actual.get(), sameBeanAs(expected));
+		
+		//Test networks
+		serviceInstance = mapper.readValue(
+				new File(RESOURCE_PATH + "ServiceInstance_getServiceInstanceNOAAIExpected.json"),
+				ServiceInstance.class);
+		L3Network l3Network = new L3Network();
+		String networkId = "networkId";
+		l3Network.setNetworkId(networkId);
+		serviceInstance.getNetworks().add(l3Network);
+		org.onap.aai.domain.yang.L3Network aaiL3Network = new org.onap.aai.domain.yang.L3Network();
+		aaiL3Network.setNetworkId(networkId);
+		doReturn(aaiL3Network).when(SPY_bbInputSetupUtils).getAAIL3Network(l3Network.getNetworkId());
+		actual = SPY_CloudInfoFromAAI.getCloudInfoFromAAI(serviceInstance);
+		assertThat(actual.get(), sameBeanAs(expected));
+		
+		//Test no relationships
+		
+		doReturn(Optional.empty()).when(SPY_CloudInfoFromAAI).getRelationshipsFromWrapper(isA(AAIResultWrapper.class));
+		actual = SPY_CloudInfoFromAAI.getCloudInfoFromAAI(serviceInstance);
+		assertEquals(actual, Optional.empty());
+		
+		//Test null
+		serviceInstance = mapper.readValue(
+				new File(RESOURCE_PATH + "ServiceInstance_getServiceInstanceNOAAIExpected.json"),
+				ServiceInstance.class);
+		actual = SPY_CloudInfoFromAAI.getCloudInfoFromAAI(serviceInstance);
+		assertEquals(actual, Optional.empty());
+	}
+	
+	@Test
+	public void testGetRelatedCloudRegionAndTenant() throws JsonProcessingException {
+		String cloudOwner = "cloudOwner";
+		String cloudRegionId = "cloudRegionId";
+		String cloudRegionVersion = "cloudRegionVersion";
+		String cloudRegionComplexName = "cloudRegionComplexName";
+		String tenantId = "tenantId";
+		CloudRegion expected = new CloudRegion();
+		expected.setCloudOwner(cloudOwner);
+		expected.setCloudRegionVersion(cloudRegionVersion);
+		expected.setComplex(cloudRegionComplexName);
+		expected.setLcpCloudRegionId(cloudRegionId);
+		expected.setTenantId(tenantId);
+		
+		Relationships relationships = Mockito.mock(Relationships.class);
+		List<AAIResultWrapper> cloudRegions = new ArrayList<>();
+		org.onap.aai.domain.yang.CloudRegion cloudRegion = new org.onap.aai.domain.yang.CloudRegion();
+		cloudRegion.setCloudOwner(cloudOwner);
+		cloudRegion.setCloudRegionId(cloudRegionId);
+		cloudRegion.setCloudRegionVersion(cloudRegionVersion);
+		cloudRegion.setComplexName(cloudRegionComplexName);
+		AAIResultWrapper cloudRegionWrapper = new AAIResultWrapper(
+				new AAICommonObjectMapperProvider().getMapper().writeValueAsString(cloudRegion));
+		cloudRegions.add(cloudRegionWrapper);
+		
+		doReturn(cloudRegions).when(relationships).getByType(AAIObjectType.CLOUD_REGION);
+		List<AAIResultWrapper> tenants = new ArrayList<>();
+		org.onap.aai.domain.yang.Tenant tenant = new org.onap.aai.domain.yang.Tenant();
+		tenant.setTenantId(tenantId);
+		AAIResultWrapper tenantWrapper = new AAIResultWrapper(
+				new AAICommonObjectMapperProvider().getMapper().writeValueAsString(tenant));
+		tenants.add(tenantWrapper);
+		doReturn(tenants).when(relationships).getByType(AAIObjectType.TENANT);
+		
+		Optional<CloudRegion> actual = SPY_CloudInfoFromAAI.getRelatedCloudRegionAndTenant(relationships);
+		
+		assertThat(actual.get(), sameBeanAs(expected));
+	}
+}