diff options
| -rw-r--r-- | datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/ThrottleFilterTest.java | 218 |
1 files changed, 218 insertions, 0 deletions
diff --git a/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/ThrottleFilterTest.java b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/ThrottleFilterTest.java new file mode 100644 index 00000000..a8e919b7 --- /dev/null +++ b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/ThrottleFilterTest.java @@ -0,0 +1,218 @@ +/******************************************************************************* + * ============LICENSE_START================================================== + * * org.onap.dmaap + * * =========================================================================== + * * Copyright © 2017 AT&T Intellectual Property. All rights reserved. + * * =========================================================================== + * * Licensed under the Apache License, Version 2.0 (the "License"); + * * you may not use this file except in compliance with the License. + * * You may obtain a copy of the License at + * * + * * http://www.apache.org/licenses/LICENSE-2.0 + * * + * * Unless required by applicable law or agreed to in writing, software + * * distributed under the License is distributed on an "AS IS" BASIS, + * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * * See the License for the specific language governing permissions and + * * limitations under the License. + * * ============LICENSE_END==================================================== + * * + * * ECOMP is a trademark and service mark of AT&T Intellectual Property. + * * + ******************************************************************************/ + +package org.onap.dmaap.datarouter.provisioning.utils; + +import org.apache.commons.lang3.reflect.FieldUtils; +import org.eclipse.jetty.continuation.Continuation; +import org.eclipse.jetty.continuation.ContinuationSupport; +import org.eclipse.jetty.io.EndPoint; +import org.eclipse.jetty.server.HttpChannel; +import org.eclipse.jetty.server.HttpConnection; +import org.eclipse.jetty.server.Request; +import org.junit.Test; + +import org.junit.runner.RunWith; +import static org.junit.Assert.*; +import static org.hamcrest.core.Is.is; +import org.mockito.Mock; + +import static org.mockito.Matchers.*; +import static org.mockito.Mockito.*; + +import org.onap.dmaap.datarouter.provisioning.beans.Parameters; +import org.powermock.api.mockito.PowerMockito; +import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor; +import org.powermock.modules.junit4.PowerMockRunner; + +import javax.servlet.FilterChain; +import javax.servlet.FilterConfig; +import javax.servlet.ServletInputStream; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.util.*; + +@RunWith(PowerMockRunner.class) +@SuppressStaticInitializationFor({"org.onap.dmaap.datarouter.provisioning.beans.Parameters", + "org.eclipse.jetty.server.Request", + "org.eclipse.jetty.continuation.ContinuationSupport", + "org.eclipse.jetty.server.HttpConnection"}) +public class ThrottleFilterTest { + + @Mock + private HttpServletRequest request; + + @Mock + private HttpServletResponse response; + + @Mock + private FilterChain filterchain; + + @Mock + private HttpConnection httpconnection; + + @Mock + private ContinuationSupport continuationsupport; + + @Mock + private Request req; + + @Mock + private HttpChannel httpchannel; + + @Mock + private Continuation continuation; + + + + @Test + public void Given_Throttle_Filter_Configure_And_Parameter_Is_Not_Null_Then_Enabled_And_Action_Is_True() throws Exception { + mockParametersWithValues("2,5,throttle"); + ThrottleFilter.configure(); + boolean enabled = (boolean) FieldUtils.readStaticField(ThrottleFilter.class, "enabled", true); + int action = (int) FieldUtils.readStaticField(ThrottleFilter.class, "action", true); + assertThat(enabled, is(true)); + assertThat(action, is(0)); + + } + + @Test + public void Given_Do_Filter_run_and_enabled_and_action_is_true_and_rate_is_0_then_continiuation_will_call_setAttribute_and_resume_once() throws Exception { + mockParametersWithValues("100,5,thing"); + ThrottleFilter throttlefilter = new ThrottleFilter(); + ThrottleFilter.configure(); + mockServletInputStream(); + FieldUtils.writeDeclaredStaticField(ThrottleFilter.class, "action", 1, true); + Map<String, List<Continuation>> suspended_requests = new HashMap<String, List<Continuation>>(); + List<Continuation> continuation_list = new ArrayList<>(); + continuation_list.add(continuation); + suspended_requests.put("null/-1", continuation_list); + FieldUtils.writeDeclaredField(throttlefilter, "suspended_requests", suspended_requests, true); + throttlefilter.doFilter(request, response, filterchain); + verify(continuation, times(1)).setAttribute(anyString(), any()); + verify(continuation, times(1)).resume(); + } + + @Test + public void Given_Do_Filter_Run_and_enabled_and_action_is_true_and_rate_is_greater_than_0_then_continuation_will_call_suspend_and_dispatch_once() throws Exception { + mockParametersWithValues("0,5,thing"); + mockContinuationSupport(); + ThrottleFilter.configure(); + mockServletInputStream(); + FieldUtils.writeDeclaredStaticField(ThrottleFilter.class, "action", 1, true); + ThrottleFilter throttlefilter = new ThrottleFilter(); + throttlefilter.doFilter(request, response, filterchain); + verify(continuation, times(1)).undispatch(); + verify(continuation, times(1)).suspend(); + } + + + @Test + public void Given_Do_Filter_Run_and_enabled_and_action_is_true_and_rate_is_greater_than_0_and_getFeedId_returns_id_then_continuation_will_call_suspend_and_dispatch_once() throws Exception { + mockParametersWithValues("0,5,thing"); + PowerMockito.mockStatic(ContinuationSupport.class); + PowerMockito.when(continuationsupport.getContinuation(any())).thenReturn(continuation); + ThrottleFilter.configure(); + mockServletInputStream(); + FieldUtils.writeDeclaredStaticField(ThrottleFilter.class, "action", 1, true); + ThrottleFilter throttlefilter = new ThrottleFilter(); + when(request.getPathInfo()).thenReturn("/123/fileName.txt"); + throttlefilter.doFilter(request, response, filterchain); + verify(continuation, times(1)).undispatch(); + verify(continuation, times(1)).suspend(); + } + + @Test + public void Given_Do_Filter_and_only_enabled_is_true_and_drop_filter_ran_then_request_will_call_getHttpChannel_and_HttpChannel_will_call_getEndPoint_once() throws Exception { + mockParametersWithValues("0,5,thing"); + ServletInputStream serverinputstream = mock(ServletInputStream.class); + mockHttpConnectionHttpChannelAndRequest(serverinputstream); + ThrottleFilter.configure(); + FieldUtils.writeDeclaredStaticField(ThrottleFilter.class, "action", 0, true); + ThrottleFilter throttlefilter = new ThrottleFilter(); + throttlefilter.doFilter(request, response, filterchain); + verify(req, times(1)).getHttpChannel(); + verify(httpchannel, times(1)).getEndPoint(); + } + + @Test + public void Given_run_is_called_then_continuation_will_call_prune_once() throws Exception { + ThrottleFilter tf = new ThrottleFilter(); + Map<String, ThrottleFilter.Counter> map = new HashMap<String, ThrottleFilter.Counter>(); + ThrottleFilter.Counter tfc = mock(ThrottleFilter.Counter.class); + map.put("Key", tfc); + when(tfc.prune()).thenReturn(-1); + FieldUtils.writeDeclaredField(tf, "map", map, true); + tf.run(); + verify(tfc, times(1)).prune(); + } + + @Test + public void Given_destroy_is_called_then_map_is_empty() throws Exception + { + ThrottleFilter throttleFilter = new ThrottleFilter(); + FilterConfig filterconfig = mock(FilterConfig.class); + mockParametersWithValues("0,5,thing"); + PowerMockito.mockStatic(ContinuationSupport.class); + PowerMockito.when(continuationsupport.getContinuation(any())).thenReturn(continuation); + + throttleFilter.init(filterconfig); + throttleFilter.destroy(); + } + + private Parameters getParameters() { + PowerMockito.mockStatic(Parameters.class); + return mock(Parameters.class); + } + + private void mockServletInputStream() throws IOException { + ServletInputStream serverinputstream = mock(ServletInputStream.class); + when(serverinputstream.read(any())).thenReturn(2).thenReturn(1).thenReturn(0); + when(request.getInputStream()).thenReturn(serverinputstream); + } + + private void mockParametersWithValues(String values) { + Parameters parameters = getParameters(); + PowerMockito.when(parameters.getParameter(anyString())).thenReturn(new Parameters("key", values)); + } + + private void mockContinuationSupport() { + PowerMockito.mockStatic(ContinuationSupport.class); + PowerMockito.when(continuationsupport.getContinuation(any())).thenReturn(continuation); + } + + private void mockHttpConnectionHttpChannelAndRequest(ServletInputStream serverinputstream) throws IOException { + PowerMockito.mockStatic(ContinuationSupport.class); + PowerMockito.when(continuationsupport.getContinuation(any())).thenReturn(continuation); + when(serverinputstream.read(any())).thenReturn(2).thenReturn(1).thenReturn(0); + when(request.getInputStream()).thenReturn(serverinputstream); + PowerMockito.mockStatic(HttpConnection.class); + EndPoint endpoint = mock(EndPoint.class); + PowerMockito.when(httpconnection.getCurrentConnection()).thenReturn(httpconnection); + PowerMockito.when(httpconnection.getHttpChannel()).thenReturn(httpchannel); + when(httpchannel.getRequest()).thenReturn(req); + when(req.getHttpChannel()).thenReturn(httpchannel); + when(httpchannel.getEndPoint()).thenReturn(endpoint); + } +} |