bpmn/so-bpmn-tasks/src/test/java/org/onap/so/client/orchestration/AAINetworkResourcesTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.so.client.orchestration;
  22
  23 import static com.shazam.shazamcrest.MatcherAssert.assertThat;
  24 import static com.shazam.shazamcrest.matcher.Matchers.sameBeanAs;
  25 import static org.junit.Assert.assertEquals;
  26 import static org.junit.Assert.assertNotNull;
  27 import static org.junit.Assert.assertTrue;
  28 import static org.mockito.Matchers.any;
  29 import static org.mockito.Matchers.eq;
  30 import static org.mockito.Matchers.isA;
  31 import static org.mockito.Mockito.doNothing;
  32 import static org.mockito.Mockito.doReturn;
  33 import static org.mockito.Mockito.times;
  34 import static org.mockito.Mockito.verify;
  35
  36 import java.nio.file.Files;
  37 import java.nio.file.Paths;
  38 import java.util.ArrayList;
  39 import java.util.List;
  40 import java.util.Optional;
  41
  42 import org.junit.Before;
  43 import org.junit.Test;
  44 import org.junit.runner.RunWith;
  45 import org.mockito.InjectMocks;
  46 import org.mockito.Mock;
  47 import org.mockito.runners.MockitoJUnitRunner;
  48 import org.onap.aai.domain.yang.NetworkPolicy;
  49 import org.onap.aai.domain.yang.RouteTableReference;
  50 import org.onap.aai.domain.yang.VpnBinding;
  51 import org.onap.so.bpmn.common.InjectionHelper;
  52 import org.onap.so.bpmn.common.data.TestDataSetup;
  53 import org.onap.so.bpmn.servicedecomposition.bbobjects.CloudRegion;
  54 import org.onap.so.bpmn.servicedecomposition.bbobjects.Collection;
  55 import org.onap.so.bpmn.servicedecomposition.bbobjects.InstanceGroup;
  56 import org.onap.so.bpmn.servicedecomposition.bbobjects.L3Network;
  57 import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
  58 import org.onap.so.client.aai.AAIObjectType;
  59 import org.onap.so.client.aai.AAIResourcesClient;
  60 import org.onap.so.client.aai.entities.AAIResultWrapper;
  61 import org.onap.so.client.aai.entities.Relationships;
  62 import org.onap.so.client.aai.entities.uri.AAIResourceUri;
  63 import org.onap.so.client.aai.entities.uri.AAIUriFactory;
  64 import org.onap.so.client.aai.mapper.AAIObjectMapper;
  65 import org.onap.so.db.catalog.beans.OrchestrationStatus;
  66
  67 @RunWith(MockitoJUnitRunner.class)
  68 public class AAINetworkResourcesTest extends TestDataSetup{
  69
  70         private final static String JSON_FILE_LOCATION = "src/test/resources/__files/BuildingBlocks/";
  71
  72         private L3Network network;
  73         private Collection collection;
  74         private InstanceGroup instanceGroup;
  75         private ServiceInstance serviceInstance;
  76         private CloudRegion cloudRegion;
  77
  78         @Mock
  79         protected AAIResourcesClient MOCK_aaiResourcesClient;
  80
  81     @Mock
  82     protected AAIObjectMapper MOCK_aaiObjectMapper;
  83
  84     @Mock
  85     protected InjectionHelper MOCK_injectionHelper;
  86
  87         @InjectMocks
  88         private AAINetworkResources aaiNetworkResources = new AAINetworkResources();
  89
  90         @Before
  91         public void before() {
  92                 network = buildL3Network();
  93
  94                 collection = buildCollection();
  95
  96                 List<L3Network> l3NetworkList = new ArrayList<L3Network>();
  97                 l3NetworkList.add(network);
  98
  99                 instanceGroup = buildInstanceGroup();
 100
 101                 serviceInstance = buildServiceInstance();
 102
 103                 cloudRegion = buildCloudRegion();
 104                 doReturn(MOCK_aaiResourcesClient).when(MOCK_injectionHelper).getAaiClient();
 105         }
 106
 107
 108         @Test
 109         public void updateNetworkTest() throws Exception {
 110
 111                 network.setOrchestrationStatus(OrchestrationStatus.ASSIGNED);
 112
 113                 doReturn(new org.onap.aai.domain.yang.L3Network()).when(MOCK_aaiObjectMapper).mapNetwork(network);
 114                 doNothing().when(MOCK_aaiResourcesClient).update(isA(AAIResourceUri.class), isA(org.onap.aai.domain.yang.L3Network.class));
 115
 116                 aaiNetworkResources.updateNetwork(network);
 117
 118                 verify(MOCK_aaiResourcesClient, times(1)).update(any(AAIResourceUri.class), isA(org.onap.aai.domain.yang.L3Network.class));
 119                 assertEquals(OrchestrationStatus.ASSIGNED, network.getOrchestrationStatus());
 120         }
 121
 122         @Test
 123         public void createNetworkConnectToServiceInstanceTest() throws Exception {
 124
 125                 network.setOrchestrationStatus(OrchestrationStatus.PRECREATED);
 126
 127                 doReturn(new org.onap.aai.domain.yang.L3Network()).when(MOCK_aaiObjectMapper).mapNetwork(network);
 128                 doReturn(MOCK_aaiResourcesClient).when(MOCK_aaiResourcesClient).createIfNotExists(any(AAIResourceUri.class), any(Optional.class));
 129                 doNothing().when(MOCK_aaiResourcesClient).connect(isA(AAIResourceUri.class), any(AAIResourceUri.class));
 130
 131                 aaiNetworkResources.createNetworkConnectToServiceInstance(network, serviceInstance);
 132
 133                 assertEquals(OrchestrationStatus.INVENTORIED, network.getOrchestrationStatus());
 134
 135         verify(MOCK_aaiResourcesClient, times(1)).createIfNotExists(any(AAIResourceUri.class), any(Optional.class));
 136         verify(MOCK_aaiResourcesClient, times(1)).connect(any(AAIResourceUri.class), any(AAIResourceUri.class));
 137         }
 138
 139         @Test
 140         public void deleteNetworkTest() throws Exception {
 141
 142                 network.setOrchestrationStatus(OrchestrationStatus.INVENTORIED);
 143
 144                 doNothing().when(MOCK_aaiResourcesClient).delete(isA(AAIResourceUri.class));
 145
 146                 aaiNetworkResources.deleteNetwork(network);
 147
 148                 verify(MOCK_aaiResourcesClient, times(1)).delete(any(AAIResourceUri.class));
 149         }
 150
 151         @Test
 152         public void getVpnBindingTest() throws Exception {
 153                 final String content = new String(Files.readAllBytes(Paths.get(JSON_FILE_LOCATION + "queryAaiVpnBinding.json")));
 154                 AAIResultWrapper aaiResultWrapper = new AAIResultWrapper(content);
 155                 Optional<VpnBinding> oVpnBinding = Optional.empty();
 156                 AAIResourceUri aaiUri = AAIUriFactory.createResourceUri(AAIObjectType.VPN_BINDING, "ModelInvariantUUID", "serviceModelVersionId");
 157
 158                 doReturn(aaiResultWrapper).when(MOCK_aaiResourcesClient).get(isA(AAIResourceUri.class));
 159                 oVpnBinding = aaiNetworkResources.getVpnBinding(aaiUri);
 160                 verify(MOCK_aaiResourcesClient, times(1)).get(any(AAIResourceUri.class));
 161
 162                 if (oVpnBinding.isPresent()) {
 163                         VpnBinding vpnBinding = oVpnBinding.get();
 164                         assertThat(aaiResultWrapper.asBean(VpnBinding.class).get(), sameBeanAs(vpnBinding));
 165                 }
 166         }
 167
 168         @Test
 169         public void getNetworkPolicyTest() throws Exception {
 170                 final String content = new String(Files.readAllBytes(Paths.get(JSON_FILE_LOCATION + "queryAaiNetworkPolicy.json")));
 171                 AAIResultWrapper aaiResultWrapper = new AAIResultWrapper(content);
 172                 Optional<NetworkPolicy> oNetPolicy = Optional.empty();
 173                 AAIResourceUri netPolicyUri = AAIUriFactory.createResourceUri(AAIObjectType.NETWORK_POLICY, "ModelInvariantUUID", "serviceModelVersionId");
 174
 175                 doReturn(aaiResultWrapper).when(MOCK_aaiResourcesClient).get(isA(AAIResourceUri.class));
 176                 oNetPolicy = aaiNetworkResources.getNetworkPolicy(netPolicyUri);
 177                 verify(MOCK_aaiResourcesClient, times(1)).get(any(AAIResourceUri.class));
 178
 179                 if (oNetPolicy.isPresent()) {
 180                         NetworkPolicy networkPolicy = oNetPolicy.get();
 181                         assertThat(aaiResultWrapper.asBean(NetworkPolicy.class).get(), sameBeanAs(networkPolicy));
 182                 }
 183         }
 184
 185         @Test
 186         public void getRouteTableTest() throws Exception {
 187                 final String content = new String(Files.readAllBytes(Paths.get(JSON_FILE_LOCATION + "queryAaiNetworkTableRefs.json")));
 188                 AAIResultWrapper aaiResultWrapper = new AAIResultWrapper(content);
 189                 Optional<RouteTableReference> oRtref = Optional.empty();
 190                 AAIResourceUri rTRefUri = AAIUriFactory.createResourceUri(AAIObjectType.ROUTE_TABLE_REFERENCE, "ModelInvariantUUID", "serviceModelVersionId");
 191
 192                 doReturn(aaiResultWrapper).when(MOCK_aaiResourcesClient).get(isA(AAIResourceUri.class));
 193                 oRtref = aaiNetworkResources.getRouteTable(rTRefUri);
 194                 verify(MOCK_aaiResourcesClient, times(1)).get(any(AAIResourceUri.class));
 195
 196                 if (oRtref.isPresent()) {
 197                         RouteTableReference rTref = oRtref.get();
 198                         assertThat(aaiResultWrapper.asBean(RouteTableReference.class).get(), sameBeanAs(rTref));
 199                 }
 200         }
 201
 202         @Test
 203         public void queryNetworkByIdTest() throws Exception {
 204                 final String content = new String(Files.readAllBytes(Paths.get(JSON_FILE_LOCATION + "aaiQueryAAIResponse-Wrapper.json")));
 205                 AAIResultWrapper aaiResultWrapper = new AAIResultWrapper(content);
 206                 L3Network network = new L3Network();
 207                 network.setNetworkId("0384d743-f69b-4cc8-9aa8-c3ae66662c44");
 208                 network.setNetworkName("Dev_Bindings_1802_020118");
 209                 network.setOrchestrationStatus(OrchestrationStatus.CREATED);
 210
 211                 doReturn(aaiResultWrapper).when(MOCK_aaiResourcesClient).get(isA(AAIResourceUri.class));
 212                 Optional<org.onap.aai.domain.yang.L3Network> l3NetworkOpt =  aaiNetworkResources.queryNetworkById(network);
 213                 org.onap.aai.domain.yang.L3Network l3Network = l3NetworkOpt.isPresent() ? l3NetworkOpt.get() : null;
 214
 215                 verify(MOCK_aaiResourcesClient, times(1)).get(isA(AAIResourceUri.class));
 216                 assertNotNull(l3Network);
 217                 assertEquals("0384d743-f69b-4cc8-9aa8-c3ae66662c44", l3Network.getNetworkId());
 218                 assertEquals("Dev_Bindings_1802_020118", l3Network.getNetworkName());
 219         }
 220
 221         @Test
 222         public void queryNetworkWrapperByIdTest() throws Exception {
 223                 final String content = new String(Files.readAllBytes(Paths.get(JSON_FILE_LOCATION + "aaiQueryAAIResponse-Wrapper.json")));
 224                 AAIResultWrapper aaiResultWrapper = new AAIResultWrapper(content);
 225                 L3Network network = new L3Network();
 226                 network.setNetworkId("0384d743-f69b-4cc8-9aa8-c3ae66662c44");
 227                 network.setNetworkName("Dev_Bindings_1802_020118");
 228                 network.setOrchestrationStatus(OrchestrationStatus.CREATED);
 229
 230                 doReturn(aaiResultWrapper).when(MOCK_aaiResourcesClient).get(isA(AAIResourceUri.class));
 231                 AAIResultWrapper result =  aaiNetworkResources.queryNetworkWrapperById(network);
 232
 233                 verify(MOCK_aaiResourcesClient, times(1)).get(isA(AAIResourceUri.class));
 234                 assertEquals(aaiResultWrapper.getJson(), result.getJson());
 235                 assertNotNull(result);
 236                 Optional<Relationships> resultNetworkRelationships = result.getRelationships();
 237                 assertTrue(resultNetworkRelationships.isPresent());
 238                 Optional<org.onap.aai.domain.yang.L3Network> aaiL3Network = result.asBean(org.onap.aai.domain.yang.L3Network.class);
 239                 assertEquals(network.getNetworkId(),aaiL3Network.get().getNetworkId());
 240                 assertEquals(network.getNetworkName(),aaiL3Network.get().getNetworkName());
 241
 242
 243         }
 244
 245         @Test
 246         public void createNetworkCollectionTest() throws Exception {
 247
 248                 doNothing().when(MOCK_aaiResourcesClient).create(isA(AAIResourceUri.class), isA(org.onap.aai.domain.yang.Collection.class));
 249                 doReturn(new org.onap.aai.domain.yang.Collection()).when(MOCK_aaiObjectMapper).mapCollection(collection);
 250                 collection.setOrchestrationStatus(OrchestrationStatus.PRECREATED);
 251
 252                 aaiNetworkResources.createNetworkCollection(collection);
 253                 assertEquals(OrchestrationStatus.INVENTORIED, collection.getOrchestrationStatus());
 254                 verify(MOCK_aaiResourcesClient, times(1)).create(any(AAIResourceUri.class), isA(org.onap.aai.domain.yang.Collection.class));
 255         }
 256
 257         @Test
 258         public void createNetworkInstanceGroupTest() throws Exception {
 259                 doReturn(new org.onap.aai.domain.yang.InstanceGroup()).when(MOCK_aaiObjectMapper).mapInstanceGroup(instanceGroup);
 260                 doNothing().when(MOCK_aaiResourcesClient).create(isA(AAIResourceUri.class), isA(org.onap.aai.domain.yang.InstanceGroup.class));
 261                 aaiNetworkResources.createNetworkInstanceGroup(instanceGroup);
 262                 verify(MOCK_aaiResourcesClient, times(1)).create(any(AAIResourceUri.class), isA(org.onap.aai.domain.yang.InstanceGroup.class));
 263         }
 264
 265         @Test
 266         public void connectNetworkToNetworkCollectionInstanceGroupTest() throws Exception {
 267                 aaiNetworkResources.connectNetworkToNetworkCollectionInstanceGroup(network, instanceGroup);
 268                 verify(MOCK_aaiResourcesClient, times(1)).connect(eq(AAIUriFactory.createResourceUri(AAIObjectType.INSTANCE_GROUP, instanceGroup.getId())), eq(AAIUriFactory.createResourceUri(AAIObjectType.L3_NETWORK, network.getNetworkId())));
 269         }
 270
 271         @Test
 272         public void connectNetworkToNetworkCollectionServiceInstanceTest() throws Exception {
 273                 aaiNetworkResources.connectNetworkToNetworkCollectionServiceInstance(network, serviceInstance);
 274                 verify(MOCK_aaiResourcesClient, times(1)).connect(any(AAIResourceUri.class), eq(AAIUriFactory.createResourceUri(AAIObjectType.L3_NETWORK, network.getNetworkId())));
 275         }
 276
 277         @Test
 278         public void connectNetworkToCloudRegionTest() throws Exception {
 279                 aaiNetworkResources.connectNetworkToCloudRegion(network, cloudRegion);
 280                 verify(MOCK_aaiResourcesClient, times(1)).connect(eq(AAIUriFactory.createResourceUri(AAIObjectType.CLOUD_REGION,
 281                                 cloudRegion.getCloudOwner(), cloudRegion.getLcpCloudRegionId())), eq(AAIUriFactory.createResourceUri(AAIObjectType.L3_NETWORK, network.getNetworkId())));
 282         }
 283
 284         @Test
 285         public void connectNetworkToTenantTest() throws Exception {
 286                 aaiNetworkResources.connectNetworkToTenant(network, cloudRegion);
 287                 verify(MOCK_aaiResourcesClient, times(1)).connect(eq(AAIUriFactory.createResourceUri(AAIObjectType.TENANT,
 288                                 cloudRegion.getCloudOwner(), cloudRegion.getLcpCloudRegionId(), cloudRegion.getTenantId())),
 289                                 eq(AAIUriFactory.createResourceUri(AAIObjectType.L3_NETWORK, network.getNetworkId())));
 290         }
 291
 292         @Test
 293         public void connectNetworkCollectionInstanceGroupToNetworkCollectionTest() throws Exception {
 294                 aaiNetworkResources.connectNetworkCollectionInstanceGroupToNetworkCollection(instanceGroup, collection);
 295                 verify(MOCK_aaiResourcesClient, times(1)).connect(eq(AAIUriFactory.createResourceUri(AAIObjectType.COLLECTION, collection.getId())), eq(AAIUriFactory.createResourceUri(AAIObjectType.INSTANCE_GROUP, instanceGroup.getId())));
 296         }
 297
 298         @Test
 299         public void connectNetworkCollectionToServiceInstanceTest() throws Exception {
 300                 aaiNetworkResources.connectNetworkCollectionToServiceInstance(collection, serviceInstance);
 301                 verify(MOCK_aaiResourcesClient, times(1)).connect(any(AAIResourceUri.class), any(AAIResourceUri.class));
 302         }
 303
 304         @Test
 305         public void deleteCollectionTest() throws Exception {
 306                 doNothing().when(MOCK_aaiResourcesClient).delete(isA(AAIResourceUri.class));
 307                 aaiNetworkResources.deleteCollection(collection);
 308                 verify(MOCK_aaiResourcesClient, times(1)).delete(any(AAIResourceUri.class));
 309         }
 310
 311         @Test
 312         public void deleteInstanceGroupTest() throws Exception {
 313                 doNothing().when(MOCK_aaiResourcesClient).delete(isA(AAIResourceUri.class));
 314                 aaiNetworkResources.deleteNetworkInstanceGroup(instanceGroup);
 315                 verify(MOCK_aaiResourcesClient, times(1)).delete(any(AAIResourceUri.class));
 316         }
 317 }