appc-config/appc-flow-controller/provider/src/test/java/org/onap/appc/flow/controller/node/CapabilitiesDataExtractorTest.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2018 Nokia. 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 package org.onap.appc.flow.controller.node;
  21
  22 import static org.mockito.Mockito.mock;
  23 import static org.mockito.Mockito.when;
  24
  25 import org.junit.Assert;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.onap.appc.flow.controller.dbervices.FlowControlDBService;
  29 import org.onap.appc.flow.controller.interfaceData.Capabilities;
  30 import org.onap.ccsdk.sli.core.sli.SvcLogicContext;
  31
  32 public class CapabilitiesDataExtractorTest {
  33
  34   private CapabilitiesDataExtractor capabilitiesDataExtractor;
  35   private FlowControlDBService dbService;
  36   private SvcLogicContext ctx;
  37
  38   @Before
  39   public void setUp() {
  40     dbService = mock(FlowControlDBService.class);
  41     ctx = mock(SvcLogicContext.class);
  42     capabilitiesDataExtractor = new CapabilitiesDataExtractor(dbService);
  43   }
  44
  45   @Test
  46   public void should_handle_capabilities_full_config() throws Exception {
  47
  48     String jsonPayload = "{'vnf':['vnf-1', 'vnf-2'],'vf-module':['vf-module-1', 'vf-module-2'],'vnfc':['vnfc-1', 'vnfc-2'],'vm':['vm-1', 'vm-2']}";
  49     when(dbService.getCapabilitiesData(ctx)).thenReturn(jsonPayload.replaceAll("'","\""));
  50
  51     Capabilities capabilitiesData = capabilitiesDataExtractor.getCapabilitiesData(ctx);
  52
  53     Assert.assertEquals("Capabilities [vnf=[vnf-1, vnf-2], vfModule=[vf-module-1, vf-module-2], vm=[vm-1, vm-2], vnfc=[vnfc-1, vnfc-2]]", capabilitiesData.toString());
  54   }
  55
  56   @Test
  57   public void should_handle_capabilities_config_with_missing_params() throws Exception {
  58
  59     // CASE: vm is empty, vnfc is absent
  60     String jsonPayload = "{'vnf':['vnf-1', 'vnf-2'],'vf-module':['vf-module-1'],'vm':[]}";
  61     when(dbService.getCapabilitiesData(ctx)).thenReturn(jsonPayload.replaceAll("'","\""));
  62
  63     Capabilities capabilitiesData = capabilitiesDataExtractor.getCapabilitiesData(ctx);
  64
  65     Assert.assertEquals("Capabilities [vnf=[vnf-1, vnf-2], vfModule=[vf-module-1], vm=[], vnfc=[]]", capabilitiesData.toString());
  66   }
  67
  68 }