@Before public void setUp() throws ServletException, ServletInitializationException { servletContainer = mock(ServletContainer.class); when(servletContainer.getServletForClass(any(Class.class), any(ServletConfig.class))). thenReturn(mock(Servlet.class)); servletContext = mock(ServletContextImpl.class); when(servletContext.getContextPath()).thenReturn("/"); ServletMapping servletMapping = new ServletMappingImpl(Pattern.compile("^.*$"), SampleServlet.class); when(servletContext.getServletMappings()).thenReturn(Arrays.asList(servletMapping)); when(servletContext.getFilterMappings()).thenReturn(Collections.<FilterMapping>emptyList()); servletResourceProvider = new ServletResourceProvider( servletContainer, Arrays.asList(servletContext) ); response = new HttpServletResponseImpl(mock( Serializer.class), mock(Serializer.class), mock(StreamHelper.class), mock(OutputStream.class)); request = mock(HttpServletRequestImpl.class); when(request.getServletContext()).thenReturn(servletContext); }
/** * Terminates servlet. Sets all necessary headers, flushes content. * * @param request * @param response * @throws IOException */ private void terminate(final HttpServletRequestImpl request, final HttpServletResponseImpl response) throws IOException { freeUploadedUnprocessedFiles(request.getUploadedFiles()); HttpSessionImpl session = (HttpSessionImpl) request.getSession(false); if (session != null) { try { ((ServletContextImpl) request.getServletContext()).handleSession(session, response); } catch (IOException e) { LOGGER.log(Level.WARNING, "Unable to persist session", e); } } if (!response.isCommitted()) { if (response.getContentType() == null) { response.setContentType(DEFAULT_RESPONSE_CONTENT_TYPE); } response.getHeaders().setHeader(Headers.HEADER_CACHE_CONTROL, HEADER_VALUE_NO_CACHE); response.getHeaders().setHeader(Headers.HEADER_PRAGMA, HEADER_VALUE_NO_CACHE); } response.flush(); }