Browse Source
Currently the getNamedDispatcher(String) method of MockServletContext always returns null. This poses a problem in certain testing scenarios since one would always expect at least a default Servlet to be present. This is specifically important for web application tests that involve the DefaultServletHttpRequestHandler which attempts to forward to the default Servlet after retrieving it by name. Furthermore, there is no way to register a named RequestDispatcher with the MockServletContext. This commit addresses these issues by introducing the following in MockServletContext. - a new defaultServletName property for configuring the name of the default Servlet, which defaults to "default" - named RequestDispatchers can be registered and unregistered - a MockRequestDispatcher is registered for the "default" Servlet automatically in the constructor - when the defaultServletName property is set to a new value the the current default RequestDispatcher is unregistered and replaced with a MockRequestDispatcher for the new defaultServletName Issue: SPR-9587pull/111/head
Sam Brannen
12 years ago
16 changed files with 1678 additions and 304 deletions
@ -0,0 +1,71 @@
@@ -0,0 +1,71 @@
|
||||
/* |
||||
* Copyright 2002-2009 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 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
package org.springframework.mock.web; |
||||
|
||||
import java.io.IOException; |
||||
import java.io.OutputStream; |
||||
import javax.servlet.ServletOutputStream; |
||||
|
||||
import org.springframework.util.Assert; |
||||
|
||||
/** |
||||
* Delegating implementation of {@link javax.servlet.ServletOutputStream}. |
||||
* |
||||
* <p>Used by {@link MockHttpServletResponse}; typically not directly |
||||
* used for testing application controllers. |
||||
* |
||||
* @author Juergen Hoeller |
||||
* @since 1.0.2 |
||||
* @see MockHttpServletResponse |
||||
*/ |
||||
public class DelegatingServletOutputStream extends ServletOutputStream { |
||||
|
||||
private final OutputStream targetStream; |
||||
|
||||
|
||||
/** |
||||
* Create a DelegatingServletOutputStream for the given target stream. |
||||
* @param targetStream the target stream (never <code>null</code>) |
||||
*/ |
||||
public DelegatingServletOutputStream(OutputStream targetStream) { |
||||
Assert.notNull(targetStream, "Target OutputStream must not be null"); |
||||
this.targetStream = targetStream; |
||||
} |
||||
|
||||
/** |
||||
* Return the underlying target stream (never <code>null</code>). |
||||
*/ |
||||
public final OutputStream getTargetStream() { |
||||
return this.targetStream; |
||||
} |
||||
|
||||
|
||||
public void write(int b) throws IOException { |
||||
this.targetStream.write(b); |
||||
} |
||||
|
||||
public void flush() throws IOException { |
||||
super.flush(); |
||||
this.targetStream.flush(); |
||||
} |
||||
|
||||
public void close() throws IOException { |
||||
super.close(); |
||||
this.targetStream.close(); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,96 @@
@@ -0,0 +1,96 @@
|
||||
/* |
||||
* Copyright 2002-2011 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 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
package org.springframework.mock.web; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.Collection; |
||||
import java.util.Collections; |
||||
import java.util.LinkedList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
import org.springframework.util.Assert; |
||||
import org.springframework.util.CollectionUtils; |
||||
|
||||
/** |
||||
* Internal helper class that serves as value holder for request headers. |
||||
* |
||||
* @author Juergen Hoeller |
||||
* @author Rick Evans |
||||
* @since 2.0.1 |
||||
*/ |
||||
class HeaderValueHolder { |
||||
|
||||
private final List<Object> values = new LinkedList<Object>(); |
||||
|
||||
|
||||
public void setValue(Object value) { |
||||
this.values.clear(); |
||||
this.values.add(value); |
||||
} |
||||
|
||||
public void addValue(Object value) { |
||||
this.values.add(value); |
||||
} |
||||
|
||||
public void addValues(Collection<?> values) { |
||||
this.values.addAll(values); |
||||
} |
||||
|
||||
public void addValueArray(Object values) { |
||||
CollectionUtils.mergeArrayIntoCollection(values, this.values); |
||||
} |
||||
|
||||
public List<Object> getValues() { |
||||
return Collections.unmodifiableList(this.values); |
||||
} |
||||
|
||||
public List<String> getStringValues() { |
||||
List<String> stringList = new ArrayList<String>(this.values.size()); |
||||
for (Object value : this.values) { |
||||
stringList.add(value.toString()); |
||||
} |
||||
return Collections.unmodifiableList(stringList); |
||||
} |
||||
|
||||
public Object getValue() { |
||||
return (!this.values.isEmpty() ? this.values.get(0) : null); |
||||
} |
||||
|
||||
public String getStringValue() { |
||||
return (!this.values.isEmpty() ? this.values.get(0).toString() : null); |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Find a HeaderValueHolder by name, ignoring casing. |
||||
* @param headers the Map of header names to HeaderValueHolders |
||||
* @param name the name of the desired header |
||||
* @return the corresponding HeaderValueHolder, |
||||
* or <code>null</code> if none found |
||||
*/ |
||||
public static HeaderValueHolder getByName(Map<String, HeaderValueHolder> headers, String name) { |
||||
Assert.notNull(name, "Header name must not be null"); |
||||
for (String headerName : headers.keySet()) { |
||||
if (headerName.equalsIgnoreCase(name)) { |
||||
return headers.get(headerName); |
||||
} |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,603 @@
@@ -0,0 +1,603 @@
|
||||
/* |
||||
* Copyright 2002-2012 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 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
package org.springframework.mock.web; |
||||
|
||||
import java.io.ByteArrayOutputStream; |
||||
import java.io.IOException; |
||||
import java.io.OutputStream; |
||||
import java.io.OutputStreamWriter; |
||||
import java.io.PrintWriter; |
||||
import java.io.UnsupportedEncodingException; |
||||
import java.io.Writer; |
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
import java.util.Locale; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
import javax.servlet.ServletOutputStream; |
||||
import javax.servlet.http.Cookie; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
|
||||
import org.springframework.util.Assert; |
||||
import org.springframework.util.LinkedCaseInsensitiveMap; |
||||
import org.springframework.web.util.WebUtils; |
||||
|
||||
/** |
||||
* Mock implementation of the {@link javax.servlet.http.HttpServletResponse} interface. |
||||
* |
||||
* <p>Compatible with Servlet 2.5 as well as Servlet 3.0. |
||||
* |
||||
* @author Juergen Hoeller |
||||
* @author Rod Johnson |
||||
* @since 1.0.2 |
||||
*/ |
||||
public class MockHttpServletResponse implements HttpServletResponse { |
||||
|
||||
private static final String CHARSET_PREFIX = "charset="; |
||||
|
||||
private static final String CONTENT_TYPE_HEADER = "Content-Type"; |
||||
|
||||
private static final String CONTENT_LENGTH_HEADER = "Content-Length"; |
||||
|
||||
private static final String LOCATION_HEADER = "Location"; |
||||
|
||||
//---------------------------------------------------------------------
|
||||
// ServletResponse properties
|
||||
//---------------------------------------------------------------------
|
||||
|
||||
private boolean outputStreamAccessAllowed = true; |
||||
|
||||
private boolean writerAccessAllowed = true; |
||||
|
||||
private String characterEncoding = WebUtils.DEFAULT_CHARACTER_ENCODING; |
||||
|
||||
private boolean charset = false; |
||||
|
||||
private final ByteArrayOutputStream content = new ByteArrayOutputStream(); |
||||
|
||||
private final ServletOutputStream outputStream = new ResponseServletOutputStream(this.content); |
||||
|
||||
private PrintWriter writer; |
||||
|
||||
private int contentLength = 0; |
||||
|
||||
private String contentType; |
||||
|
||||
private int bufferSize = 4096; |
||||
|
||||
private boolean committed; |
||||
|
||||
private Locale locale = Locale.getDefault(); |
||||
|
||||
|
||||
//---------------------------------------------------------------------
|
||||
// HttpServletResponse properties
|
||||
//---------------------------------------------------------------------
|
||||
|
||||
private final List<Cookie> cookies = new ArrayList<Cookie>(); |
||||
|
||||
private final Map<String, HeaderValueHolder> headers = new LinkedCaseInsensitiveMap<HeaderValueHolder>(); |
||||
|
||||
private int status = HttpServletResponse.SC_OK; |
||||
|
||||
private String errorMessage; |
||||
|
||||
private String forwardedUrl; |
||||
|
||||
private final List<String> includedUrls = new ArrayList<String>(); |
||||
|
||||
|
||||
//---------------------------------------------------------------------
|
||||
// ServletResponse interface
|
||||
//---------------------------------------------------------------------
|
||||
|
||||
/** |
||||
* Set whether {@link #getOutputStream()} access is allowed. |
||||
* <p>Default is <code>true</code>. |
||||
*/ |
||||
public void setOutputStreamAccessAllowed(boolean outputStreamAccessAllowed) { |
||||
this.outputStreamAccessAllowed = outputStreamAccessAllowed; |
||||
} |
||||
|
||||
/** |
||||
* Return whether {@link #getOutputStream()} access is allowed. |
||||
*/ |
||||
public boolean isOutputStreamAccessAllowed() { |
||||
return this.outputStreamAccessAllowed; |
||||
} |
||||
|
||||
/** |
||||
* Set whether {@link #getWriter()} access is allowed. |
||||
* <p>Default is <code>true</code>. |
||||
*/ |
||||
public void setWriterAccessAllowed(boolean writerAccessAllowed) { |
||||
this.writerAccessAllowed = writerAccessAllowed; |
||||
} |
||||
|
||||
/** |
||||
* Return whether {@link #getOutputStream()} access is allowed. |
||||
*/ |
||||
public boolean isWriterAccessAllowed() { |
||||
return this.writerAccessAllowed; |
||||
} |
||||
|
||||
public void setCharacterEncoding(String characterEncoding) { |
||||
this.characterEncoding = characterEncoding; |
||||
this.charset = true; |
||||
updateContentTypeHeader(); |
||||
} |
||||
|
||||
private void updateContentTypeHeader() { |
||||
if (this.contentType != null) { |
||||
StringBuilder sb = new StringBuilder(this.contentType); |
||||
if (this.contentType.toLowerCase().indexOf(CHARSET_PREFIX) == -1 && this.charset) { |
||||
sb.append(";").append(CHARSET_PREFIX).append(this.characterEncoding); |
||||
} |
||||
doAddHeaderValue(CONTENT_TYPE_HEADER, sb.toString(), true); |
||||
} |
||||
} |
||||
|
||||
public String getCharacterEncoding() { |
||||
return this.characterEncoding; |
||||
} |
||||
|
||||
public ServletOutputStream getOutputStream() { |
||||
if (!this.outputStreamAccessAllowed) { |
||||
throw new IllegalStateException("OutputStream access not allowed"); |
||||
} |
||||
return this.outputStream; |
||||
} |
||||
|
||||
public PrintWriter getWriter() throws UnsupportedEncodingException { |
||||
if (!this.writerAccessAllowed) { |
||||
throw new IllegalStateException("Writer access not allowed"); |
||||
} |
||||
if (this.writer == null) { |
||||
Writer targetWriter = (this.characterEncoding != null ? |
||||
new OutputStreamWriter(this.content, this.characterEncoding) : new OutputStreamWriter(this.content)); |
||||
this.writer = new ResponsePrintWriter(targetWriter); |
||||
} |
||||
return this.writer; |
||||
} |
||||
|
||||
public byte[] getContentAsByteArray() { |
||||
flushBuffer(); |
||||
return this.content.toByteArray(); |
||||
} |
||||
|
||||
public String getContentAsString() throws UnsupportedEncodingException { |
||||
flushBuffer(); |
||||
return (this.characterEncoding != null) ? |
||||
this.content.toString(this.characterEncoding) : this.content.toString(); |
||||
} |
||||
|
||||
public void setContentLength(int contentLength) { |
||||
this.contentLength = contentLength; |
||||
doAddHeaderValue(CONTENT_LENGTH_HEADER, contentLength, true); |
||||
} |
||||
|
||||
public int getContentLength() { |
||||
return this.contentLength; |
||||
} |
||||
|
||||
public void setContentType(String contentType) { |
||||
this.contentType = contentType; |
||||
if (contentType != null) { |
||||
int charsetIndex = contentType.toLowerCase().indexOf(CHARSET_PREFIX); |
||||
if (charsetIndex != -1) { |
||||
String encoding = contentType.substring(charsetIndex + CHARSET_PREFIX.length()); |
||||
this.characterEncoding = encoding; |
||||
this.charset = true; |
||||
} |
||||
updateContentTypeHeader(); |
||||
} |
||||
} |
||||
|
||||
public String getContentType() { |
||||
return this.contentType; |
||||
} |
||||
|
||||
public void setBufferSize(int bufferSize) { |
||||
this.bufferSize = bufferSize; |
||||
} |
||||
|
||||
public int getBufferSize() { |
||||
return this.bufferSize; |
||||
} |
||||
|
||||
public void flushBuffer() { |
||||
setCommitted(true); |
||||
} |
||||
|
||||
public void resetBuffer() { |
||||
if (isCommitted()) { |
||||
throw new IllegalStateException("Cannot reset buffer - response is already committed"); |
||||
} |
||||
this.content.reset(); |
||||
} |
||||
|
||||
private void setCommittedIfBufferSizeExceeded() { |
||||
int bufSize = getBufferSize(); |
||||
if (bufSize > 0 && this.content.size() > bufSize) { |
||||
setCommitted(true); |
||||
} |
||||
} |
||||
|
||||
public void setCommitted(boolean committed) { |
||||
this.committed = committed; |
||||
} |
||||
|
||||
public boolean isCommitted() { |
||||
return this.committed; |
||||
} |
||||
|
||||
public void reset() { |
||||
resetBuffer(); |
||||
this.characterEncoding = null; |
||||
this.contentLength = 0; |
||||
this.contentType = null; |
||||
this.locale = null; |
||||
this.cookies.clear(); |
||||
this.headers.clear(); |
||||
this.status = HttpServletResponse.SC_OK; |
||||
this.errorMessage = null; |
||||
} |
||||
|
||||
public void setLocale(Locale locale) { |
||||
this.locale = locale; |
||||
} |
||||
|
||||
public Locale getLocale() { |
||||
return this.locale; |
||||
} |
||||
|
||||
|
||||
//---------------------------------------------------------------------
|
||||
// HttpServletResponse interface
|
||||
//---------------------------------------------------------------------
|
||||
|
||||
public void addCookie(Cookie cookie) { |
||||
Assert.notNull(cookie, "Cookie must not be null"); |
||||
this.cookies.add(cookie); |
||||
} |
||||
|
||||
public Cookie[] getCookies() { |
||||
return this.cookies.toArray(new Cookie[this.cookies.size()]); |
||||
} |
||||
|
||||
public Cookie getCookie(String name) { |
||||
Assert.notNull(name, "Cookie name must not be null"); |
||||
for (Cookie cookie : this.cookies) { |
||||
if (name.equals(cookie.getName())) { |
||||
return cookie; |
||||
} |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
public boolean containsHeader(String name) { |
||||
return (HeaderValueHolder.getByName(this.headers, name) != null); |
||||
} |
||||
|
||||
/** |
||||
* Return the names of all specified headers as a Set of Strings. |
||||
* <p>As of Servlet 3.0, this method is also defined HttpServletResponse. |
||||
* @return the <code>Set</code> of header name <code>Strings</code>, or an empty <code>Set</code> if none |
||||
*/ |
||||
public Set<String> getHeaderNames() { |
||||
return this.headers.keySet(); |
||||
} |
||||
|
||||
/** |
||||
* Return the primary value for the given header as a String, if any. |
||||
* Will return the first value in case of multiple values. |
||||
* <p>As of Servlet 3.0, this method is also defined in HttpServletResponse. |
||||
* As of Spring 3.1, it returns a stringified value for Servlet 3.0 compatibility. |
||||
* Consider using {@link #getHeaderValue(String)} for raw Object access. |
||||
* @param name the name of the header |
||||
* @return the associated header value, or <code>null<code> if none |
||||
*/ |
||||
public String getHeader(String name) { |
||||
HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name); |
||||
return (header != null ? header.getStringValue() : null); |
||||
} |
||||
|
||||
/** |
||||
* Return all values for the given header as a List of Strings. |
||||
* <p>As of Servlet 3.0, this method is also defined in HttpServletResponse. |
||||
* As of Spring 3.1, it returns a List of stringified values for Servlet 3.0 compatibility. |
||||
* Consider using {@link #getHeaderValues(String)} for raw Object access. |
||||
* @param name the name of the header |
||||
* @return the associated header values, or an empty List if none |
||||
*/ |
||||
public List<String> getHeaders(String name) { |
||||
HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name); |
||||
if (header != null) { |
||||
return header.getStringValues(); |
||||
} |
||||
else { |
||||
return Collections.emptyList(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Return the primary value for the given header, if any. |
||||
* <p>Will return the first value in case of multiple values. |
||||
* @param name the name of the header |
||||
* @return the associated header value, or <code>null<code> if none |
||||
*/ |
||||
public Object getHeaderValue(String name) { |
||||
HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name); |
||||
return (header != null ? header.getValue() : null); |
||||
} |
||||
|
||||
/** |
||||
* Return all values for the given header as a List of value objects. |
||||
* @param name the name of the header |
||||
* @return the associated header values, or an empty List if none |
||||
*/ |
||||
public List<Object> getHeaderValues(String name) { |
||||
HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name); |
||||
if (header != null) { |
||||
return header.getValues(); |
||||
} |
||||
else { |
||||
return Collections.emptyList(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* The default implementation returns the given URL String as-is. |
||||
* <p>Can be overridden in subclasses, appending a session id or the like. |
||||
*/ |
||||
public String encodeURL(String url) { |
||||
return url; |
||||
} |
||||
|
||||
/** |
||||
* The default implementation delegates to {@link #encodeURL}, |
||||
* returning the given URL String as-is. |
||||
* <p>Can be overridden in subclasses, appending a session id or the like |
||||
* in a redirect-specific fashion. For general URL encoding rules, |
||||
* override the common {@link #encodeURL} method instead, applying |
||||
* to redirect URLs as well as to general URLs. |
||||
*/ |
||||
public String encodeRedirectURL(String url) { |
||||
return encodeURL(url); |
||||
} |
||||
|
||||
public String encodeUrl(String url) { |
||||
return encodeURL(url); |
||||
} |
||||
|
||||
public String encodeRedirectUrl(String url) { |
||||
return encodeRedirectURL(url); |
||||
} |
||||
|
||||
public void sendError(int status, String errorMessage) throws IOException { |
||||
if (isCommitted()) { |
||||
throw new IllegalStateException("Cannot set error status - response is already committed"); |
||||
} |
||||
this.status = status; |
||||
this.errorMessage = errorMessage; |
||||
setCommitted(true); |
||||
} |
||||
|
||||
public void sendError(int status) throws IOException { |
||||
if (isCommitted()) { |
||||
throw new IllegalStateException("Cannot set error status - response is already committed"); |
||||
} |
||||
this.status = status; |
||||
setCommitted(true); |
||||
} |
||||
|
||||
public void sendRedirect(String url) throws IOException { |
||||
if (isCommitted()) { |
||||
throw new IllegalStateException("Cannot send redirect - response is already committed"); |
||||
} |
||||
Assert.notNull(url, "Redirect URL must not be null"); |
||||
setHeader(LOCATION_HEADER, url); |
||||
setStatus(HttpServletResponse.SC_MOVED_TEMPORARILY); |
||||
setCommitted(true); |
||||
} |
||||
|
||||
public String getRedirectedUrl() { |
||||
return getHeader(LOCATION_HEADER); |
||||
} |
||||
|
||||
public void setDateHeader(String name, long value) { |
||||
setHeaderValue(name, value); |
||||
} |
||||
|
||||
public void addDateHeader(String name, long value) { |
||||
addHeaderValue(name, value); |
||||
} |
||||
|
||||
public void setHeader(String name, String value) { |
||||
setHeaderValue(name, value); |
||||
} |
||||
|
||||
public void addHeader(String name, String value) { |
||||
addHeaderValue(name, value); |
||||
} |
||||
|
||||
public void setIntHeader(String name, int value) { |
||||
setHeaderValue(name, value); |
||||
} |
||||
|
||||
public void addIntHeader(String name, int value) { |
||||
addHeaderValue(name, value); |
||||
} |
||||
|
||||
private void setHeaderValue(String name, Object value) { |
||||
if (setSpecialHeader(name, value)) { |
||||
return; |
||||
} |
||||
doAddHeaderValue(name, value, true); |
||||
} |
||||
|
||||
private void addHeaderValue(String name, Object value) { |
||||
if (setSpecialHeader(name, value)) { |
||||
return; |
||||
} |
||||
doAddHeaderValue(name, value, false); |
||||
} |
||||
|
||||
private boolean setSpecialHeader(String name, Object value) { |
||||
if (CONTENT_TYPE_HEADER.equalsIgnoreCase(name)) { |
||||
setContentType((String) value); |
||||
return true; |
||||
} |
||||
else if (CONTENT_LENGTH_HEADER.equalsIgnoreCase(name)) { |
||||
setContentLength(Integer.parseInt((String) value)); |
||||
return true; |
||||
} |
||||
else { |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
private void doAddHeaderValue(String name, Object value, boolean replace) { |
||||
HeaderValueHolder header = HeaderValueHolder.getByName(this.headers, name); |
||||
Assert.notNull(value, "Header value must not be null"); |
||||
if (header == null) { |
||||
header = new HeaderValueHolder(); |
||||
this.headers.put(name, header); |
||||
} |
||||
if (replace) { |
||||
header.setValue(value); |
||||
} |
||||
else { |
||||
header.addValue(value); |
||||
} |
||||
} |
||||
|
||||
public void setStatus(int status) { |
||||
this.status = status; |
||||
} |
||||
|
||||
public void setStatus(int status, String errorMessage) { |
||||
this.status = status; |
||||
this.errorMessage = errorMessage; |
||||
} |
||||
|
||||
public int getStatus() { |
||||
return this.status; |
||||
} |
||||
|
||||
public String getErrorMessage() { |
||||
return this.errorMessage; |
||||
} |
||||
|
||||
|
||||
//---------------------------------------------------------------------
|
||||
// Methods for MockRequestDispatcher
|
||||
//---------------------------------------------------------------------
|
||||
|
||||
public void setForwardedUrl(String forwardedUrl) { |
||||
this.forwardedUrl = forwardedUrl; |
||||
} |
||||
|
||||
public String getForwardedUrl() { |
||||
return this.forwardedUrl; |
||||
} |
||||
|
||||
public void setIncludedUrl(String includedUrl) { |
||||
this.includedUrls.clear(); |
||||
if (includedUrl != null) { |
||||
this.includedUrls.add(includedUrl); |
||||
} |
||||
} |
||||
|
||||
public String getIncludedUrl() { |
||||
int count = this.includedUrls.size(); |
||||
if (count > 1) { |
||||
throw new IllegalStateException( |
||||
"More than 1 URL included - check getIncludedUrls instead: " + this.includedUrls); |
||||
} |
||||
return (count == 1 ? this.includedUrls.get(0) : null); |
||||
} |
||||
|
||||
public void addIncludedUrl(String includedUrl) { |
||||
Assert.notNull(includedUrl, "Included URL must not be null"); |
||||
this.includedUrls.add(includedUrl); |
||||
} |
||||
|
||||
public List<String> getIncludedUrls() { |
||||
return this.includedUrls; |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Inner class that adapts the ServletOutputStream to mark the |
||||
* response as committed once the buffer size is exceeded. |
||||
*/ |
||||
private class ResponseServletOutputStream extends DelegatingServletOutputStream { |
||||
|
||||
public ResponseServletOutputStream(OutputStream out) { |
||||
super(out); |
||||
} |
||||
|
||||
public void write(int b) throws IOException { |
||||
super.write(b); |
||||
super.flush(); |
||||
setCommittedIfBufferSizeExceeded(); |
||||
} |
||||
|
||||
public void flush() throws IOException { |
||||
super.flush(); |
||||
setCommitted(true); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Inner class that adapts the PrintWriter to mark the |
||||
* response as committed once the buffer size is exceeded. |
||||
*/ |
||||
private class ResponsePrintWriter extends PrintWriter { |
||||
|
||||
public ResponsePrintWriter(Writer out) { |
||||
super(out, true); |
||||
} |
||||
|
||||
public void write(char buf[], int off, int len) { |
||||
super.write(buf, off, len); |
||||
super.flush(); |
||||
setCommittedIfBufferSizeExceeded(); |
||||
} |
||||
|
||||
public void write(String s, int off, int len) { |
||||
super.write(s, off, len); |
||||
super.flush(); |
||||
setCommittedIfBufferSizeExceeded(); |
||||
} |
||||
|
||||
public void write(int c) { |
||||
super.write(c); |
||||
super.flush(); |
||||
setCommittedIfBufferSizeExceeded(); |
||||
} |
||||
|
||||
public void flush() { |
||||
super.flush(); |
||||
setCommitted(true); |
||||
} |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,93 @@
@@ -0,0 +1,93 @@
|
||||
/* |
||||
* Copyright 2002-2012 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 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
package org.springframework.mock.web; |
||||
|
||||
import javax.servlet.RequestDispatcher; |
||||
import javax.servlet.ServletRequest; |
||||
import javax.servlet.ServletResponse; |
||||
import javax.servlet.http.HttpServletResponseWrapper; |
||||
|
||||
import org.apache.commons.logging.Log; |
||||
import org.apache.commons.logging.LogFactory; |
||||
|
||||
import org.springframework.util.Assert; |
||||
|
||||
/** |
||||
* Mock implementation of the {@link javax.servlet.RequestDispatcher} interface. |
||||
* |
||||
* <p>Used for testing the web framework; typically not necessary for |
||||
* testing application controllers. |
||||
* |
||||
* @author Rod Johnson |
||||
* @author Juergen Hoeller |
||||
* @author Sam Brannen |
||||
* @since 1.0.2 |
||||
*/ |
||||
public class MockRequestDispatcher implements RequestDispatcher { |
||||
|
||||
private final Log logger = LogFactory.getLog(getClass()); |
||||
|
||||
private final String resource; |
||||
|
||||
|
||||
/** |
||||
* Create a new MockRequestDispatcher for the given resource. |
||||
* @param resource the server resource to dispatch to, located at a |
||||
* particular path or given by a particular name |
||||
*/ |
||||
public MockRequestDispatcher(String resource) { |
||||
Assert.notNull(resource, "resource must not be null"); |
||||
this.resource = resource; |
||||
} |
||||
|
||||
|
||||
public void forward(ServletRequest request, ServletResponse response) { |
||||
Assert.notNull(request, "Request must not be null"); |
||||
Assert.notNull(response, "Response must not be null"); |
||||
if (response.isCommitted()) { |
||||
throw new IllegalStateException("Cannot perform forward - response is already committed"); |
||||
} |
||||
getMockHttpServletResponse(response).setForwardedUrl(this.resource); |
||||
if (logger.isDebugEnabled()) { |
||||
logger.debug("MockRequestDispatcher: forwarding to [" + this.resource + "]"); |
||||
} |
||||
} |
||||
|
||||
public void include(ServletRequest request, ServletResponse response) { |
||||
Assert.notNull(request, "Request must not be null"); |
||||
Assert.notNull(response, "Response must not be null"); |
||||
getMockHttpServletResponse(response).addIncludedUrl(this.resource); |
||||
if (logger.isDebugEnabled()) { |
||||
logger.debug("MockRequestDispatcher: including [" + this.resource + "]"); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Obtain the underlying {@link MockHttpServletResponse}, unwrapping |
||||
* {@link HttpServletResponseWrapper} decorators if necessary. |
||||
*/ |
||||
protected MockHttpServletResponse getMockHttpServletResponse(ServletResponse response) { |
||||
if (response instanceof MockHttpServletResponse) { |
||||
return (MockHttpServletResponse) response; |
||||
} |
||||
if (response instanceof HttpServletResponseWrapper) { |
||||
return getMockHttpServletResponse(((HttpServletResponseWrapper) response).getResponse()); |
||||
} |
||||
throw new IllegalArgumentException("MockRequestDispatcher requires MockHttpServletResponse"); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue