ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/controller/FirewallDictionaryControllerTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP-PAP-REST
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. 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.policy.pap.xacml.rest.controller;
  21
  22 import static org.junit.Assert.assertTrue;
  23 import static org.junit.Assert.fail;
  24 import static org.mockito.Matchers.any;
  25 import static org.mockito.Mockito.doNothing;
  26 import static org.mockito.Mockito.mock;
  27 import static org.mockito.Mockito.never;
  28 import static org.mockito.Mockito.verify;
  29 import static org.mockito.Mockito.when;
  30
  31 import java.io.BufferedReader;
  32 import java.io.IOException;
  33 import java.io.StringReader;
  34 import java.util.ArrayList;
  35 import java.util.List;
  36
  37 import javax.servlet.http.HttpServletRequest;
  38
  39 import org.junit.Before;
  40 import org.junit.Test;
  41 import org.mockito.Mockito;
  42 import org.onap.policy.common.logging.flexlogger.FlexLogger;
  43 import org.onap.policy.common.logging.flexlogger.Logger;
  44 import org.onap.policy.pap.xacml.rest.util.DictionaryUtils;
  45 import org.onap.policy.rest.adapter.Term;
  46 import org.onap.policy.rest.dao.CommonClassDao;
  47 import org.onap.policy.rest.jpa.ActionList;
  48 import org.onap.policy.rest.jpa.AddressGroup;
  49 import org.onap.policy.rest.jpa.FWTag;
  50 import org.onap.policy.rest.jpa.FWTagPicker;
  51 import org.onap.policy.rest.jpa.FirewallDictionaryList;
  52 import org.onap.policy.rest.jpa.GroupServiceList;
  53 import org.onap.policy.rest.jpa.PortList;
  54 import org.onap.policy.rest.jpa.PrefixList;
  55 import org.onap.policy.rest.jpa.ProtocolList;
  56 import org.onap.policy.rest.jpa.SecurityZone;
  57 import org.onap.policy.rest.jpa.ServiceList;
  58 import org.onap.policy.rest.jpa.TermList;
  59 import org.onap.policy.rest.jpa.UserInfo;
  60 import org.onap.policy.rest.jpa.Zone;
  61 import org.springframework.mock.web.MockHttpServletResponse;
  62
  63 public class FirewallDictionaryControllerTest {
  64
  65         private static Logger logger = FlexLogger.getLogger(FirewallDictionaryControllerTest.class);
  66         private static CommonClassDao commonClassDao;
  67         private String jsonString = null;
  68         private HttpServletRequest request = null;
  69         private FirewallDictionaryController controller = null;
  70         private MockHttpServletResponse response = null;
  71         private UserInfo userInfo;
  72         private List<String> data;
  73
  74         @Before
  75         public void setUp() throws Exception {
  76                 logger.info("setUp: Entering");
  77                 commonClassDao = Mockito.mock(CommonClassDao.class);
  78
  79                 data = new ArrayList<>();
  80                 data.add("Test");
  81
  82                 userInfo = new UserInfo();
  83                 userInfo.setUserLoginId("Test");
  84                 userInfo.setUserName("Test");
  85
  86                 doNothing().when(commonClassDao).delete(any(Term.class));
  87                 doNothing().when(commonClassDao).save(any(Term.class));
  88
  89                 controller = new FirewallDictionaryController();
  90                 FirewallDictionaryController.setCommonClassDao(commonClassDao);
  91
  92                 request = Mockito.mock(HttpServletRequest.class);
  93                 response = new MockHttpServletResponse();
  94                 new DictionaryUtils(commonClassDao);
  95                 DictionaryUtils.setDictionaryUtils(new DictionaryUtils());
  96                 mock(DictionaryUtils.class);
  97                 logger.info("setUp: exit");
  98         }
  99
 100         @Test
 101         public void testGetPrefixListDictionaryEntityDataByName() {
 102                 test_WithGetDataByColumn(PrefixList.class, "prefixListDictionaryDatas", "prefixListName",
 103                                 () -> controller.getPrefixListDictionaryEntityDataByName(response));
 104         }
 105
 106         @Test
 107         public void testGetPrefixListDictionaryEntityData() {
 108                 test_WithGetData(PrefixList.class, "prefixListDictionaryDatas",
 109                                 () -> controller.getPrefixListDictionaryEntityData(response));
 110         }
 111
 112         @Test
 113         public void testGetPortListDictionaryEntityData() {
 114                 test_WithGetData(PortList.class, "portListDictionaryDatas",
 115                                 () -> controller.getPortListDictionaryEntityData(response));
 116         }
 117
 118         @Test
 119         public void testGetProtocolListDictionaryEntityDataByName() {
 120                 test_WithGetDataByColumn(ProtocolList.class, "protocolListDictionaryDatas", "protocolName",
 121                                 () -> controller.getProtocolListDictionaryEntityDataByName(response));
 122         }
 123
 124         @Test
 125         public void testGetProtocolListDictionaryEntityData() {
 126                 test_WithGetData(ProtocolList.class, "protocolListDictionaryDatas",
 127                                 () -> controller.getProtocolListDictionaryEntityData(response));
 128         }
 129
 130         @Test
 131         public void testGetAddressGroupDictionaryEntityDataByName() {
 132                 test_WithGetDataByColumn(AddressGroup.class, "addressGroupDictionaryDatas", "name",
 133                                 () -> controller.getAddressGroupDictionaryEntityDataByName(response));
 134         }
 135
 136         @Test
 137         public void testGetAddressGroupDictionaryEntityData() {
 138                 test_WithGetData(AddressGroup.class, "addressGroupDictionaryDatas",
 139                                 () -> controller.getAddressGroupDictionaryEntityData(response));
 140         }
 141
 142         @Test
 143         public void testGetActionListDictionaryEntityDataByName() {
 144                 test_WithGetDataByColumn(ActionList.class, "actionListDictionaryDatas", "actionName",
 145                                 () -> controller.getActionListDictionaryEntityDataByName(response));
 146         }
 147
 148         @Test
 149         public void testGetActionListDictionaryEntityData() {
 150                 test_WithGetData(ActionList.class, "actionListDictionaryDatas",
 151                                 () -> controller.getActionListDictionaryEntityData(response));
 152         }
 153
 154         @Test
 155         public void testGetServiceGroupDictionaryEntityDataByName() {
 156                 test_WithGetDataByColumn(GroupServiceList.class, "serviceGroupDictionaryDatas", "name",
 157                                 () -> controller.getServiceGroupDictionaryEntityDataByName(response));
 158         }
 159
 160         @Test
 161         public void testGetServiceGroupDictionaryEntityData() {
 162                 test_WithGetData(GroupServiceList.class, "serviceGroupDictionaryDatas",
 163                                 () -> controller.getServiceGroupDictionaryEntityData(response));
 164         }
 165
 166         @Test
 167         public void testGetSecurityZoneDictionaryEntityDataByName() {
 168                 test_WithGetDataByColumn(SecurityZone.class, "securityZoneDictionaryDatas", "zoneName",
 169                                 () -> controller.getSecurityZoneDictionaryEntityDataByName(response));
 170         }
 171
 172         @Test
 173         public void testGetSecurityZoneDictionaryEntityData() {
 174                 test_WithGetData(SecurityZone.class, "securityZoneDictionaryDatas",
 175                                 () -> controller.getSecurityZoneDictionaryEntityData(response));
 176         }
 177
 178         @Test
 179         public void testGetServiceListDictionaryEntityDataByName() {
 180                 test_WithGetDataByColumn(ServiceList.class, "serviceListDictionaryDatas", "serviceName",
 181                                 () -> controller.getServiceListDictionaryEntityDataByName(response));
 182         }
 183
 184         @Test
 185         public void testGetServiceListDictionaryEntityData() {
 186                 test_WithGetData(ServiceList.class, "serviceListDictionaryDatas",
 187                                 () -> controller.getServiceListDictionaryEntityData(response));
 188         }
 189
 190         @Test
 191         public void testGetZoneDictionaryEntityDataByName() {
 192                 test_WithGetDataByColumn(Zone.class, "zoneDictionaryDatas", "zoneName",
 193                                 () -> controller.getZoneDictionaryEntityDataByName(response));
 194         }
 195
 196         @Test
 197         public void testGetZoneDictionaryEntityData() {
 198                 test_WithGetData(Zone.class, "zoneDictionaryDatas", () -> controller.getZoneDictionaryEntityData(response));
 199         }
 200
 201         @Test
 202         public void testGetTermListDictionaryEntityDataByName() {
 203                 test_WithGetDataByColumn(TermList.class, "termListDictionaryDatas", "termName",
 204                                 () -> controller.getTermListDictionaryEntityDataByName(response));
 205         }
 206
 207         @Test
 208         public void testGetTermListDictionaryEntityData() {
 209                 test_WithGetData(TermList.class, "termListDictionaryDatas",
 210                                 () -> controller.getTermListDictionaryEntityData(response));
 211         }
 212
 213         @Test
 214         public void testGetFWDictListDictionaryEntityDataByName() {
 215                 test_WithGetDataByColumn(FirewallDictionaryList.class, "fwDictListDictionaryDatas", "parentItemName",
 216                                 () -> controller.getFWDictListDictionaryEntityDataByName(response));
 217         }
 218
 219         @Test
 220         public void testGetFWDictionaryListEntityData() {
 221                 test_WithGetData(FirewallDictionaryList.class, "fwDictListDictionaryDatas",
 222                                 () -> controller.getFWDictionaryListEntityData(response));
 223         }
 224
 225         @Test
 226         public void testGetTagPickerNameEntityDataByName() {
 227                 test_WithGetDataByColumn(FWTagPicker.class, "fwTagPickerDictionaryDatas", "tagPickerName",
 228                                 () -> controller.getTagPickerNameEntityDataByName(response));
 229         }
 230
 231         @Test
 232         public void testGetTagPickerDictionaryEntityData() {
 233                 test_WithGetData(FWTagPicker.class, "fwTagPickerDictionaryDatas",
 234                                 () -> controller.getTagPickerDictionaryEntityData(response));
 235         }
 236
 237         @Test
 238         public void testGetTagNameEntityDataByName() {
 239                 test_WithGetDataByColumn(FWTag.class, "fwTagDictionaryDatas", "fwTagName",
 240                                 () -> controller.getTagNameEntityDataByName(response));
 241         }
 242
 243         @Test
 244         public void testGetTagDictionaryEntityData() {
 245                 test_WithGetData(FWTag.class, "fwTagDictionaryDatas", () -> controller.getTagDictionaryEntityData(response));
 246         }
 247
 248         @Test
 249         public void testSavePrefixListDictionary() {
 250                 jsonString = "{\"userid\":\"demo\",\"prefixListDictionaryData\":{\"description\":\"test\",\"prefixListName\":\"Test\"}}";
 251                 testSave(PrefixList.class, "prefixListDictionaryDatas", "prefixListName",
 252                                 () -> controller.savePrefixListDictionary(request, response));
 253         }
 254
 255         @Test
 256         public void testUpdatePrefixListDictionary() {
 257                 jsonString = "{\"userid\":\"demo\",\"prefixListDictionaryData\":{\"id\":1,\"description\":\"test\",\"prefixListName\":\"Test\"}}";
 258                 testUpdate(PrefixList.class, "prefixListDictionaryDatas", "prefixListName",
 259                                 () -> controller.savePrefixListDictionary(request, response));
 260         }
 261
 262         @Test
 263         public void testRemovePrefixListDictionary() {
 264                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"prefixListName\":\"Test\"}}";
 265                 testRemove(PrefixList.class, "prefixListDictionaryDatas",
 266                                 () -> controller.removePrefixListDictionary(request, response));
 267         }
 268
 269         @Test
 270         public void testValidatePrefixListDictionary() {
 271                 jsonString = "{\"userid\":\"demo\",\"prefixListDictionaryData\":{\"id\":1,\"description\":\"test\",\"prefixListName\":\"Test\",\"prefixListValue\":\"10.10.10\"}}";
 272                 testValidate(PrefixList.class, "result", () -> controller.validatePrefixListDictionary(request, response));
 273         }
 274
 275         @Test
 276         public void testSavePortListDictionary() {
 277                 jsonString = "{\"userid\":\"demo\",\"portListDictionaryData\":{\"description\":\"test\",\"portName\":\"Test\"}}";
 278                 testSave(PortList.class, "portListDictionaryDatas", "portName",
 279                                 () -> controller.savePortListDictionary(request, response));
 280         }
 281
 282         @Test
 283         public void testUpdatePortListDictionary() {
 284                 jsonString = "{\"userid\":\"demo\",\"portListDictionaryData\":{\"id\":1,\"description\":\"test\",\"portName\":\"Test\"}}";
 285                 testUpdate(PortList.class, "portListDictionaryDatas", "portName",
 286                                 () -> controller.savePortListDictionary(request, response));
 287         }
 288
 289         @Test
 290         public void testRemovePortListDictionary() {
 291                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"portName\":\"Test\"}}";
 292                 testRemove(PortList.class, "portListDictionaryDatas",
 293                                 () -> controller.removePortListDictionary(request, response));
 294         }
 295
 296         @Test
 297         public void testSaveProtocolListDictionary() {
 298                 jsonString = "{\"userid\":\"demo\",\"protocolListDictionaryData\":{\"description\":\"test\",\"protocolName\":\"Test\"}}";
 299                 testSave(ProtocolList.class, "protocolListDictionaryDatas", "protocolName",
 300                                 () -> controller.saveProtocolListDictionary(request, response));
 301         }
 302
 303         @Test
 304         public void testUpdateProtocolListDictionary() {
 305                 jsonString = "{\"userid\":\"demo\",\"protocolListDictionaryData\":{\"id\":1,\"description\":\"test\",\"protocolName\":\"Test\"}}";
 306                 testUpdate(ProtocolList.class, "protocolListDictionaryDatas", "protocolName",
 307                                 () -> controller.saveProtocolListDictionary(request, response));
 308         }
 309
 310         @Test
 311         public void testRemoveProtocolListDictionary() {
 312                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"protocolName\":\"Test\"}}";
 313                 testRemove(ProtocolList.class, "protocolListDictionaryDatas",
 314                                 () -> controller.removeProtocolListDictionary(request, response));
 315         }
 316
 317         @Test
 318         public void testSaveAddressGroupDictionary() {
 319                 jsonString = "{\"addressGroupDictionaryData\":{\"attributes\":[{\"$$hashKey\":\"object:409\",\"id\":\"choice1\",\"option\":\"Test\"}],\"description\":\"test\",\"groupName\":\"Test\"},\"userid\":\"demo\"}";
 320                 testSave(AddressGroup.class, "addressGroupDictionaryDatas", "name", "Group_Test",
 321                                 () -> controller.saveAddressGroupDictionary(request, response));
 322         }
 323
 324         @Test
 325         public void testUpdateAddressGroupDictionary() {
 326                 jsonString = "{\"addressGroupDictionaryData\":{\"id\":1, \"attributes\":[{\"$$hashKey\":\"object:409\",\"id\":\"choice1\",\"option\":\"Test\"}],\"description\":\"test\",\"groupName\":\"Test\"},\"userid\":\"demo\"}";
 327                 testUpdate(AddressGroup.class, "addressGroupDictionaryDatas", "name", "Group_Test",
 328                                 () -> controller.saveAddressGroupDictionary(request, response));
 329         }
 330
 331         @Test
 332         public void testRemoveAddressGroupDictionary() {
 333                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"name\":\"Test\"}}";
 334                 testRemove(AddressGroup.class, "addressGroupDictionaryDatas",
 335                                 () -> controller.removeAddressGroupDictionary(request, response));
 336         }
 337
 338         @Test
 339         public void testSaveActionListDictionary() {
 340                 jsonString = "{\"userid\":\"demo\",\"actionListDictionaryData\":{\"description\":\"test\",\"actionName\":\"Test\"}}";
 341                 testSave(ActionList.class, "actionListDictionaryDatas", "actionName",
 342                                 () -> controller.saveActionListDictionary(request, response));
 343         }
 344
 345         @Test
 346         public void testUpdateActionListDictionary() {
 347                 jsonString = "{\"userid\":\"demo\",\"actionListDictionaryData\":{\"id\":1,\"description\":\"test\",\"actionName\":\"Test\"}}";
 348                 testUpdate(ActionList.class, "actionListDictionaryDatas", "actionName",
 349                                 () -> controller.saveActionListDictionary(request, response));
 350         }
 351
 352         @Test
 353         public void testRemoveActionListDictionary() {
 354                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"actionName\":\"Test\"}}";
 355                 testRemove(ActionList.class, "actionListDictionaryDatas",
 356                                 () -> controller.removeActionListDictionary(request, response));
 357         }
 358
 359         @Test
 360         public void testSaveServiceGroupDictionary() {
 361                 jsonString = "{\"serviceGroupDictionaryData\":{\"attributes\":[{\"$$hashKey\":\"object:657\",\"id\":\"choice1\",\"option\":\"Test\"}],\"groupName\":\"Test\"},\"userid\":\"demo\"}";
 362                 testSave(GroupServiceList.class, "serviceGroupDictionaryDatas", "name", "Group_Test",
 363                                 () -> controller.saveServiceGroupDictionary(request, response));
 364         }
 365
 366         @Test
 367         public void testUpdateServiceGroupDictionary() {
 368                 jsonString = "{\"serviceGroupDictionaryData\":{\"id\":1, \"attributes\":[{\"$$hashKey\":\"object:657\",\"id\":\"choice1\",\"option\":\"Test\"}],\"groupName\":\"Test\"},\"userid\":\"demo\"}";
 369                 testUpdate(GroupServiceList.class, "serviceGroupDictionaryDatas", "name", "Group_Test",
 370                                 () -> controller.saveServiceGroupDictionary(request, response));
 371         }
 372
 373         @Test
 374         public void testRemoveServiceGroupDictionary() {
 375                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"name\":\"Test\"}}";
 376                 testRemove(GroupServiceList.class, "serviceGroupDictionaryDatas",
 377                                 () -> controller.removeServiceGroupDictionary(request, response));
 378         }
 379
 380         @Test
 381         public void testSaveSecurityZoneDictionary() {
 382                 jsonString = "{\"userid\":\"demo\",\"securityZoneDictionaryData\":{\"description\":\"test\",\"zoneName\":\"Test\"}}";
 383                 testSave(SecurityZone.class, "securityZoneDictionaryDatas", "zoneName",
 384                                 () -> controller.saveSecurityZoneDictionary(request, response));
 385         }
 386
 387         @Test
 388         public void testUpdateSecurityZoneDictionary() {
 389                 jsonString = "{\"userid\":\"demo\",\"securityZoneDictionaryData\":{\"id\":1,\"description\":\"test\",\"zoneName\":\"Test\"}}";
 390                 testUpdate(SecurityZone.class, "securityZoneDictionaryDatas", "zoneName",
 391                                 () -> controller.saveSecurityZoneDictionary(request, response));
 392         }
 393
 394         @Test
 395         public void testRemoveSecurityZoneDictionary() {
 396                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"zoneName\":\"Test\"}}";
 397                 testRemove(SecurityZone.class, "securityZoneDictionaryDatas",
 398                                 () -> controller.removeSecurityZoneDictionary(request, response));
 399         }
 400
 401         @Test
 402         public void testSaveServiceListDictionary() {
 403                 jsonString = "{\"serviceListDictionaryData\":{\"appProtocols\":[{\"$$hashKey\":\"object:560\",\"id\":\"choice1\",\"option\":\"Test\"}],\"serviceDescription\":\"test\",\"serviceName\":\"Test\",\"servicePorts\":\"1010\",\"transportProtocols\":[{\"$$hashKey\":\"object:555\",\"id\":\"choice1\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 404                 testSave(ServiceList.class, "serviceListDictionaryDatas", "serviceName",
 405                                 () -> controller.saveServiceListDictionary(request, response));
 406         }
 407
 408         @Test
 409         public void testUpdateServiceListDictionary() {
 410                 jsonString = "{\"serviceListDictionaryData\":{\"appProtocols\":[{\"$$hashKey\":\"object:560\",\"id\":\"choice1\",\"option\":\"Test\"}],\"serviceDescription\":\"test\",\"id\":1,\"serviceName\":\"Test\",\"servicePorts\":\"1010\",\"transportProtocols\":[{\"$$hashKey\":\"object:555\",\"id\":\"choice1\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 411                 testUpdate(ServiceList.class, "serviceListDictionaryDatas", "serviceName",
 412                                 () -> controller.saveServiceListDictionary(request, response));
 413         }
 414
 415         @Test
 416         public void testRemoveServiceListDictionary() {
 417                 jsonString = "{\"data\":{\"appProtocols\":[{\"$$hashKey\":\"object:560\",\"id\":\"choice1\",\"option\":\"Test\"}],\"serviceDescription\":\"test\",\"id\":1,\"serviceName\":\"Test\",\"servicePorts\":\"1010\",\"transportProtocols\":[{\"$$hashKey\":\"object:555\",\"id\":\"choice1\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 418                 testRemove(ServiceList.class, "serviceListDictionaryDatas",
 419                                 () -> controller.removeServiceListDictionary(request, response));
 420         }
 421
 422         @Test
 423         public void testSaveZoneDictionary() {
 424                 jsonString = "{\"userid\":\"demo\",\"zoneDictionaryData\":{\"zoneValue\":\"test\",\"zoneName\":\"Test\"}}";
 425                 testSave(Zone.class, "zoneDictionaryDatas", "zoneName", () -> controller.saveZoneDictionary(request, response));
 426         }
 427
 428         @Test
 429         public void testUpdateZoneDictionary() {
 430                 jsonString = "{\"userid\":\"demo\",\"zoneDictionaryData\":{\"id\":1,\"zoneValue\":\"test\",\"zoneName\":\"Test\"}}";
 431                 testUpdate(Zone.class, "zoneDictionaryDatas", "zoneName",
 432                                 () -> controller.saveZoneDictionary(request, response));
 433         }
 434
 435         @Test
 436         public void testRemoveZoneDictionary() {
 437                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"zoneValue\":\"test\",\"zoneName\":\"Test\"}}";
 438                 testRemove(Zone.class, "zoneDictionaryDatas", () -> controller.removeZoneDictionary(request, response));
 439         }
 440
 441         @Test
 442         public void testSaveTermListDictionary() {
 443                 jsonString = "{\"termListDictionaryData\":{\"actionListDatas\":[{\"$$hashKey\":\"object:1220\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"destinationListDatas\":[{\"$$hashKey\":\"object:1220\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"destinationServiceDatas\":[{\"$$hashKey\":\"object:1230\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"fromZoneDatas\":[{\"$$hashKey\":\"object:1245\",\"id\":\"choice1\",\"option\":\"Test\"}],\"sourceListDatas\":[{\"$$hashKey\":\"object:1215\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"sourceServiceDatas\":[{\"$$hashKey\":\"object:1225\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"termDescription\":\"test\",\"termName\":\"Test\",\"toZoneDatas\":[{\"$$hashKey\":\"object:1240\",\"id\":\"choice1\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 444                 testSave(TermList.class, "termListDictionaryDatas", "termName",
 445                                 () -> controller.saveTermListDictionary(request, response));
 446         }
 447
 448         @Test
 449         public void testUpdateTermListDictionary() {
 450                 jsonString = "{\"termListDictionaryData\":{\"id\":1,\"actionListDatas\":[{\"$$hashKey\":\"object:1220\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"destinationListDatas\":[{\"$$hashKey\":\"object:1220\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"destinationServiceDatas\":[{\"$$hashKey\":\"object:1230\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"fromZoneDatas\":[{\"$$hashKey\":\"object:1245\",\"id\":\"choice1\",\"option\":\"Test\"}],\"sourceListDatas\":[{\"$$hashKey\":\"object:1215\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"sourceServiceDatas\":[{\"$$hashKey\":\"object:1225\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"termDescription\":\"test\",\"termName\":\"Test\",\"toZoneDatas\":[{\"$$hashKey\":\"object:1240\",\"id\":\"choice1\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 451                 testUpdate(TermList.class, "termListDictionaryDatas", "termName",
 452                                 () -> controller.saveTermListDictionary(request, response));
 453         }
 454
 455         @Test
 456         public void testRemoveTermListDictionary() {
 457                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"termDescription\":\"test\",\"termName\":\"Test\"}}";
 458                 testRemove(TermList.class, "termListDictionaryDatas",
 459                                 () -> controller.removeTermListDictionary(request, response));
 460         }
 461
 462         @Test
 463         public void testSaveFWDictionaryList() {
 464                 jsonString = "{\"fwDictListDictionaryData\":{\"alAttributes\":[{\"$$hashKey\":\"object:1379\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"attributes\":[{\"$$hashKey\":\"object:1374\",\"id\":\"choice1\",\"option\":\"Test\"}],\"description\":\"test\",\"parentItemName\":\"Test\"},\"userid\":\"demo\"}";
 465                 testSave(FirewallDictionaryList.class, "fwDictListDictionaryDatas", "parentItemName",
 466                                 () -> controller.saveFWDictionaryList(request, response));
 467         }
 468
 469         @Test
 470         public void testUpdateFWDictionaryList() {
 471                 jsonString = "{\"fwDictListDictionaryData\":{\"id\":1,\"alAttributes\":[{\"$$hashKey\":\"object:1379\",\"id\":\"choice1\",\"option\":\"Group_Test\"}],\"attributes\":[{\"$$hashKey\":\"object:1374\",\"id\":\"choice1\",\"option\":\"Test\"}],\"description\":\"test\",\"parentItemName\":\"Test\"},\"userid\":\"demo\"}";
 472                 testUpdate(FirewallDictionaryList.class, "fwDictListDictionaryDatas", "parentItemName",
 473                                 () -> controller.saveFWDictionaryList(request, response));
 474         }
 475
 476         @Test
 477         public void testRemoveFWDictionaryList() {
 478                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"parentItemName\":\"Test\"}}";
 479                 testRemove(FirewallDictionaryList.class, "fwDictListDictionaryDatas",
 480                                 () -> controller.removeFWDictionaryList(request, response));
 481         }
 482
 483         @Test
 484         public void testSaveFirewallTagPickerDictionary() {
 485                 jsonString = "{\"fwTagPickerDictionaryData\":{\"description\":\"test\",\"networkRole\":\"test\",\"tagPickerName\":\"Test\",\"tags\":[{\"$$hashKey\":\"object:1855\",\"id\":\"choice1\",\"number\":\"test\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 486                 testSave(FWTagPicker.class, "fwTagPickerDictionaryDatas", "tagPickerName",
 487                                 () -> controller.saveFirewallTagPickerDictionary(request, response));
 488         }
 489
 490         @Test
 491         public void testUpdateFirewallTagPickerDictionary() {
 492                 jsonString = "{\"fwTagPickerDictionaryData\":{\"id\":1,\"description\":\"test\",\"networkRole\":\"test\",\"tagPickerName\":\"Test\",\"tags\":[{\"$$hashKey\":\"object:1855\",\"id\":\"choice1\",\"number\":\"test\",\"option\":\"Test\"}]},\"userid\":\"demo\"}";
 493                 testUpdate(FWTagPicker.class, "fwTagPickerDictionaryDatas", "tagPickerName",
 494                                 () -> controller.saveFirewallTagPickerDictionary(request, response));
 495         }
 496
 497         @Test
 498         public void testRemoveFirewallTagPickerDictionary() {
 499                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"tagPickerName\":\"Test\"}}";
 500                 testRemove(FWTagPicker.class, "fwTagPickerDictionaryDatas",
 501                                 () -> controller.removeFirewallTagPickerDictionary(request, response));
 502         }
 503
 504         @Test
 505         public void testSaveFirewallTagDictionary() {
 506                 jsonString = "{\"fwTagDictionaryData\":{\"description\":\"test\",\"fwTagName\":\"Test\",\"tags\":[{\"$$hashKey\":\"object:1690\",\"id\":\"choice1\",\"tags\":\"test\"}]},\"userid\":\"demo\"}";
 507                 testSave(FWTag.class, "fwTagDictionaryDatas", "fwTagName",
 508                                 () -> controller.saveFirewallTagDictionary(request, response));
 509         }
 510
 511         @Test
 512         public void testUpdateFirewallTagDictionary() {
 513                 jsonString = "{\"fwTagDictionaryData\":{\"id\":1,\"description\":\"test\",\"fwTagName\":\"Test\",\"tags\":[{\"$$hashKey\":\"object:1690\",\"id\":\"choice1\",\"tags\":\"test\"}]},\"userid\":\"demo\"}";
 514                 testUpdate(FWTag.class, "fwTagDictionaryDatas", "fwTagName",
 515                                 () -> controller.saveFirewallTagDictionary(request, response));
 516         }
 517
 518         @Test
 519         public void testRemoveFirewallTagDictionary() {
 520                 jsonString = "{\"userid\":\"demo\",\"data\":{\"id\":1,\"description\":\"test\",\"fwTagName\":\"Test\"}}";
 521                 testRemove(FWTag.class, "fwTagDictionaryDatas",
 522                                 () -> controller.removeFirewallTagDictionary(request, response));
 523         }
 524
 525         /**
 526          * Tests a "get" function that uses commonClassDao.getDataByColumn().
 527          *
 528          * @param clazz
 529          * @param contentData
 530          * @param contentName
 531          * @param func
 532          */
 533         private void test_WithGetDataByColumn(Class<?> clazz, String contentData, String contentName, VoidFunc func) {
 534                 when(commonClassDao.getDataByColumn(clazz, contentName)).thenReturn(data);
 535                 try {
 536                         func.apply();
 537                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 538                         verify(commonClassDao).getDataByColumn(clazz, contentName);
 539                 } catch (Exception e) {
 540                         fail("get " + clazz.getName() + e);
 541                 }
 542         }
 543
 544         /**
 545          * Tests a "get" function that uses commonClassDao.getData().
 546          *
 547          * @param clazz
 548          * @param contentData
 549          * @param contentName
 550          * @param func
 551          */
 552         private void test_WithGetData(Class<?> clazz, String contentData, VoidFunc func) {
 553                 when(commonClassDao.getData(clazz)).thenReturn(new ArrayList<>());
 554                 try {
 555                         func.apply();
 556                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 557                         verify(commonClassDao).getData(clazz);
 558                 } catch (Exception e) {
 559                         fail("get " + clazz.getName() + e);
 560                 }
 561         }
 562
 563         /**
 564          * Tests a function that uses commonClassDao.save().
 565          *
 566          * @param clazz
 567          * @param contentData
 568          * @param contentName
 569          * @param func
 570          */
 571         private void testSave(Class<?> clazz, String contentData, String contentName, VoidFunc func) {
 572                 testSave(clazz, contentData, contentName, "Test", func);
 573         }
 574
 575         /**
 576          * Tests a function that uses commonClassDao.save().
 577          *
 578          * @param clazz
 579          * @param contentData
 580          * @param contentName
 581          * @param testName
 582          * @param func
 583          */
 584         private void testSave(Class<?> clazz, String contentData, String contentName, String testName, VoidFunc func) {
 585                 try (BufferedReader br = new BufferedReader(new StringReader(jsonString))) {
 586                         when(request.getReader()).thenReturn(br);
 587                         func.apply();
 588                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 589                         verify(commonClassDao).checkDuplicateEntry(testName, contentName, clazz);
 590                         verify(commonClassDao).save(any());
 591                         verify(commonClassDao, never()).update(any());
 592                         verify(commonClassDao).getData(clazz);
 593
 594                 } catch (IOException e) {
 595                         fail("save " + clazz.getName() + e);
 596                 }
 597         }
 598
 599         /**
 600          * Tests a function that uses commonClassDao.update().
 601          *
 602          * @param clazz
 603          * @param contentData
 604          * @param contentName
 605          * @param func
 606          */
 607         private void testUpdate(Class<?> clazz, String contentData, String contentName, VoidFunc func) {
 608                 testUpdate(clazz, contentData, contentName, "Test", func);
 609         }
 610
 611         /**
 612          * Tests a function that uses commonClassDao.update().
 613          *
 614          * @param clazz
 615          * @param contentData
 616          * @param contentName
 617          * @param testName
 618          * @param func
 619          */
 620         private void testUpdate(Class<?> clazz, String contentData, String contentName, String testName, VoidFunc func) {
 621                 try (BufferedReader br = new BufferedReader(new StringReader(jsonString))) {
 622                         when(request.getReader()).thenReturn(br);
 623                         func.apply();
 624                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 625                         verify(commonClassDao).checkDuplicateEntry(testName, contentName, clazz);
 626                         verify(commonClassDao, never()).save(any());
 627                         verify(commonClassDao).update(any());
 628                         verify(commonClassDao).getData(clazz);
 629
 630                 } catch (IOException e) {
 631                         fail("update " + clazz.getName() + e);
 632                 }
 633         }
 634
 635         /**
 636          * Tests a function that uses commonClassDao.delete() and
 637          * commonClassDao.getData().
 638          *
 639          * @param clazz
 640          * @param contentData
 641          * @param func
 642          */
 643         private void testRemove(Class<?> clazz, String contentData, VoidFunc func) {
 644                 try (BufferedReader br = new BufferedReader(new StringReader(jsonString))) {
 645                         when(request.getReader()).thenReturn(br);
 646                         func.apply();
 647                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 648                         verify(commonClassDao).delete(any());
 649                         verify(commonClassDao).getData(clazz);
 650
 651                 } catch (IOException e) {
 652                         fail("remove " + clazz.getName() + e);
 653                 }
 654         }
 655
 656         /**
 657          * Tests a "validate" function.
 658          *
 659          * @param clazz
 660          * @param contentData
 661          * @param func
 662          */
 663         private void testValidate(Class<?> clazz, String contentData, VoidFunc func) {
 664                 try (BufferedReader br = new BufferedReader(new StringReader(jsonString))) {
 665                         when(request.getReader()).thenReturn(br);
 666                         func.apply();
 667                         assertTrue(response.getContentAsString() != null && response.getContentAsString().contains(contentData));
 668
 669                 } catch (IOException e) {
 670                         fail("save " + clazz.getName() + e);
 671                 }
 672         }
 673
 674         @FunctionalInterface
 675         private static interface VoidFunc {
 676                 public void apply() throws IOException;
 677         }
 678 }