org.mockserver.proxy.ProxyRunner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockserver-jetty Show documentation
A MockServer that uses embedded Jetty to run
There is a newer version: 2.9
package org.mockserver.proxy;

import com.google.common.annotations.VisibleForTesting;
import org.eclipse.jetty.server.ForwardedRequestCustomizer;
import org.eclipse.jetty.server.HttpConfiguration;
import org.mockserver.model.HttpRequest;
import org.mockserver.proxy.filters.ProxyRequestFilter;
import org.mockserver.proxy.filters.ProxyResponseFilter;
import org.mockserver.runner.AbstractRunner;
import org.mockserver.socket.PortFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServlet;
import java.io.IOException;
import java.net.*;
import java.util.Arrays;
import java.util.List;

/**
 * @author jamesdbloom
 */
public class ProxyRunner extends AbstractRunner {

    public static final int PROXY_PORT = PortFactory.findFreePort();
    public static final int PROXY_SECURE_PORT = PortFactory.findFreePort();
    private Logger logger = LoggerFactory.getLogger(this.getClass());

    @VisibleForTesting
    ProxyServlet proxyServlet = new ProxyServlet();

    protected HttpServlet getServlet() {
        return proxyServlet;
    }

    @Override
    protected void extendHTTPConfig(HttpConfiguration https_config) {
        https_config.addCustomizer(new ForwardedRequestCustomizer());
    }

    @Override
    protected void serverStarted(final Integer port, final Integer securePort) {
        // todo - need to support SOCKS protocol for this solution to work - jamesdbloom 12/01/2014
//        System.setProperty("proxySet", "true");
//        System.setProperty("http.proxyHost", "localhost");
//        System.setProperty("http.proxyPort", port.toString());
//        System.setProperty("java.net.useSystemProxies","true");
//        java.net.ProxySelector.setDefault(new ProxySelector() {
//            @Override
//            public List select(URI uri) {
//                return Arrays.asList(new Proxy(Proxy.Type.HTTP, new InetSocketAddress("localhost", port)));
//            }
//
//            @Override
//            public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
//                logger.error("Connection could not be established to proxy at socket [" + sa + "]", ioe);
//            }
//        });
    }

    /**
     * Add filter for HTTP requests, each filter get called before each request is proxied, if the filter return null then the request is not proxied
     *
     * @param httpRequest the request to match against for this filter
     * @param filter the filter to execute for this request, if the filter returns null the request will not be proxied
     */
    public ProxyRunner withFilter(HttpRequest httpRequest, ProxyRequestFilter filter) {
        proxyServlet.withFilter(httpRequest, filter);
        return this;
    }

    /**
     * Add filter for HTTP response, each filter get called after each request has been proxied
     *
     * @param httpRequest the request to match against for this filter
     * @param filter the filter that is executed after this request has been proxied
     */
    public ProxyRunner withFilter(HttpRequest httpRequest, ProxyResponseFilter filter) {
        proxyServlet.withFilter(httpRequest, filter);
        return this;
    }
}