Add unit test for PNF registration function.
[aai/esr-server.git] / esr-mgr / src / test / java / org / onap / aai / esr / wrapper / PnfManagerWrapperTest.java
1 /**
2  * Copyright 2018 ZTE Corporation.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.onap.aai.esr.wrapper;
17
18 import static org.junit.Assert.assertEquals;
19 import java.util.ArrayList;
20 import java.util.List;
21 import javax.ws.rs.core.Response;
22 import org.junit.Assert;
23 import org.junit.Test;
24 import org.mockito.Mockito;
25 import org.onap.aai.esr.common.MsbConfig;
26 import org.onap.aai.esr.entity.aai.Pnf;
27 import org.onap.aai.esr.entity.rest.PnfRegisterInfo;
28 import org.onap.aai.esr.exception.ExtsysException;
29 import org.onap.aai.esr.externalservice.aai.NetworkProxy;
30 import org.onap.aai.esr.util.ExtsysUtil;
31
32 public class PnfManagerWrapperTest {
33
34     static {
35         MsbConfig.setMsbServerAddr("http://127.0.0.1:80");
36     }
37
38     @Test
39     public void test_registerPnf() throws ExtsysException {
40         PnfRegisterInfo pnfRegisterInfo = new PnfRegisterInfo();
41         pnfRegisterInfo.setPnfId("pnf1");
42         pnfRegisterInfo.setUserLabel("PNF test");
43         pnfRegisterInfo.setSubnetId("subnetId1");
44         pnfRegisterInfo.setNeId("neId1");
45         pnfRegisterInfo.setManagementType("Test");
46         pnfRegisterInfo.setVendor("ZTE");
47         pnfRegisterInfo.setPnfdId("pnfdId1");
48         pnfRegisterInfo.setEmsId("emsId1");
49         pnfRegisterInfo.setLattitude("121.546");
50         pnfRegisterInfo.setLongitude("14.22");
51         NetworkProxy mockNetworkProxy = Mockito.mock(NetworkProxy.class);
52         Mockito.doNothing().when(mockNetworkProxy).registerPnf(Mockito.anyString(), (Pnf)Mockito.anyObject());
53         PnfManagerWrapper pnfManagerWrapper = new PnfManagerWrapper(mockNetworkProxy);
54         Response response = pnfManagerWrapper.registerPnf(pnfRegisterInfo);
55         if (response != null) {
56             Assert.assertTrue(response.getStatus() == 200);
57         }
58     }
59     
60     @Test
61     public void test_queryPnfById() throws ExtsysException {
62         ExtsysUtil extsysUtil = new ExtsysUtil();
63         PnfRegisterInfo pnfRegisterInfo = new PnfRegisterInfo();
64         pnfRegisterInfo.setPnfId("pnf1");
65         pnfRegisterInfo.setUserLabel("PNF test");
66         pnfRegisterInfo.setSubnetId("subnetId1");
67         pnfRegisterInfo.setNeId("neId1");
68         pnfRegisterInfo.setManagementType("Test");
69         pnfRegisterInfo.setVendor("ZTE");
70         pnfRegisterInfo.setPnfdId("pnfdId1");
71         pnfRegisterInfo.setEmsId("emsId1");
72         pnfRegisterInfo.setLattitude("121.546");
73         pnfRegisterInfo.setLongitude("14.22");
74         String pnfStr = "{\"pnf-name\": \"pnf1\","
75                 + "\"pnf-name2\": \"PNF test\","
76                 + "\"pnf-id\": \"subnetId1-neId1\","
77                 + "\"equip-type\": \"Test\","
78                 + "\"equip-vendor\": \"ZTE\","
79                 + "\"equip-model\": \"pnfdId1\","
80                 + "\"management-option\": \"emsId1\","
81                 + "\"in-maint\": false,"
82                 + "\"frame-id\": \"121.546-14.22\"}";
83         NetworkProxy mockNetworkProxy = Mockito.mock(NetworkProxy.class);
84         Mockito.when(mockNetworkProxy.queryPNF(Mockito.anyString())).thenReturn(pnfStr);
85         PnfManagerWrapper pnfManagerWrapper = new PnfManagerWrapper(mockNetworkProxy);
86         Response response = pnfManagerWrapper.queryPnfById("pnf1");
87         if (response != null) {
88             Assert.assertTrue(response.getStatus() == 200);
89             assertEquals(extsysUtil.objectToString(pnfRegisterInfo), extsysUtil.objectToString(response.getEntity()));
90         }
91     }
92     
93     @Test
94     public void test_queryPnfList() throws ExtsysException {
95         ExtsysUtil extsysUtil = new ExtsysUtil();
96         List<PnfRegisterInfo> esrPnfList = new ArrayList<>();
97         PnfRegisterInfo pnfRegisterInfo = new PnfRegisterInfo();
98         pnfRegisterInfo.setPnfId("pnf1");
99         pnfRegisterInfo.setUserLabel("PNF test");
100         pnfRegisterInfo.setSubnetId("subnetId1");
101         pnfRegisterInfo.setNeId("neId1");
102         pnfRegisterInfo.setManagementType("Test");
103         pnfRegisterInfo.setVendor("ZTE");
104         pnfRegisterInfo.setPnfdId("pnfdId1");
105         pnfRegisterInfo.setEmsId("emsId1");
106         pnfRegisterInfo.setLattitude("121.546");
107         pnfRegisterInfo.setLongitude("14.22");
108         esrPnfList.add(pnfRegisterInfo);
109         String pnfListStr = "{\"pnf\": [{\"pnf-name\": \"pnf1\","
110                 + "\"pnf-name2\": \"PNF test\","
111                 + "\"pnf-id\": \"subnetId1-neId1\","
112                 + "\"equip-type\": \"Test\","
113                 + "\"equip-vendor\": \"ZTE\","
114                 + "\"equip-model\": \"pnfdId1\","
115                 + "\"management-option\": \"emsId1\","
116                 + "\"in-maint\": false,"
117                 + "\"frame-id\": \"121.546-14.22\"}]}";
118         NetworkProxy mockNetworkProxy = Mockito.mock(NetworkProxy.class);
119         Mockito.when(mockNetworkProxy.queryPnfList()).thenReturn(pnfListStr);
120         PnfManagerWrapper pnfManagerWrapper = new PnfManagerWrapper(mockNetworkProxy);
121         Response response = pnfManagerWrapper.queryPnfList();
122         if (response != null) {
123             Assert.assertTrue(response.getStatus() == 200);
124             assertEquals(extsysUtil.objectToString(esrPnfList), extsysUtil.objectToString(response.getEntity()));
125         }
126     }
127     
128     @Test
129     public void test_delPnf() throws ExtsysException {
130         NetworkProxy mockNetworkProxy = Mockito.mock(NetworkProxy.class);
131         Mockito.doNothing().when(mockNetworkProxy).deletePnf(Mockito.anyString(), Mockito.anyString());
132         PnfManagerWrapper pnfManagerWrapper = new PnfManagerWrapper(mockNetworkProxy);
133         Response response = pnfManagerWrapper.delPnf("pnf1");
134         if (response != null) {
135             Assert.assertTrue(response.getStatus() == 204);
136         }
137     }
138     
139     @Test
140     public void test_updatePnf() throws ExtsysException {
141         PnfRegisterInfo pnfRegisterInfo = new PnfRegisterInfo();
142         pnfRegisterInfo.setPnfId("pnf1");
143         pnfRegisterInfo.setUserLabel("PNF test");
144         pnfRegisterInfo.setSubnetId("subnetId1");
145         pnfRegisterInfo.setNeId("neId1");
146         pnfRegisterInfo.setManagementType("Test");
147         pnfRegisterInfo.setVendor("ZTE");
148         pnfRegisterInfo.setPnfdId("pnfdId1");
149         pnfRegisterInfo.setEmsId("emsId1");
150         pnfRegisterInfo.setLattitude("121.546");
151         pnfRegisterInfo.setLongitude("14.22");
152         String pnfStr = "{\"pnf-name\": \"pnf1\","
153                 + "\"pnf-name2\": \"PNF test\","
154                 + "\"pnf-id\": \"subnetId1-neId1\","
155                 + "\"equip-type\": \"Test\","
156                 + "\"equip-vendor\": \"ZTE\","
157                 + "\"equip-model\": \"pnfdId1\","
158                 + "\"management-option\": \"emsId1\","
159                 + "\"in-maint\": false,"
160                 + "\"frame-id\": \"121.546-14.22\"}";
161         NetworkProxy mockNetworkProxy = Mockito.mock(NetworkProxy.class);
162         Mockito.doNothing().when(mockNetworkProxy).registerPnf(Mockito.anyString(), (Pnf)Mockito.anyObject());
163         Mockito.when(mockNetworkProxy.queryPNF(Mockito.anyString())).thenReturn(pnfStr);
164         PnfManagerWrapper vnfmManagerWrapper = new PnfManagerWrapper(mockNetworkProxy);
165         Response response = vnfmManagerWrapper.updatePnf(pnfRegisterInfo, "pnf1");
166         if (response != null) {
167             Assert.assertTrue(response.getStatus() == 200);
168         }
169     }
170 }