2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2020 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.sdc.be.filters;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.junit.runner.RunWith;
26 import org.mockito.InjectMocks;
27 import org.mockito.Mock;
28 import org.mockito.Mockito;
29 import org.mockito.MockitoAnnotations;
30 import org.mockito.Spy;
31 import org.mockito.junit.MockitoJUnitRunner;
32 import org.openecomp.sdc.be.components.impl.ResponseFormatManager;
33 import org.openecomp.sdc.be.config.Configuration;
34 import org.openecomp.sdc.be.config.ConfigurationManager;
35 import org.openecomp.sdc.be.servlets.exception.ComponentExceptionMapper;
36 import org.openecomp.sdc.common.api.ConfigurationSource;
37 import org.openecomp.sdc.common.api.FilterDecisionEnum;
38 import org.openecomp.sdc.common.impl.ExternalConfiguration;
39 import org.openecomp.sdc.common.impl.FSConfigurationSource;
40 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
42 import javax.servlet.FilterChain;
43 import javax.servlet.ServletException;
44 import javax.servlet.http.Cookie;
45 import javax.servlet.http.HttpServletRequest;
46 import javax.servlet.http.HttpServletResponse;
47 import javax.ws.rs.core.HttpHeaders;
48 import java.io.IOException;
49 import java.util.Arrays;
50 import java.util.Collections;
51 import java.util.Enumeration;
52 import java.util.List;
54 import static org.junit.Assert.assertNotNull;
55 import static org.junit.Assert.assertNull;
56 import static org.junit.Assert.assertTrue;
57 import static org.mockito.Mockito.any;
58 import static org.mockito.Mockito.doNothing;
59 import static org.mockito.Mockito.doThrow;
60 import static org.mockito.Mockito.eq;
61 import static org.mockito.Mockito.times;
62 import static org.mockito.Mockito.when;
65 @RunWith(MockitoJUnitRunner.class)
66 public class GatewayFilterTest {
68 private static final List<String> excludedUrls = Arrays.asList("test1", "test2");
69 private static final String cookieName = "myCookie";
71 static ResponseFormatManager responseFormatManager = new ResponseFormatManager();
72 static ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), "src/test/resources/config/catalog-be");
73 static ConfigurationManager configurationManager = new ConfigurationManager(configurationSource);
76 private GatewayFilter filter;
78 private ThreadLocalUtils threadLocalUtils;
80 private Configuration.CookieConfig authCookieConf;
82 private Configuration configuration;
84 private HttpServletRequest request;
86 private FilterChain filterChain;
88 private HttpServletResponse response;
90 private ComponentExceptionMapper componentExceptionMapper;
95 public void initMocks(){
96 MockitoAnnotations.openMocks(this);
100 public void setUp() throws ServletException {
101 doNothing().when(threadLocalUtils).setUserContextFromDB(request);
102 when(configuration.getAuthCookie()).thenReturn(authCookieConf);
103 this.filter = new GatewayFilter(configuration);
104 ThreadLocalsHolder.setApiType(null);
105 assertNotNull(filter);
109 public void validateRequestFromWhiteList() throws ServletException, IOException {
110 when(authCookieConf.getExcludedUrls()).thenReturn(excludedUrls);
111 when(request.getPathInfo()).thenReturn("test1");
112 filter.doFilter(request, response, filterChain);
113 assertTrue(ThreadLocalsHolder.getApiType().equals(FilterDecisionEnum.NA));
114 Mockito.verify(filterChain, times(1)).doFilter(request, response);
117 private Enumeration getHeaderEnumerationObj(List<String> arrlist){
119 // creating object of type Enumeration<String>
120 Enumeration<String> enumer = Collections.enumeration(arrlist);
129 private Cookie[] getCookiesFromReq(boolean isFromRequest) {
130 Cookie[] cookies = new Cookie [1];
132 cookies[0] = new Cookie(cookieName, "cookieData");
135 cookies[0] = new Cookie("dummy", "cookieData");
140 private String getCookieNameFromConf(boolean isFromConfiguration) {
141 Cookie[] cookies = new Cookie [1];
142 if (isFromConfiguration) {
143 cookies[0] = new Cookie(cookieName, "cookieData");
146 cookies[0] = new Cookie("dummy", "cookieData");
148 return cookies[0].getName();