POLICY-SDK-APP/src/test/java/org/onap/policy/admin/PolicyRestControllerTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP Policy Engine
   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
  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  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.policy.admin;
  24
  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;
  34
  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;
  70
  71 @RunWith(PowerMockRunner.class)
  72 @PowerMockIgnore({"com.sun.org.apache.xerces.*", "javax.xml.*", "org.xml.*", "com.sun.org.apache.xalan.*"})
  73 public class PolicyRestControllerTest {
  74
  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;
  89
  90     /**
  91      * Before.
  92      *
  93      * @throws Exception exception
  94      */
  95     @Before
  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\"}]}}";
 148
 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\"}}}";
 162
 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\"}}";
 169
 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);
 176
 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);
 182
 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);
 192
 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);
 199
 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);
 206
 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);
 213
 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);
 220
 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);
 234     }
 235
 236     @Test
 237     public final void testPolicyCreationController() {
 238         assertNull(PolicyRestController.getCommonClassDao());
 239         PolicyRestController.setCommonClassDao(commonClassDao);
 240         assertNotNull(PolicyRestController.getCommonClassDao());
 241
 242         PolicyRestController controller = new PolicyRestController();
 243         BufferedReader reader = new BufferedReader(new StringReader(clRequestString));
 244         try {
 245             Mockito.when(request.getReader()).thenReturn(reader);
 246             controller.policyCreationController(request, response);
 247         } catch (IOException e) {
 248             fail();
 249         }
 250         PolicyRestController controller1 = new PolicyRestController();
 251         CreateFirewallController.setCommonClassDao(commonClassDao);
 252         assertNotNull(CreateFirewallController.getCommonClassDao());
 253         BufferedReader reader1 = new BufferedReader(new StringReader(fwRequestString));
 254         try {
 255             Mockito.when(request.getReader()).thenReturn(reader1);
 256             controller1.policyCreationController(request, response);
 257         } catch (IOException e) {
 258             fail();
 259         }
 260
 261         CreateFirewallController fwController = new CreateFirewallController();
 262         CreateFirewallController.setCommonClassDao(commonClassDao);
 263         BufferedReader reader2 = new BufferedReader(new StringReader(fwViewRequestString));
 264         try {
 265             Mockito.when(request.getReader()).thenReturn(reader2);
 266             fwController.setFWViewRule(request, response);
 267         } catch (IOException e) {
 268             fail();
 269         }
 270     }
 271
 272     @Test
 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));
 282     }
 283
 284     @Test
 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));
 291     }
 292
 293     @Test
 294     public final void testDeleteElasticData() {
 295         PolicyRestController controller = new PolicyRestController();
 296         try {
 297             controller.deleteElasticData("Test");
 298         } catch (Exception e) {
 299             fail();
 300         }
 301     }
 302
 303     @Test
 304     public final void testNotifyOtherPaps() {
 305         PolicyRestController controller = new PolicyRestController();
 306         String strReturn = controller.notifyOtherPapsToUpdateConfigurations("mode", "newName", "oldName");
 307         assertNull(strReturn);
 308     }
 309
 310     @SuppressWarnings("unchecked")
 311     @Test
 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);
 317
 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();
 323
 324         reset(mockResponse);
 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();
 330     }
 331
 332     @PrepareForTest(UserUtils.class)
 333     @Test
 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);
 341
 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");
 346
 347         controller.deletetDictionaryController(mockRequest, mockResponse);
 348         verify(mockRequest).getRequestURI();
 349         verify(mockResponse).getWriter();
 350         verify(mockUser).getOrgUserId();
 351
 352         when(mockRequest.getRequestURI()).thenReturn("testRequestURI///deleteDictionary");
 353         controller.deletetDictionaryController(mockRequest, mockResponse);
 354         verify(mockRequest, atLeast(1)).getRequestURI();
 355     }
 356
 357     @PrepareForTest(UserUtils.class)
 358     @Test
 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);
 366
 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");
 371
 372         controller.saveDictionaryController(mockRequest, mockResponse);
 373         verify(mockRequest).getRequestURI();
 374         verify(mockResponse).getWriter();
 375         verify(mockUser).getOrgUserId();
 376
 377         when(mockRequest.getRequestURI()).thenReturn("testRequestURI///saveDictionary");
 378         controller.saveDictionaryController(mockRequest, mockResponse);
 379         verify(mockRequest, atLeast(1)).getRequestURI();
 380     }
 381
 382     @Test
 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);
 388
 389         when(mockConnection.getResponseCode()).thenReturn(200);
 390         when(mockConnection.getInputStream()).thenReturn(mockInputStream);
 391
 392         Whitebox.invokeMethod(controller, "doConnect", mockConnection);
 393         verify(mockConnection).getResponseCode();
 394         verify(mockConnection).getInputStream();
 395
 396         when(mockConnection.getResponseCode()).thenReturn(201);
 397         assertNull(Whitebox.invokeMethod(controller, "doConnect", mockConnection));
 398     }
 399 }