2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017, 2019-2020 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Modifications Copyright (C) 2019 Samsung
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
13 * http://www.apache.org/licenses/LICENSE-2.0
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 * ============LICENSE_END=========================================================
23 package org.onap.policy.admin;
25 import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
26 import static org.junit.Assert.assertNotNull;
27 import static org.junit.Assert.assertNull;
28 import static org.junit.Assert.fail;
29 import static org.mockito.Mockito.atLeast;
30 import static org.mockito.Mockito.mock;
31 import static org.mockito.Mockito.reset;
32 import static org.mockito.Mockito.verify;
33 import static org.mockito.Mockito.when;
35 import java.io.BufferedReader;
36 import java.io.IOException;
37 import java.io.InputStream;
38 import java.io.PrintWriter;
39 import java.io.StringReader;
40 import java.net.HttpURLConnection;
41 import java.util.ArrayList;
42 import java.util.List;
43 import javax.servlet.http.HttpServletRequest;
44 import javax.servlet.http.HttpServletResponse;
45 import javax.servlet.http.HttpSession;
46 import org.junit.Before;
47 import org.junit.Test;
48 import org.junit.runner.RunWith;
49 import org.mockito.Mockito;
50 import org.onap.policy.controller.CreateFirewallController;
51 import org.onap.policy.controller.PolicyController;
52 import org.onap.policy.rest.dao.CommonClassDao;
53 import org.onap.policy.rest.jpa.ActionList;
54 import org.onap.policy.rest.jpa.AddressGroup;
55 import org.onap.policy.rest.jpa.FwTagPicker;
56 import org.onap.policy.rest.jpa.GroupServiceList;
57 import org.onap.policy.rest.jpa.PrefixList;
58 import org.onap.policy.rest.jpa.SecurityZone;
59 import org.onap.policy.rest.jpa.ServiceList;
60 import org.onap.policy.rest.jpa.TermList;
61 import org.onap.portalsdk.core.domain.User;
62 import org.onap.portalsdk.core.util.SystemProperties;
63 import org.onap.portalsdk.core.web.support.UserUtils;
64 import org.powermock.api.mockito.PowerMockito;
65 import org.powermock.core.classloader.annotations.PowerMockIgnore;
66 import org.powermock.core.classloader.annotations.PrepareForTest;
67 import org.powermock.modules.junit4.PowerMockRunner;
68 import org.powermock.reflect.Whitebox;
69 import org.springframework.mock.web.MockHttpServletResponse;
71 @RunWith(PowerMockRunner.class)
72 @PowerMockIgnore({"com.sun.org.apache.xerces.*", "javax.xml.*", "org.xml.*", "com.sun.org.apache.xalan.*"})
73 public class PolicyRestControllerTest {
75 private String clRequestString;
76 private String fwRequestString;
77 private String fwViewRequestString;
78 private HttpServletRequest request;
79 private MockHttpServletResponse response;
80 private static CommonClassDao commonClassDao;
81 private List<Object> prefixListData;
82 private List<Object> actionListData;
83 private List<Object> serviceListData;
84 private List<Object> addressGroupData;
85 private List<Object> securityZoneData;
86 private List<Object> serviceGroupData;
87 private List<Object> tagListData;
88 private List<Object> termListData;
93 * @throws Exception exception
96 public void setUp() throws Exception {
97 commonClassDao = mock(CommonClassDao.class);
98 final HttpSession mockSession = mock(HttpSession.class);
99 request = mock(HttpServletRequest.class);
100 response = new MockHttpServletResponse();
101 User user = new User();
102 user.setOrgUserId("Test");
103 Mockito.when(mockSession.getAttribute(SystemProperties.getProperty("user_attribute_name"))).thenReturn(user);
104 Mockito.when(request.getSession(false)).thenReturn(mockSession);
105 clRequestString = "{\"policyData\":{\"error\":\"\",\"inprocess\":false,\"model\":{\"name\":\"com\","
106 + "\"subScopename\":\"\",\"path\":[],\"type\":\"dir\",\"size\":0,"
107 + "\"date\":\"2017-06-01T15:45:36.000Z\",\"version\":\"\",\"createdBy\":\"Demo\","
108 + "\"modifiedBy\":\"Demo\",\"content\":\"\",\"recursive\":false},"
109 + "\"tempModel\":{\"name\":\"com\",\"subScopename\":\"\",\"path\":[],\"type\":\"dir\","
110 + "\"size\":0,\"date\":\"2017-06-01T15:45:36.000Z\",\"version\":\"\",\"createdBy\":\"Demo\","
111 + "\"modifiedBy\":\"Demo\",\"content\":\"\",\"recursive\":false},"
112 + "\"$$hashKey\":\"object:1439\",\"policy\":{\"policyType\":\"Config\","
113 + "\"configPolicyType\":\"ClosedLoop_Fault\",\"triggerTrapSignatures\":[1,1,2,3],"
114 + "\"triggerfaultSignatures\":[1,1,2,3],\"traptriggerSignatures\":[{\"id\":\"Trap1\","
115 + "\"$$hashKey\":\"object:1526\"},{\"id\":\"Trap2\",\"$$hashKey\":\"object:1534\"}],"
116 + "\"connecttriggerSignatures\":[{\"id\":\"C1\",\"$$hashKey\":\"object:1554\","
117 + "\"notBox\":\"NOT\",\"connectTrap1\":\"Trap1\",\"trapCount1\":\"12\","
118 + "\"operatorBox\":\"AND\",\"connectTrap2\":\"Trap2\",\"trapCount2\":\"14\"}],"
119 + "\"faulttriggerSignatures\":[{\"id\":\"Fault1\",\"$$hashKey\":\"object:1566\"},"
120 + "{\"id\":\"Fault2\",\"$$hashKey\":\"object:1575\"}],\"connectVerificationSignatures\":"
121 + "[{\"id\":\"C1\",\"$$hashKey\":\"object:1595\",\"notBox\":\"NOT\","
122 + "\"connectTrap1\":\"Fault1\",\"trapCount1\":\"11\",\"operatorBox\":\"AND\","
123 + "\"connectTrap2\":\"Fault2\",\"trapCount2\":\"12\"}],\"jsonBodyData\":"
124 + "{\"trapMaxAge\":\"300\",\"vnfType\":\"Test\",\"closedLoopPolicyStatus\":\"Active\","
125 + "\"vUSP\":true,\"trinity\":true,\"vDNS\":true,\"mcr\":true,\"gamma\":true,"
126 + "\"actions\":\"Test\",\"timeInterval\":\"11\",\"timeOutvPRO\":\"11\",\"timeOutRuby\":\"11\","
127 + "\"retrys\":\"1\",\"agingWindow\":\"12\",\"geoLink\":\"test\","
128 + "\"emailAddress\":\"aa@test.com\",\"pepName\":\"Test\",\"pepAction\":\"test\","
129 + "\"conditions\":\"SEND\"},\"policyName\":\"SampleTest\","
130 + "\"policyDescription\":\"SampleTest\",\"riskType\":\"SampleRiskType\",\"riskLevel\":\"1\","
131 + "\"guard\":\"True\",\"onapName\":\"SampleDemo\",\"ttlDate\":\"14/09/2017\","
132 + "\"clearTimeOut\":\"123\",\"trapMaxAge\":\"11\",\"verificationclearTimeOut\":\"13\"}},"
133 + "\"trapData\":{\"trap1\":[{\"id\":\"A1\",\"$$hashKey\":\"object:1528\","
134 + "\"notBox\":\"NOT\",\"trigger1\":\"Test\",\"operatorBox\":\"AND\",\"trigger2\":\"Test\"}],"
135 + "\"trap2\":[{\"id\":\"A1\",\"$$hashKey\":\"object:1536\",\"notBox\":\"NOT\","
136 + "\"trigger1\":\"Test\",\"operatorBox\":\"AND\",\"trigger2\":\"Test\"},{\"id\":\"A2\","
137 + "\"$$hashKey\":\"object:1542\",\"notBox\":\"NOT\",\"trigger1\":\"A1\","
138 + "\"operatorBox\":\"AND\",\"trigger2\":\"Test\"},{\"id\":\"A3\","
139 + "\"$$hashKey\":\"object:1548\",\"notBox\":\"NOT\",\"trigger1\":\"A1\","
140 + "\"operatorBox\":\"OR\",\"trigger2\":\"A2\"}]},\"faultData\":{\"trap1\":[{\"id\":\"A1\","
141 + "\"$$hashKey\":\"object:1568\",\"notBox\":\"NOT\",\"trigger1\":\"Test\","
142 + "\"operatorBox\":\"AND\",\"trigger2\":\"Test\"}],\"trap2\":[{\"id\":\"A1\","
143 + "\"$$hashKey\":\"object:1577\",\"notBox\":\"NOT\",\"trigger1\":\"Test\","
144 + "\"operatorBox\":\"AND\",\"trigger2\":\"Test\"},{\"id\":\"A2\",\"$$hashKey\":"
145 + "\"object:1583\",\"notBox\":\"NOT\",\"trigger1\":\"Test\",\"operatorBox\":\"OR\","
146 + "\"trigger2\":\"Test\"},{\"id\":\"A3\",\"$$hashKey\":\"object:1589\",\"notBox\":\"NOT\","
147 + "\"trigger1\":\"A1\",\"operatorBox\":\"AND\",\"trigger2\":\"A2\"}]}}";
149 fwRequestString = "{\"policyData\":{\"error\":\"\",\"inprocess\":false,\"model\":{\"name\":\"com\","
150 + "\"subScopename\":\"\",\"path\":[],\"type\":\"dir\",\"size\":0,"
151 + "\"date\":\"2017-06-01T15:45:36.000Z\",\"version\":\"\",\"createdBy\":\"Demo\","
152 + "\"modifiedBy\":\"Demo\",\"content\":\"\",\"recursive\":false},\"tempModel\":"
153 + "{\"name\":\"com\",\"subScopename\":\"\",\"path\":[],\"type\":\"dir\",\"size\":0,"
154 + "\"date\":\"2017-06-01T15:45:36.000Z\",\"version\":\"\",\"createdBy\":\"Demo\","
155 + "\"modifiedBy\":\"Demo\",\"content\":\"\",\"recursive\":false},\"$$hashKey\":\"object:260\","
156 + "\"policy\":{\"policyType\":\"Config\",\"configPolicyType\":\"Firewall Config\","
157 + "\"attributes\":[{\"id\":\"choice1\",\"$$hashKey\":\"object:338\",\"key\":\"Test\","
158 + "\"value\":\"Test\"}],\"fwattributes\":[],\"policyName\":\"SampleTest\","
159 + "\"policyDescription\":\"SampleTest\",\"riskType\":\"SampleRiskType\",\"riskLevel\":\"1\","
160 + "\"guard\":\"True\",\"configName\":\"SampleTest\",\"ttlDate\":\"14/09/2017\","
161 + "\"securityZone\":\"Test\"}}}";
163 fwViewRequestString =
164 "{\"policyData\":{\"policyType\":\"Config\",\"configPolicyType\":\"Firewall Config\",\"attributes\":"
165 + "[{\"id\":\"choice1\",\"$$hashKey\":\"object:338\",\"key\":\"Test\",\"value\":\"Test\"}],"
166 + "\"fwattributes\":[],\"policyName\":\"SampleTest\",\"policyDescription\":\"SampleTest\","
167 + "\"riskType\":\"SampleRiskType\",\"riskLevel\":\"1\",\"guard\":\"True\","
168 + "\"configName\":\"SampleTest\",\"ttlDate\":\"14/09/2017\",\"securityZone\":\"Test\"}}";
170 prefixListData = new ArrayList<>();
171 PrefixList prefixList = new PrefixList();
172 prefixList.setPrefixListName("Test");
173 prefixList.setPrefixListValue("10.10.10.10/12");
174 prefixListData.add(prefixList);
175 when(commonClassDao.getData(PrefixList.class)).thenReturn(prefixListData);
177 actionListData = new ArrayList<>();
178 ActionList actionList = new ActionList();
179 actionList.setActionName("Test");
180 actionListData.add(actionList);
181 when(commonClassDao.getData(ActionList.class)).thenReturn(actionListData);
183 serviceListData = new ArrayList<>();
184 ServiceList serviceList = new ServiceList();
185 serviceList.setServiceName("Test");
186 serviceList.setServiceType("SERVICE");
187 serviceList.setServiceTransportProtocol("Test");
188 serviceList.setServiceAppProtocol("Test");
189 serviceList.setServicePorts("8080");
190 serviceListData.add(serviceList);
191 when(commonClassDao.getData(ServiceList.class)).thenReturn(serviceListData);
193 addressGroupData = new ArrayList<>();
194 AddressGroup addressGroup = new AddressGroup();
195 addressGroup.setGroupName("Group_Test");
196 addressGroup.setServiceList("Test");
197 addressGroupData.add(addressGroup);
198 when(commonClassDao.getData(AddressGroup.class)).thenReturn(addressGroupData);
200 securityZoneData = new ArrayList<>();
201 SecurityZone securityZone = new SecurityZone();
202 securityZone.setZoneName("Test");
203 securityZone.setZoneValue("Test");
204 securityZoneData.add(securityZone);
205 when(commonClassDao.getData(SecurityZone.class)).thenReturn(securityZoneData);
207 serviceGroupData = new ArrayList<>();
208 GroupServiceList serviceGroup = new GroupServiceList();
209 serviceGroup.setGroupName("Group_Test");
210 serviceGroup.setServiceList("Test");
211 serviceGroupData.add(serviceGroup);
212 when(commonClassDao.getData(GroupServiceList.class)).thenReturn(serviceGroupData);
214 tagListData = new ArrayList<>();
215 FwTagPicker fwPicker = new FwTagPicker();
216 fwPicker.setTagPickerName("Test");
217 fwPicker.setTagValues("Test:8080");
218 tagListData.add(fwPicker);
219 when(commonClassDao.getData(FwTagPicker.class)).thenReturn(tagListData);
221 termListData = new ArrayList<>();
222 TermList termList = new TermList();
223 termList.setTermName("Test");
224 termList.setFromZone("Test");
225 termList.setToZone("Test");
226 termList.setSrcIpList("Test");
227 termList.setDestIpList("Test");
228 termList.setSrcPortList("Test");
229 termList.setDestPortList("Test");
230 termList.setAction("Test");
231 termListData.add(termList);
232 when(commonClassDao.getData(TermList.class)).thenReturn(termListData);
233 when(commonClassDao.getDataById(TermList.class, "termName", "Test")).thenReturn(termListData);
237 public final void testPolicyCreationController() {
238 assertNull(PolicyRestController.getCommonClassDao());
239 PolicyRestController.setCommonClassDao(commonClassDao);
240 assertNotNull(PolicyRestController.getCommonClassDao());
242 PolicyRestController controller = new PolicyRestController();
243 BufferedReader reader = new BufferedReader(new StringReader(clRequestString));
245 Mockito.when(request.getReader()).thenReturn(reader);
246 controller.policyCreationController(request, response);
247 } catch (IOException e) {
250 PolicyRestController controller1 = new PolicyRestController();
251 CreateFirewallController.setCommonClassDao(commonClassDao);
252 assertNotNull(CreateFirewallController.getCommonClassDao());
253 BufferedReader reader1 = new BufferedReader(new StringReader(fwRequestString));
255 Mockito.when(request.getReader()).thenReturn(reader1);
256 controller1.policyCreationController(request, response);
257 } catch (IOException e) {
261 CreateFirewallController fwController = new CreateFirewallController();
262 CreateFirewallController.setCommonClassDao(commonClassDao);
263 BufferedReader reader2 = new BufferedReader(new StringReader(fwViewRequestString));
265 Mockito.when(request.getReader()).thenReturn(reader2);
266 fwController.setFWViewRule(request, response);
267 } catch (IOException e) {
273 public final void testSearchPolicy() throws IOException {
274 PolicyController.setjUnit(true);
275 PolicyController.setPapUrl("http://localhost:8070/pap/");
276 PolicyRestController controller = new PolicyRestController();
277 BufferedReader reader = new BufferedReader(new StringReader("{\"foo\":\"bar\"}"));
278 Mockito.when(request.getReader()).thenReturn(reader);
279 Mockito.when(request.getRequestURI()).thenReturn("/pap/foo/");
280 assertThatExceptionOfType(NullPointerException.class)
281 .isThrownBy(() -> controller.searchPolicy(request, response));
285 public final void testSearchDictionaryController() throws IOException {
286 PolicyRestController controller = new PolicyRestController();
287 BufferedReader reader = new BufferedReader(new StringReader("{\"foo\":\"bar\"}"));
288 Mockito.when(request.getReader()).thenReturn(reader);
289 Mockito.when(request.getRequestURI()).thenReturn("/pap/foo/");
290 assertNull(controller.searchDictionaryController(request, response));
294 public final void testDeleteElasticData() {
295 PolicyRestController controller = new PolicyRestController();
297 controller.deleteElasticData("Test");
298 } catch (Exception e) {
304 public final void testNotifyOtherPaps() {
305 PolicyRestController controller = new PolicyRestController();
306 String strReturn = controller.notifyOtherPapsToUpdateConfigurations("mode", "newName", "oldName");
307 assertNull(strReturn);
310 @SuppressWarnings("unchecked")
312 public void testGetDictionaryController() throws IOException {
313 PolicyRestController controller = new PolicyRestController();
314 HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class);
315 HttpServletResponse mockResponse = Mockito.mock(HttpServletResponse.class);
316 PrintWriter mockPrintWriter = Mockito.mock(PrintWriter.class);
318 when(mockRequest.getRequestURI()).thenReturn("//testRequestURI///getDictionary");
319 when(mockResponse.getWriter()).thenThrow(IOException.class);
320 controller.getDictionaryController(mockRequest, mockResponse);
321 verify(mockRequest, atLeast(1)).getRequestURI();
322 verify(mockResponse, atLeast(1)).getWriter();
325 when(mockRequest.getRequestURI()).thenReturn("testRequestURI///getDictionary");
326 when(mockResponse.getWriter()).thenReturn(mockPrintWriter);
327 controller.getDictionaryController(mockRequest, mockResponse);
328 verify(mockRequest, atLeast(1)).getRequestURI();
329 verify(mockResponse, atLeast(1)).getWriter();
332 @PrepareForTest(UserUtils.class)
334 public void testdeletetDictionaryController() throws IOException {
335 HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class);
336 HttpServletResponse mockResponse = Mockito.mock(HttpServletResponse.class);
337 PolicyRestController controller = new PolicyRestController();
338 User mockUser = Mockito.mock(User.class);
339 PowerMockito.mockStatic(UserUtils.class);
340 PrintWriter mockPrintWriter = Mockito.mock(PrintWriter.class);
342 when(mockRequest.getRequestURI()).thenReturn("//testRequestURI///deleteDictionary");
343 when(mockResponse.getWriter()).thenReturn(mockPrintWriter);
344 when(UserUtils.getUserSession(mockRequest)).thenReturn(mockUser);
345 when(mockUser.getOrgUserId()).thenReturn("testUserId");
347 controller.deletetDictionaryController(mockRequest, mockResponse);
348 verify(mockRequest).getRequestURI();
349 verify(mockResponse).getWriter();
350 verify(mockUser).getOrgUserId();
352 when(mockRequest.getRequestURI()).thenReturn("testRequestURI///deleteDictionary");
353 controller.deletetDictionaryController(mockRequest, mockResponse);
354 verify(mockRequest, atLeast(1)).getRequestURI();
357 @PrepareForTest(UserUtils.class)
359 public void testSaveDictionaryController() throws IOException {
360 HttpServletRequest mockRequest = Mockito.mock(HttpServletRequest.class);
361 HttpServletResponse mockResponse = Mockito.mock(HttpServletResponse.class);
362 PolicyRestController controller = new PolicyRestController();
363 User mockUser = Mockito.mock(User.class);
364 PowerMockito.mockStatic(UserUtils.class);
365 PrintWriter mockPrintWriter = Mockito.mock(PrintWriter.class);
367 when(mockRequest.getRequestURI()).thenReturn("//testRequestURI///saveDictionary//import_dictionary");
368 when(mockResponse.getWriter()).thenReturn(mockPrintWriter);
369 when(UserUtils.getUserSession(mockRequest)).thenReturn(mockUser);
370 when(mockUser.getOrgUserId()).thenReturn("testUserId");
372 controller.saveDictionaryController(mockRequest, mockResponse);
373 verify(mockRequest).getRequestURI();
374 verify(mockResponse).getWriter();
375 verify(mockUser).getOrgUserId();
377 when(mockRequest.getRequestURI()).thenReturn("testRequestURI///saveDictionary");
378 controller.saveDictionaryController(mockRequest, mockResponse);
379 verify(mockRequest, atLeast(1)).getRequestURI();
383 public void testDoConnect() throws Exception {
384 HttpURLConnection mockConnection = Mockito.mock(HttpURLConnection.class);
385 PolicyRestController controller = new PolicyRestController();
386 Mockito.doNothing().when(mockConnection).connect();
387 InputStream mockInputStream = Mockito.mock(InputStream.class);
389 when(mockConnection.getResponseCode()).thenReturn(200);
390 when(mockConnection.getInputStream()).thenReturn(mockInputStream);
392 Whitebox.invokeMethod(controller, "doConnect", mockConnection);
393 verify(mockConnection).getResponseCode();
394 verify(mockConnection).getInputStream();
396 when(mockConnection.getResponseCode()).thenReturn(201);
397 assertNull(Whitebox.invokeMethod(controller, "doConnect", mockConnection));