ecomp-portal-BE-os/src/test/java/org/onap/portalapp/util/SecurityXssValidatorTest.java

   1 /*-
   2  * ============LICENSE_START==========================================
   3  * ONAP Portal
   4  * ===================================================================
   5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * ===================================================================
   7  *
   8  * Unless otherwise specified, all software contained herein is licensed
   9  * under the Apache License, Version 2.0 (the "License");
  10  * you may not use this software 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  * Unless otherwise specified, all documentation contained herein is licensed
  22  * under the Creative Commons License, Attribution 4.0 Intl. (the "License");
  23  * you may not use this documentation except in compliance with the License.
  24  * You may obtain a copy of the License at
  25  *
  26  *             https://creativecommons.org/licenses/by/4.0/
  27  *
  28  * Unless required by applicable law or agreed to in writing, documentation
  29  * distributed under the License is distributed on an "AS IS" BASIS,
  30  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  31  * See the License for the specific language governing permissions and
  32  * limitations under the License.
  33  *
  34  * ============LICENSE_END============================================
  35  *
  36  *
  37  */
  38 package org.onap.portalapp.util;
  39
  40 import org.apache.commons.lang.StringUtils;
  41 import org.junit.Assert;
  42 import org.junit.Test;
  43 import org.junit.runner.RunWith;
  44 import org.mockito.Mockito;
  45 import org.onap.portalsdk.core.util.SystemProperties;
  46 import org.owasp.esapi.ESAPI;
  47 import org.owasp.esapi.Encoder;
  48 import org.owasp.esapi.codecs.Codec;
  49 import org.powermock.api.mockito.PowerMockito;
  50 import org.powermock.core.classloader.annotations.PrepareForTest;
  51 import org.powermock.modules.junit4.PowerMockRunner;
  52
  53 @RunWith(PowerMockRunner.class)
  54 @PrepareForTest({ESAPI.class, SystemProperties.class})
  55 public class SecurityXssValidatorTest {
  56
  57         @Test
  58         public void stripXSSTest() {
  59                 String value ="Test";
  60                 PowerMockito.mockStatic(ESAPI.class);
  61                 Encoder mockEncoder = Mockito.mock(Encoder.class);
  62                 Mockito.when(ESAPI.encoder()).thenReturn(mockEncoder);
  63                 Mockito.when(mockEncoder.canonicalize(value)).thenReturn(value);
  64                 SecurityXssValidator validator = SecurityXssValidator.getInstance();
  65                 String reponse = validator.stripXSS(value);
  66                 Assert.assertEquals(value, reponse);;
  67         }
  68
  69         @Test
  70         public void stripXSSExceptionTest() {
  71                 String value ="Test";
  72                 SecurityXssValidator validator = SecurityXssValidator.getInstance();
  73                 String reponse = validator.stripXSS(value);
  74                 Assert.assertEquals(value, reponse);;
  75         }
  76
  77         @Test
  78         public void denyXSSTest() {
  79                 String value ="<script>Test</script>";
  80                 PowerMockito.mockStatic(ESAPI.class);
  81                 Encoder mockEncoder = Mockito.mock(Encoder.class);
  82                 Mockito.when(ESAPI.encoder()).thenReturn(mockEncoder);
  83                 Mockito.when(mockEncoder.canonicalize(value)).thenReturn(value);
  84                 SecurityXssValidator validator = SecurityXssValidator.getInstance();
  85                 Boolean flag = validator.denyXSS(value);
  86                 Assert.assertTrue(flag);
  87         }
  88
  89         @Test
  90         public void denyXSSFalseTest() {
  91                 String value ="test";
  92                 PowerMockito.mockStatic(ESAPI.class);
  93                 Encoder mockEncoder = Mockito.mock(Encoder.class);
  94                 Mockito.when(ESAPI.encoder()).thenReturn(mockEncoder);
  95                 Mockito.when(mockEncoder.canonicalize(value)).thenReturn(value);
  96                 SecurityXssValidator validator = SecurityXssValidator.getInstance();
  97                 Boolean flag = validator.denyXSS(value);
  98                 Assert.assertFalse(flag);
  99         }
 100
 101         @Test
 102         public void getCodecMySqlTest() {
 103                 PowerMockito.mockStatic(SystemProperties.class);
 104                 Mockito.when(SystemProperties.getProperty(SystemProperties.DB_DRIVER)).thenReturn("mysql");
 105                 SecurityXssValidator validator = SecurityXssValidator.getInstance();
 106                 Codec codec = validator.getCodec();
 107                 Assert.assertNotNull(codec);
 108         }
 109
 110 }