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.components.impl.exceptions.ByActionStatusComponentException;
34 import org.openecomp.sdc.be.components.impl.exceptions.ComponentException;
35 import org.openecomp.sdc.be.config.Configuration;
36 import org.openecomp.sdc.be.config.ConfigurationManager;
37 import org.openecomp.sdc.be.servlets.exception.ComponentExceptionMapper;
38 import org.openecomp.sdc.common.api.ConfigurationSource;
39 import org.openecomp.sdc.common.api.FilterDecisionEnum;
40 import org.openecomp.sdc.common.impl.ExternalConfiguration;
41 import org.openecomp.sdc.common.impl.FSConfigurationSource;
42 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
44 import javax.servlet.FilterChain;
45 import javax.servlet.ServletException;
46 import javax.servlet.http.Cookie;
47 import javax.servlet.http.HttpServletRequest;
48 import javax.servlet.http.HttpServletResponse;
49 import javax.ws.rs.core.HttpHeaders;
50 import java.io.IOException;
51 import java.util.Arrays;
52 import java.util.Collections;
53 import java.util.Enumeration;
54 import java.util.List;
56 import static org.junit.Assert.assertNotNull;
57 import static org.junit.Assert.assertNull;
58 import static org.junit.Assert.assertTrue;
59 import static org.mockito.Mockito.any;
60 import static org.mockito.Mockito.doNothing;
61 import static org.mockito.Mockito.doThrow;
62 import static org.mockito.Mockito.eq;
63 import static org.mockito.Mockito.times;
64 import static org.mockito.Mockito.when;
67 @RunWith(MockitoJUnitRunner.class)
68 public class GatewayFilterTest {
70 private static final List<String> excludedUrls = Arrays.asList("test1", "test2");
71 private static final String cookieName = "myCookie";
73 static ResponseFormatManager responseFormatManager = new ResponseFormatManager();
74 static ConfigurationSource configurationSource = new FSConfigurationSource(ExternalConfiguration.getChangeListener(), "src/test/resources/config/catalog-be");
75 static ConfigurationManager configurationManager = new ConfigurationManager(configurationSource);
78 private GatewayFilter filter;
80 private ThreadLocalUtils threadLocalUtils;
82 private Configuration.CookieConfig authCookieConf;
84 private Configuration configuration;
86 private HttpServletRequest request;
88 private FilterChain filterChain;
90 private HttpServletResponse response;
92 private ComponentExceptionMapper componentExceptionMapper;
97 public void initMocks(){
98 MockitoAnnotations.initMocks(this);
102 public void setUp() throws ServletException {
103 doNothing().when(threadLocalUtils).setUserContextFromDB(request);
104 when(configuration.getAuthCookie()).thenReturn(authCookieConf);
105 this.filter = new GatewayFilter(configuration);
106 ThreadLocalsHolder.setApiType(null);
107 assertNotNull(filter);
111 public void validateRequestFromWhiteList() throws ServletException, IOException {
112 when(authCookieConf.getExcludedUrls()).thenReturn(excludedUrls);
113 when(request.getPathInfo()).thenReturn("test1");
114 filter.doFilter(request, response, filterChain);
115 assertTrue(ThreadLocalsHolder.getApiType().equals(FilterDecisionEnum.NA));
116 Mockito.verify(filterChain, times(1)).doFilter(request, response);
119 private Enumeration getHeaderEnumerationObj(List<String> arrlist){
121 // creating object of type Enumeration<String>
122 Enumeration<String> enumer = Collections.enumeration(arrlist);
131 private Cookie[] getCookiesFromReq(boolean isFromRequest) {
132 Cookie[] cookies = new Cookie [1];
134 cookies[0] = new Cookie(cookieName, "cookieData");
137 cookies[0] = new Cookie("dummy", "cookieData");
142 private String getCookieNameFromConf(boolean isFromConfiguration) {
143 Cookie[] cookies = new Cookie [1];
144 if (isFromConfiguration) {
145 cookies[0] = new Cookie(cookieName, "cookieData");
148 cookies[0] = new Cookie("dummy", "cookieData");
150 return cookies[0].getName();