src/test/java/com/att/nsa/dmaap/util/DMaaPAuthFilterTest.java

   1 /*-\r
   2  * ============LICENSE_START=======================================================\r
   3  * ONAP Policy Engine\r
   4  * ================================================================================\r
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * ================================================================================\r
   7  * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * you may not use this file except in compliance with the License.\r
   9  * You may obtain a copy of the License at\r
  10  * \r
  11  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * \r
  13  * Unless required by applicable law or agreed to in writing, software\r
  14  * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * See the License for the specific language governing permissions and\r
  17  * limitations under the License.\r
  18  * ============LICENSE_END=========================================================\r
  19  */\r
  20 \r
  21 package com.att.nsa.dmaap.util;\r
  22 \r
  23 import java.io.IOException;\r
  24 \r
  25 import javax.servlet.FilterChain;\r
  26 import javax.servlet.ServletException;\r
  27 import javax.servlet.ServletResponse;\r
  28 import javax.servlet.http.HttpServletRequest;\r
  29 \r
  30 import org.junit.After;\r
  31 import org.junit.Before;\r
  32 import org.junit.Test;\r
  33 import org.junit.runner.RunWith;\r
  34 import org.mockito.InjectMocks;\r
  35 import org.mockito.Mock;\r
  36 import org.mockito.MockitoAnnotations;\r
  37 import org.powermock.api.mockito.PowerMockito;\r
  38 import org.powermock.core.classloader.annotations.PrepareForTest;\r
  39 import org.powermock.modules.junit4.PowerMockRunner;\r
  40 \r
  41 import com.att.ajsc.beans.PropertiesMapBean;\r
  42 import com.att.nsa.cambria.beans.DMaaPContext;\r
  43 import com.att.nsa.cambria.exception.DMaaPResponseCode;\r
  44 \r
  45 @RunWith(PowerMockRunner.class)\r
  46 @PrepareForTest({ PropertiesMapBean.class, DMaaPResponseCode.class })\r
  47 public class DMaaPAuthFilterTest {\r
  48 \r
  49         @InjectMocks\r
  50         DMaaPAuthFilter filter;\r
  51 \r
  52         @Mock\r
  53         HttpServletRequest req;\r
  54 \r
  55         @Mock\r
  56         ServletResponse res;\r
  57 \r
  58         @Mock\r
  59         FilterChain chain;\r
  60 \r
  61         @Mock\r
  62         DMaaPContext dmaapContext;\r
  63 \r
  64         @Before\r
  65         public void setUp() throws Exception {\r
  66                 MockitoAnnotations.initMocks(this);\r
  67         }\r
  68 \r
  69         @After\r
  70         public void tearDown() throws Exception {\r
  71         }\r
  72 \r
  73         @Test\r
  74         public void testDoFilter() throws IOException, ServletException {\r
  75 \r
  76                 PowerMockito.when(dmaapContext.getRequest()).thenReturn(req);\r
  77                 PowerMockito.when(req.getHeader("Authorization")).thenReturn("Authorization");\r
  78                 // when(dmaapContext.getResponse()).thenReturn(res);\r
  79                 filter.doFilter(req, res, chain);\r
  80 \r
  81         }\r
  82 \r
  83         @Test\r
  84         public void testDoFilter_nullAuth() throws IOException, ServletException {\r
  85 \r
  86                 PowerMockito.when(dmaapContext.getRequest()).thenReturn(req);\r
  87                 PowerMockito.when(req.getHeader("Authorization")).thenReturn("Authorization");\r
  88 \r
  89                 // when(dmaapContext.getResponse()).thenReturn(res);\r
  90                 filter.doFilter(req, res, chain);\r
  91 \r
  92         }\r
  93 \r
  94 }