net.anthavio.httl.util.MockTransport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hatatitla Show documentation
Compact but tweakable REST client library you have been dreaming of
The newest version!
package net.anthavio.httl.util;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.util.concurrent.atomic.AtomicInteger;

import net.anthavio.httl.HttlBody;
import net.anthavio.httl.HttlRequest;
import net.anthavio.httl.HttlResponse;
import net.anthavio.httl.HttlSender.Multival;
import net.anthavio.httl.HttlTransport;
import net.anthavio.httl.TransportBuilder.BaseTransBuilder;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Sometimes we need to test what is sent remote server without actualy sending it...
 * 
 * @author martin.vanek
 *
 */
public class MockTransport extends BaseTransBuilder implements HttlTransport {

	private final Logger logger = LoggerFactory.getLogger(getClass());

	private HttlRequest lastRequest; //from last doExecute invocation
	private HttlResponse lastResponse; //from last doExecute invocation

	private HttlResponse staticResponse; //what to return from doExecute

	private IOException exception; //throw from doExecute

	private boolean closed;

	private AtomicInteger executions = new AtomicInteger(0);

	public MockTransport(String url) {
		super(url);
	}

	public MockTransport() {
		this((HttlResponse) null);
	}

	public MockTransport(HttlResponse response) {
		super("http://mock.mock.mock:6363");
		this.staticResponse = response;
	}

	public MockTransport(int responseCode, String contentType, String responseBody) {
		super("http://mock.mock.mock:6363");
		setStaticResponse(responseCode, contentType, responseBody);
	}

	@Override
	public MockTransport getSelf() {
		return this;
	}

	@Override
	public MockTransport getConfig() {
		return this;
	}

	@Override
	public MockTransport build() {
		return this;
	}

	/**
	 * Test Sender closing
	 */
	public boolean isClosed() {
		return closed;
	}

	@Override
	public void close() {
		this.closed = true;
	}

	@Override
	public HttlResponse call(HttlRequest request) throws IOException {
		if (closed) {
			throw new IllegalStateException("Sender is closed");
		}
		this.lastRequest = request;
		executions.incrementAndGet();
		if (exception != null) {
			throw exception;
		}
		if (this.staticResponse != null) {
			this.lastResponse = staticResponse;
			return this.staticResponse;
		} else {
			MockResponse response;
			//copy request into response
			if (request.getBody() != null) {
				HttlBody body = request.getBody();
				logger.debug("Payload " + body.getType() + " : " + body.getPayload());

				switch (body.getType()) {
				case MARSHALL:
					ByteArrayOutputStream baos = new ByteArrayOutputStream();
					request.getSender().getMarshaller()
							.marshall(request.getBody().getPayload(), request.getMediaType(), request.getCharset(), baos);
					response = new MockResponse(request, 200, "OK", request.getHeaders(), baos.toByteArray());
					break;
				case STRING:
					response = new MockResponse(request, 200, "OK", request.getHeaders(), (String) body.getPayload());
					break;
				case BYTES:
					response = new MockResponse(request, 200, "OK", request.getHeaders(), (byte[]) body.getPayload());
					break;
				case STREAM:
					response = new MockResponse(request, 200, "OK", request.getHeaders(), (InputStream) body.getPayload());
					break;
				case READER:
					response = new MockResponse(request, 200, "OK", request.getHeaders(), new ReaderInputStream(
							(Reader) body.getPayload()));
					break;
				default:
					throw new IllegalStateException("Unsupported HttlBody type: " + body.getType());
				}

			} else {
				//No request body, but it still would be nice to return something according to Accept header
				String responseBody = "MockResponse to " + request.getMethod() + " " + request.getPathAndQuery();
				response = new MockResponse(request, 200, "OK", request.getHeaders(), responseBody);
			}
			this.lastResponse = response;
			return response;
		}

	}

	/**
	 * Set Exception thrown from doExecute
	 */
	public void setException(IOException exception) {
		this.exception = exception;
	}

	/**
	 * Set Response returned from doExecute
	 */
	public void setStaticResponse(int code, String contentType, String body) {
		Multival headers = new Multival();
		headers.set("Content-Type", contentType);
		this.staticResponse = new MockResponse(null, code, "OK", headers, body);
	}

	public void setStaticResponse(HttlResponse response) {
		this.staticResponse = response;
	}

	/**
	 * @return Response returned from doExecute
	 */
	public HttlResponse getStaticResponse() {
		return staticResponse;
	}

	/**
	 * Test for number of executions
	 */
	public int getExecutionCount() {
		return executions.intValue();
	}

	/**
	 * @return request from last doExecute invocation
	 */
	public HttlRequest getLastRequest() {
		return lastRequest;
	}

	/**
	 * @return response from last doExecute invocation
	 */
	public HttlResponse getLastResponse() {
		return lastResponse;
	}

}