/* * Copyright 2002-2017 the original author or authors. * * 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 * * https://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. */package org.springframework.mock.web;import java.io.IOException;import java.io.UnsupportedEncodingException;import java.util.Collections;import java.util.Enumeration;import java.util.LinkedHashMap;import java.util.LinkedHashSet;import java.util.Map;import javax.el.ELContext;import javax.servlet.Servlet;import javax.servlet.ServletConfig;import javax.servlet.ServletContext;import javax.servlet.ServletException;import javax.servlet.ServletRequest;import javax.servlet.ServletResponse;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import javax.servlet.http.HttpSession;import javax.servlet.jsp.JspWriter;import javax.servlet.jsp.PageContext;import org.springframework.util.Assert;/** * Mock implementation of the {@link javax.servlet.jsp.PageContext} interface. * Only necessary for testing applications when testing custom JSP tags. * * <p>Note: Expects initialization via the constructor rather than via the * {@code PageContext.initialize} method. Does not support writing to a * JspWriter, request dispatching, or {@code handlePageException} calls. * * @author Juergen Hoeller * @since 1.0.2 */public class MockPageContext extends PageContext { private final ServletContext servletContext; private final HttpServletRequest request; private final HttpServletResponse response; private final ServletConfig servletConfig; private final Map<String, Object> attributes = new LinkedHashMap<String, Object>(); private JspWriter out; /** * Create new MockPageContext with a default {@link MockServletContext}, * {@link MockHttpServletRequest}, {@link MockHttpServletResponse}, * {@link MockServletConfig}. */ public MockPageContext() { this(null, null, null, null); } /** * Create new MockPageContext with a default {@link MockHttpServletRequest}, * {@link MockHttpServletResponse}, {@link MockServletConfig}. * @param servletContext the ServletContext that the JSP page runs in * (only necessary when actually accessing the ServletContext) */ public MockPageContext(ServletContext servletContext) { this(servletContext, null, null, null); } /** * Create new MockPageContext with a MockHttpServletResponse, * MockServletConfig. * @param servletContext the ServletContext that the JSP page runs in * @param request the current HttpServletRequest * (only necessary when actually accessing the request) */ public MockPageContext(ServletContext servletContext, HttpServletRequest request) { this(servletContext, request, null, null); } /** * Create new MockPageContext with a MockServletConfig. * @param servletContext the ServletContext that the JSP page runs in * @param request the current HttpServletRequest * @param response the current HttpServletResponse * (only necessary when actually writing to the response) */ public MockPageContext(ServletContext servletContext, HttpServletRequest request, HttpServletResponse response) { this(servletContext, request, response, null); } /** * Create new MockServletConfig. * @param servletContext the ServletContext that the JSP page runs in * @param request the current HttpServletRequest * @param response the current HttpServletResponse * @param servletConfig the ServletConfig (hardly ever accessed from within a tag) */ public MockPageContext(ServletContext servletContext, HttpServletRequest request, HttpServletResponse response, ServletConfig servletConfig) { this.servletContext = (servletContext != null ? servletContext : new MockServletContext()); this.request = (request != null ? request : new MockHttpServletRequest(servletContext)); this.response = (response != null ? response : new MockHttpServletResponse()); this.servletConfig = (servletConfig != null ? servletConfig : new MockServletConfig(servletContext)); } @Override public void initialize( Servlet servlet, ServletRequest request, ServletResponse response, String errorPageURL, boolean needsSession, int bufferSize, boolean autoFlush) { throw new UnsupportedOperationException("Use appropriate constructor"); } @Override public void release() { } @Override public void setAttribute(String name, Object value) { Assert.notNull(name, "Attribute name must not be null"); if (value != null) { this.attributes.put(name, value); } else { this.attributes.remove(name); } } @Override public void setAttribute(String name, Object value, int scope) { Assert.notNull(name, "Attribute name must not be null"); switch (scope) { case PAGE_SCOPE: setAttribute(name, value); break; case REQUEST_SCOPE: this.request.setAttribute(name, value); break;