1 files changed, 337 insertions, 0 deletions
diff --git a/nokiav2/driver/src/test/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/onap/vfc/TestVfcGrantManager.java b/nokiav2/driver/src/test/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/onap/vfc/TestVfcGrantManager.java
new file mode 100644
index 00000000..7ca19d63
--- /dev/null
+++ b/nokiav2/driver/src/test/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/onap/vfc/TestVfcGrantManager.java
@@ -0,0 +1,337 @@
+/*
+ * Copyright 2016-2017, Nokia Corporation
+ *
+ * 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.
+ */
+package org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.vfc;
+
+import com.google.common.collect.Lists;
+import com.google.gson.Gson;
+import com.google.gson.JsonObject;
+import com.nokia.cbam.lcm.v32.model.*;
+import com.nokia.cbam.lcm.v32.model.VnfInfo;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.ArgumentCaptor;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CatalogManager;
+import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.LifecycleManager;
+import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.TestBase;
+import org.onap.vnfmdriver.ApiException;
+import org.onap.vnfmdriver.model.*;
+import org.onap.vnfmdriver.model.ScaleDirection;
+
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+import static java.nio.file.Files.readAllBytes;
+import static junit.framework.TestCase.*;
+import static org.mockito.Mockito.*;
+import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CbamRestApiProvider.NOKIA_LCM_API_VERSION;
+import static org.springframework.test.util.ReflectionTestUtils.setField;
+
+public class TestVfcGrantManager extends TestBase {
+
+    private ArgumentCaptor<GrantVNFRequest> grantRequest = ArgumentCaptor.forClass(GrantVNFRequest.class);
+    private GrantVNFResponseVim vim = new GrantVNFResponseVim();
+    private GrantVNFResponse grantResponse = new GrantVNFResponse();
+    @Mock
+    private CatalogManager cbamCatalogManager;
+    @InjectMocks
+    private VfcGrantManager vfcGrantManager;
+
+    @Before
+    public void initMocks() throws Exception {
+        setField(VfcGrantManager.class, "logger", logger);
+        when(nsLcmApi.grantvnf(grantRequest.capture())).thenReturn(grantResponse);
+        grantResponse.setVim(vim);
+    }
+
+    /**
+     * test grant request for instantiation
+     */
+    @Test
+    public void testGrantDuringInstantiation() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.instantiation.yaml").toURI())));
+        //when
+        vfcGrantManager.requestGrantForInstantiate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, "level1", cbamVnfdContent, JOB_ID);
+        //verify
+        assertEquals(1, grantRequest.getAllValues().size());
+        GrantVNFRequest request = grantRequest.getValue();
+        assertVduInGrant(request.getAddResource(), "vdu1", 1);
+        assertVduInGrant(request.getAddResource(), "vdu2", 2);
+        assertEquals(0, request.getRemoveResource().size());
+        assertBasicGrantAttributes(request, org.onap.vnfmdriver.model.OperationType.INSTANTIATE);
+    }
+
+    /**
+     * test failure logging & propagation during grant request for instantiation
+     */
+    @Test
+    public void testFailureDuringGrantPreparation() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.instantiation.yaml").toURI())));
+        //when
+        try {
+            vfcGrantManager.requestGrantForInstantiate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, "missingLevel", cbamVnfdContent, JOB_ID);
+            //verify
+            fail();
+        } catch (RuntimeException e) {
+            verify(logger).error(Mockito.eq("Unable to prepare grant request for instantiation"), Mockito.any(RuntimeException.class));
+            verifyNoMoreInteractions(nsLcmApi);
+        }
+    }
+
+    /**
+     * test grant request for instantiation
+     */
+    @Test
+    public void testFailureDuringGrantRequest() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.instantiation.yaml").toURI())));
+        ApiException expectedException = new ApiException("a");
+        when(nsLcmApi.grantvnf(Mockito.any())).thenThrow(expectedException);
+        //when
+        try {
+            vfcGrantManager.requestGrantForInstantiate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, "level1", cbamVnfdContent, JOB_ID);
+            //verify
+            fail();
+        } catch (RuntimeException e) {
+            verify(logger).error("Unable to request grant", expectedException);
+        }
+    }
+
+    /**
+     * No grant is requested for termination if the the VNF is not instantiated
+     */
+    @Test
+    public void testNoGrantIsRequestedIfNotInstantiated() {
+        VnfInfo vnf = new VnfInfo();
+        vnf.setId(VNF_ID);
+        vnf.setInstantiationState(InstantiationState.NOT_INSTANTIATED);
+        //when
+        vfcGrantManager.requestGrantForTerminate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, vnf, JOB_ID);
+        //verify
+        verifyNoMoreInteractions(nsLcmApi);
+    }
+
+    /**
+     * grant is requested for termination if the the VNF is instantiated
+     */
+    @Test
+    public void testGrantIsRequestedIfInstantiated() {
+        VnfInfo vnf = new VnfInfo();
+        vnf.setId(VNF_ID);
+        vnf.setInstantiationState(InstantiationState.INSTANTIATED);
+        InstantiatedVnfInfo instantiatedVnfInfo = new InstantiatedVnfInfo();
+        VnfcResourceInfo vnfc = new VnfcResourceInfo();
+        vnfc.setId("vnfcId1");
+        vnfc.setVduId("vdu1");
+        instantiatedVnfInfo.setVnfcResourceInfo(new ArrayList<>());
+        instantiatedVnfInfo.getVnfcResourceInfo().add(vnfc);
+        vnf.setInstantiatedVnfInfo(instantiatedVnfInfo);
+        VnfProperty prop = new VnfProperty();
+        prop.setName(LifecycleManager.ONAP_CSAR_ID);
+        prop.setValue(ONAP_CSAR_ID);
+        vnf.setVnfConfigurableProperties(new ArrayList<>());
+        vnf.getVnfConfigurableProperties().add(prop);
+        //when
+        vfcGrantManager.requestGrantForTerminate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, vnf, JOB_ID);
+        //verify
+        assertEquals(1, grantRequest.getAllValues().size());
+        GrantVNFRequest request = grantRequest.getValue();
+        assertVduInGrant(request.getRemoveResource(), "vdu1", 1);
+        assertVduInGrant(request.getRemoveResource(), "vdu2", 0);
+        assertEquals(0, request.getAddResource().size());
+        assertBasicGrantAttributes(request, org.onap.vnfmdriver.model.OperationType.TERMINAL);
+    }
+
+    /**
+     * test failure logging & propagation during grant request for instantiation
+     */
+    @Test
+    public void testFailureDuringTerminationGrantPreparation() throws Exception {
+        VnfInfo vnf = new VnfInfo();
+        vnf.setId(VNF_ID);
+        vnf.setInstantiatedVnfInfo(null);
+        vnf.setInstantiationState(InstantiationState.INSTANTIATED);
+        //when
+        try {
+            vfcGrantManager.requestGrantForTerminate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, vnf, JOB_ID);
+            //verify
+            fail();
+        } catch (RuntimeException e) {
+            verify(logger).error(Mockito.eq("Unable to prepare grant request for termination"), Mockito.any(RuntimeException.class));
+            verifyNoMoreInteractions(nsLcmApi);
+        }
+    }
+
+    /**
+     * failuire is to request grant is logged
+     */
+    @Test
+    public void testFailureToRequestGrantIsLogged() throws Exception {
+        VnfInfo vnf = new VnfInfo();
+        vnf.setId(VNF_ID);
+        vnf.setInstantiationState(InstantiationState.INSTANTIATED);
+        InstantiatedVnfInfo instantiatedVnfInfo = new InstantiatedVnfInfo();
+        VnfcResourceInfo vnfc = new VnfcResourceInfo();
+        vnfc.setId("vnfcId1");
+        vnfc.setVduId("vdu1");
+        instantiatedVnfInfo.setVnfcResourceInfo(new ArrayList<>());
+        instantiatedVnfInfo.getVnfcResourceInfo().add(vnfc);
+        vnf.setInstantiatedVnfInfo(instantiatedVnfInfo);
+        VnfProperty prop = new VnfProperty();
+        prop.setName(LifecycleManager.ONAP_CSAR_ID);
+        prop.setValue(ONAP_CSAR_ID);
+        vnf.setVnfConfigurableProperties(new ArrayList<>());
+        vnf.getVnfConfigurableProperties().add(prop);
+        ApiException expectedException = new ApiException();
+        when(nsLcmApi.grantvnf(Mockito.any())).thenThrow(expectedException);
+        //when
+        try {
+            vfcGrantManager.requestGrantForTerminate(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, vnf, JOB_ID);
+            //verify
+            fail();
+        } catch (RuntimeException e) {
+            verify(logger).error(Mockito.eq("Unable to request grant"), Mockito.eq(expectedException));
+        }
+    }
+
+    /**
+     * failuire is to request grant is logged
+     */
+    @Test
+    public void testFailureToRequestGrantForScaleIsLogged() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.scale.yaml").toURI())));
+        VnfScaleRequest scaleRequest = new VnfScaleRequest();
+        scaleRequest.setType(ScaleDirection.OUT);
+        scaleRequest.setAspectId("aspect1");
+        scaleRequest.setNumberOfSteps("2");
+        com.nokia.cbam.lcm.v32.ApiException expectedException = new com.nokia.cbam.lcm.v32.ApiException();
+        when(vnfApi.vnfsVnfInstanceIdGet(VNF_ID, NOKIA_LCM_API_VERSION)).thenThrow(expectedException);
+        //when
+        try {
+            vfcGrantManager.requestGrantForScale(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, scaleRequest, JOB_ID);
+            //verify
+            fail();
+        } catch (RuntimeException e) {
+            verify(logger).error(Mockito.eq("Unable to query VNF myVnfId"), Mockito.eq(expectedException));
+            assertEquals(e.getCause(), expectedException);
+        }
+    }
+
+    /**
+     * test grant request for scale out
+     */
+    @Test
+    public void testGrantDuringScaleOut() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.scale.yaml").toURI())));
+        VnfScaleRequest scaleRequest = new VnfScaleRequest();
+        scaleRequest.setType(ScaleDirection.OUT);
+        scaleRequest.setAspectId("aspect1");
+        scaleRequest.setNumberOfSteps("2");
+        VnfInfo vnf = new VnfInfo();
+        when(vnfApi.vnfsVnfInstanceIdGet(VNF_ID, NOKIA_LCM_API_VERSION)).thenReturn(vnf);
+        vnf.setVnfdId(CBAM_VNFD_ID);
+        when(cbamCatalogManager.getCbamVnfdContent(VNFM_ID, CBAM_VNFD_ID)).thenReturn(cbamVnfdContent);
+        //when
+        vfcGrantManager.requestGrantForScale(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, scaleRequest, JOB_ID);
+        //verify
+        assertEquals(1, grantRequest.getAllValues().size());
+        GrantVNFRequest request = grantRequest.getValue();
+        assertVduInGrant(request.getAddResource(), "vdu1", 4);
+        assertVduInGrant(request.getAddResource(), "vdu2", 2);
+        assertEquals(0, request.getRemoveResource().size());
+        assertBasicGrantAttributes(request, org.onap.vnfmdriver.model.OperationType.SCALEOUT);
+    }
+
+    /**
+     * test grant request for scale in
+     */
+    @Test
+    public void testGrantDuringScaleIn() throws Exception {
+        String cbamVnfdContent = new String(readAllBytes(Paths.get(TestVfcGrantManager.class.getResource("/unittests/vnfd.scale.yaml").toURI())));
+        VnfScaleRequest scaleRequest = new VnfScaleRequest();
+        scaleRequest.setType(ScaleDirection.IN);
+        scaleRequest.setAspectId("aspect1");
+        scaleRequest.setNumberOfSteps("2");
+        VnfInfo vnf = new VnfInfo();
+        when(vnfApi.vnfsVnfInstanceIdGet(VNF_ID, NOKIA_LCM_API_VERSION)).thenReturn(vnf);
+        vnf.setVnfdId(CBAM_VNFD_ID);
+        when(cbamCatalogManager.getCbamVnfdContent(VNFM_ID, CBAM_VNFD_ID)).thenReturn(cbamVnfdContent);
+        //when
+        vfcGrantManager.requestGrantForScale(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, scaleRequest, JOB_ID);
+        //verify
+        assertEquals(1, grantRequest.getAllValues().size());
+        GrantVNFRequest request = grantRequest.getValue();
+        assertVduInGrant(request.getRemoveResource(), "vdu1", 4);
+        assertVduInGrant(request.getRemoveResource(), "vdu2", 2);
+        assertEquals(0, request.getAddResource().size());
+        assertBasicGrantAttributes(request, org.onap.vnfmdriver.model.OperationType.SCALEIN);
+    }
+
+
+    /**
+     * test grant request for healing
+     */
+    @Test
+    public void testGrantDuringHealing() throws Exception {
+        //when
+        VnfHealRequest healRequest = new VnfHealRequest();
+        VnfHealRequestAffectedvm affectedVm = new VnfHealRequestAffectedvm();
+        affectedVm.setVduid("vdu1");
+        healRequest.setAffectedvm(affectedVm);
+        vfcGrantManager.requestGrantForHeal(VNFM_ID, VNF_ID, VIM_ID, ONAP_CSAR_ID, healRequest, JOB_ID);
+        //verify
+        assertEquals(1, grantRequest.getAllValues().size());
+        GrantVNFRequest request = grantRequest.getValue();
+        assertVduInGrant(request.getAddResource(), "vdu1", 1);
+        assertVduInGrant(request.getRemoveResource(), "vdu1", 1);
+        assertBasicGrantAttributes(request, org.onap.vnfmdriver.model.OperationType.HEAL);
+    }
+
+    private void assertBasicGrantAttributes(GrantVNFRequest request, org.onap.vnfmdriver.model.OperationType type) {
+        assertEquals(JOB_ID, request.getJobId());
+        assertEquals(type, request.getLifecycleOperation());
+        assertEquals(ONAP_CSAR_ID, request.getVnfDescriptorId());
+        assertEquals(VNF_ID, request.getVnfInstanceId());
+        JsonObject additionalParams = new Gson().toJsonTree(request.getAdditionalParam()).getAsJsonObject();
+        assertEquals(VIM_ID, additionalParams.get("vimId").getAsString());
+        assertEquals(VNFM_ID, additionalParams.get("vnfmId").getAsString());
+    }
+
+    private void assertVduInGrant(List<ResourceChange> changes, String vduName, int count) {
+        ArrayList<ResourceChange> clonedChanges = Lists.newArrayList(changes);
+        for (int i = 0; i < count + 1; i++) {
+            Iterator<ResourceChange> iter = clonedChanges.iterator();
+            boolean found = false;
+            while (iter.hasNext()) {
+                ResourceChange change = iter.next();
+                if (change.getVdu().equals(vduName)) {
+                    iter.remove();
+                    found = true;
+                    break;
+                }
+            }
+            if (i >= count) {
+                assertFalse(found);
+            } else {
+                assertTrue(found);
+            }
+        }
+    }
+
+}