use of com.yahoo.jdisc.http.filter.RequestFilter in project vespa by vespa-engine.
the class JDiscFilterForServletTest method bindings.
private Module bindings(BindingRepository<RequestFilter> requestFilters, BindingRepository<ResponseFilter> responseFilters) {
return Modules.combine(new AbstractModule() {
@Override
protected void configure() {
bind(FilterBindings.class).toInstance(new FilterBindings(requestFilters, responseFilters));
bind(FilterInvoker.class).toInstance(new FilterInvoker() {
@Override
public HttpServletRequest invokeRequestFilterChain(RequestFilter requestFilter, URI uri, HttpServletRequest httpRequest, ResponseHandler responseHandler) {
TestRequestFilter filter = (TestRequestFilter) requestFilter;
filter.runAsSecurityFilter(httpRequest, responseHandler);
return httpRequest;
}
@Override
public void invokeResponseFilterChain(ResponseFilter responseFilter, URI uri, HttpServletRequest request, HttpServletResponse response) {
TestResponseFilter filter = (TestResponseFilter) responseFilter;
filter.runAsSecurityFilter(request, response);
}
});
}
}, guiceModule());
}
use of com.yahoo.jdisc.http.filter.RequestFilter in project vespa by vespa-engine.
the class FilterTestCase method requireThatResponseFilterIsRunOnBoundPath.
@Test
public void requireThatResponseFilterIsRunOnBoundPath() throws Exception {
final ResponseFilter filter = mock(ResponseFilterMockBase.class);
final BindingRepository<RequestFilter> requestFilters = null;
final BindingRepository<ResponseFilter> responseFilters = new BindingRepository<>();
responseFilters.bind("http://*/filtered/*", filter);
final MyRequestHandler requestHandler = new MyRequestHandler();
final TestDriver testDriver = newDriver(requestHandler, requestFilters, responseFilters);
testDriver.client().get("/filtered/status.html");
assertThat(requestHandler.awaitInvocation(), is(true));
verify(filter, times(1)).filter(any(Response.class), any(Request.class));
assertThat(testDriver.close(), is(true));
}
use of com.yahoo.jdisc.http.filter.RequestFilter in project vespa by vespa-engine.
the class FilterTestCase method requireThatRequestFilterCanTerminateChain.
@Test
public void requireThatRequestFilterCanTerminateChain() throws Exception {
final RequestFilter requestFilter1 = new RespondForbiddenFilter();
final RequestFilter requestFilter2 = mock(RequestFilter.class);
final RequestFilter requestFilterChain = RequestFilterChain.newInstance(requestFilter1, requestFilter2);
final HttpRequest request = null;
final ResponseHandler responseHandler = mock(ResponseHandler.class);
when(responseHandler.handleResponse(any(Response.class))).thenReturn(mock(ContentChannel.class));
requestFilterChain.filter(request, responseHandler);
verify(requestFilter2, never()).filter(any(HttpRequest.class), any(ResponseHandler.class));
final ArgumentCaptor<Response> responseCaptor = ArgumentCaptor.forClass(Response.class);
verify(responseHandler).handleResponse(responseCaptor.capture());
assertThat(responseCaptor.getValue().getStatus(), is(Response.Status.FORBIDDEN));
}
use of com.yahoo.jdisc.http.filter.RequestFilter in project vespa by vespa-engine.
the class FilterTestCase method requireThatRequestFilterIsRunOnBoundPath.
@Test
public void requireThatRequestFilterIsRunOnBoundPath() throws Exception {
final RequestFilter filter = mock(RequestFilterMockBase.class);
final BindingRepository<RequestFilter> requestFilters = new BindingRepository<>();
requestFilters.bind("http://*/filtered/*", filter);
final BindingRepository<ResponseFilter> responseFilters = null;
final MyRequestHandler requestHandler = new MyRequestHandler();
final TestDriver testDriver = newDriver(requestHandler, requestFilters, responseFilters);
testDriver.client().get("/filtered/status.html");
assertThat(requestHandler.awaitInvocation(), is(true));
verify(filter, times(1)).filter(any(HttpRequest.class), any(ResponseHandler.class));
assertThat(testDriver.close(), is(true));
}
use of com.yahoo.jdisc.http.filter.RequestFilter in project vespa by vespa-engine.
the class FilterTestCase method requireThatRequestFilterChainCallsFilterWithOriginalResponseHandler.
@Test
public void requireThatRequestFilterChainCallsFilterWithOriginalResponseHandler() throws Exception {
final RequestFilter requestFilter = mock(RequestFilter.class);
final RequestFilter requestFilterChain = RequestFilterChain.newInstance(requestFilter);
final HttpRequest request = null;
final ResponseHandler responseHandler = mock(ResponseHandler.class);
requestFilterChain.filter(request, responseHandler);
// Check that the filter is called with the same response handler argument as the chain was,
// in a manner that allows the handler object to be wrapped.
final ArgumentCaptor<ResponseHandler> responseHandlerCaptor = ArgumentCaptor.forClass(ResponseHandler.class);
verify(requestFilter).filter(any(HttpRequest.class), responseHandlerCaptor.capture());
verify(responseHandler, never()).handleResponse(any(Response.class));
responseHandlerCaptor.getValue().handleResponse(mock(Response.class));
verify(responseHandler, times(1)).handleResponse(any(Response.class));
}
Aggregations