Improve coverage FlowControlNode #7
[appc.git] / appc-config / appc-flow-controller / provider / src / test / java / org / onap / appc / flow / controller / node / FlowControlNodeTest.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP : APPC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Copyright (C) 2017 Amdocs
8  * =============================================================================
9  * Licensed under the Apache License, Version 2.0 (the "License");
10  * you may not use this file except in compliance with the License.
11  * You may obtain a copy of the License at
12  *
13  *      http://www.apache.org/licenses/LICENSE-2.0
14  *
15  * Unless required by applicable law or agreed to in writing, software
16  * distributed under the License is distributed on an "AS IS" BASIS,
17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18  * See the License for the specific language governing permissions and
19  * limitations under the License.
20  *
21  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22  * ============LICENSE_END=========================================================
23  */
24
25 package org.onap.appc.flow.controller.node;
26
27 import static org.mockito.Mockito.mock;
28 import static org.mockito.Mockito.when;
29
30 import org.junit.Assert;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.onap.appc.flow.controller.dbervices.FlowControlDBService;
34 import org.onap.appc.flow.controller.interfaceData.Capabilities;
35 import org.onap.ccsdk.sli.core.sli.SvcLogicContext;
36
37 public class FlowControlNodeTest {
38
39   private SvcLogicContext ctx;
40   private FlowControlDBService dbService;
41
42   @Before
43   public void setUp() {
44     ctx = mock(SvcLogicContext.class);
45     dbService = mock(FlowControlDBService.class);
46   }
47
48   @Test
49   public void should_handle_capabilities_full_config() throws Exception {
50
51     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']}";
52     when(dbService.getCapabilitiesData(ctx)).thenReturn(jsonPayload.replaceAll("'","\""));
53
54     FlowControlNode flowControlNode = new FlowControlNode(null, dbService);
55     Capabilities capabilitiesData = flowControlNode.getCapabilitiesData(ctx);
56
57     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());
58   }
59
60   @Test
61   public void should_handle_capabilities_config_with_missing_params() throws Exception {
62
63     // vm is empty, vnfc is absent
64     String jsonPayload = "{'vnf':['vnf-1', 'vnf-2'],'vf-module':['vf-module-1'],'vm':[]}";
65     when(dbService.getCapabilitiesData(ctx)).thenReturn(jsonPayload.replaceAll("'","\""));
66
67     FlowControlNode flowControlNode = new FlowControlNode(null, dbService);
68     Capabilities capabilitiesData = flowControlNode.getCapabilitiesData(ctx);
69
70     Assert.assertEquals("Capabilities [vnf=[vnf-1, vnf-2], vfModule=[vf-module-1], vm=[], vnfc=[]]", capabilitiesData.toString());
71   }
72
73 }