MockHttpServletResponseImpl.java

/*
 *    Copyright 2011 Bryn Cooke
 *
 * 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.jglue.cdiunit.internal.servlet;

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.TimeZone;
import java.util.TreeMap;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletResponse;

/**
 * Shamlessly ripped from mockrunner. If mockrunner supports servlet 3.1 https://github.com/mockrunner/mockrunner/issues/4 then this class can extend mockrunner instead.
 * 
 * @author Various
 * 
 */
@CdiUnitServlet
public class MockHttpServletResponseImpl implements HttpServletResponse {
	private PrintWriter writer;
	private MockServletOutputStream outputStream;
	private Map headers;
	private Locale locale;
	private String characterEncoding;
	private int bufferSize;
	private boolean wasErrorSent;
	private boolean wasRedirectSent;
	private int errorCode;
	private int statusCode;
	private List cookies;
	private long contentLength;

	public MockHttpServletResponseImpl() {
		resetAll();
	}

	/**
	 * Resets the state of this object to the default values
	 */
	public void resetAll() {
		headers = new TreeMap(String.CASE_INSENSITIVE_ORDER);
		characterEncoding = "ISO-8859-1";
		bufferSize = 8192;
		wasErrorSent = false;
		wasRedirectSent = false;
		errorCode = SC_OK;
		statusCode = SC_OK;
		cookies = new ArrayList();
		outputStream = new MockServletOutputStream(characterEncoding);
		try {
			writer = new PrintWriter(new OutputStreamWriter(outputStream,
					characterEncoding), true);
		} catch (UnsupportedEncodingException exc) {
			throw new NestedApplicationException(exc);
		}
	}

	public String encodeURL(String url) {
		return url;
	}

	public String encodeRedirectUrl(String url) {
		return url;
	}

	public String encodeRedirectURL(String url) {
		return url;
	}

	public String encodeUrl(String url) {
		return url;
	}

	public PrintWriter getWriter() throws IOException {
		return writer;
	}

	public ServletOutputStream getOutputStream() throws IOException {
		return outputStream;
	}

	public String getOutputStreamContent() {
		return outputStream.getContent();
	}

	public void addCookie(Cookie cookie) {
		cookies.add(cookie);
	}

	public void addDateHeader(String key, long date) {
		addHeader(key, getDateString(date));
	}

	public void addHeader(String key, String value) {
		List valueList = (List) headers.get(key);
		if (null == valueList) {
			valueList = new ArrayList();
			headers.put(key, valueList);
		}
		valueList.add(value);
	}

	public void addIntHeader(String key, int value) {
		String stringValue = new Integer(value).toString();
		addHeader(key, stringValue);
	}

	public boolean containsHeader(String key) {
		return headers.containsKey(key);
	}

	public void sendError(int code, String message) throws IOException {
		errorCode = code;
		wasErrorSent = true;
	}

	public void sendError(int code) throws IOException {
		errorCode = code;
		wasErrorSent = true;
	}

	public void sendRedirect(String location) throws IOException {
		setHeader("Location", location);
		wasRedirectSent = true;
	}

	public void setDateHeader(String key, long date) {
		setHeader(key, getDateString(date));
	}

	public void setHeader(String key, String value) {
		List valueList = new ArrayList();
		headers.put(key, valueList);
		valueList.add(value);
	}

	public void setIntHeader(String key, int value) {
		String stringValue = new Integer(value).toString();
		setHeader(key, stringValue);
	}

	public void setStatus(int code, String message) {
		statusCode = code;
	}

	public void setStatus(int code) {
		statusCode = code;
	}

	public void flushBuffer() throws IOException {
		writer.flush();
		outputStream.flush();
	}

	public int getBufferSize() {
		return bufferSize;
	}

	public String getCharacterEncoding() {
		return characterEncoding;
	}

	public void setCharacterEncoding(String encoding) {
		characterEncoding = encoding;
		outputStream.setEncoding(encoding);
		try {
			writer = new PrintWriter(new OutputStreamWriter(outputStream,
					characterEncoding), true);
		} catch (UnsupportedEncodingException exc) {
			throw new NestedApplicationException(exc);
		}
	}

	public Locale getLocale() {
		return locale;
	}

	public void setLocale(Locale locale) {
		this.locale = locale;
	}

	public boolean isCommitted() {
		return false;
	}

	public void reset() {
		errorCode = SC_OK;
		statusCode = SC_OK;
		clearHeaders();
		resetBuffer();
	}

	public void resetBuffer() {
		outputStream.clearContent();
	}

	public void clearHeaders() {
		headers.clear();
	}

	public void setBufferSize(int size) {
		bufferSize = size;
	}

	public void setContentLength(int length) {
		setIntHeader("Content-Length", length);
	}

	public String getContentType() {
		return getHeader("Content-Type");
	}

	public void setContentType(String type) {
		setHeader("Content-Type", type);
	}

	public Collection getHeaderNames() {
		return headers.keySet();
	}

	public Collection getHeaders(String name) {
		List headerList = (List) headers.get(name);
		if (null == headerList)
			return null;
		return headerList;
	}

	public List getHeaderList(String key) {
		return (List) headers.get(key);
	}

	public String getHeader(String key) {
		List list = getHeaderList(key);
		if (null == list || 0 == list.size())
			return null;
		return (String) list.get(0);
	}

	public int getStatusCode() {
		return statusCode;
	}

	public int getStatus() {
		return getStatusCode();
	}

	public int getErrorCode() {
		return errorCode;
	}

	public List getCookies() {
		return cookies;
	}

	public boolean wasErrorSent() {
		return wasErrorSent;
	}

	public boolean wasRedirectSent() {
		return wasRedirectSent;
	}

	private String getDateString(long date) {
		Date dateValue = new Date(date);
		SimpleDateFormat dateFormat = new SimpleDateFormat(WebConstants.DATE_FORMAT_HEADER, Locale.US);
		dateFormat.setTimeZone(TimeZone.getTimeZone("GMT"));
		return dateFormat.format(dateValue);
	}

	
	public void setContentLengthLong(long len) {
		contentLength = len;
	}
}