From 46047e789101c765e748390ef4b538fadd7787a3 Mon Sep 17 00:00:00 2001
From: Frank Ganske <frank.ganske@web.de>
Date: Wed, 29 Jul 2015 23:13:25 +0200
Subject: [PATCH] Added a test to answer responses with content or to redirect.

It's a simple trick to add a web interface to a proxy application.
---
 .../org/littleshoot/proxy/ProxyWebUiTest.java | 96 +++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 src/test/java/org/littleshoot/proxy/ProxyWebUiTest.java

diff --git a/src/test/java/org/littleshoot/proxy/ProxyWebUiTest.java b/src/test/java/org/littleshoot/proxy/ProxyWebUiTest.java
new file mode 100644
index 000000000..2439df553
--- /dev/null
+++ b/src/test/java/org/littleshoot/proxy/ProxyWebUiTest.java
@@ -0,0 +1,96 @@
+package org.littleshoot.proxy;
+
+import static org.junit.Assert.*;
+import static org.hamcrest.Matchers.*;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import io.netty.handler.codec.http.DefaultFullHttpResponse;
+import io.netty.handler.codec.http.DefaultHttpResponse;
+import io.netty.handler.codec.http.HttpHeaders;
+import io.netty.handler.codec.http.HttpHeaders.Names;
+import io.netty.handler.codec.http.HttpHeaders.Values;
+import io.netty.handler.codec.http.HttpObject;
+import io.netty.handler.codec.http.HttpRequest;
+import io.netty.handler.codec.http.HttpResponse;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.handler.codec.http.HttpVersion;
+
+import java.util.concurrent.TimeUnit;
+
+import org.junit.Test;
+import org.littleshoot.proxy.impl.DefaultHttpProxyServer;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.firefox.FirefoxDriver;
+import org.openqa.selenium.remote.DesiredCapabilities;
+
+public class ProxyWebUiTest {
+
+    private static final String REDIRECT_PATH = "/";
+
+    private HttpFiltersAdapter answerRedirectFilter(HttpRequest originalRequest) {
+        return new HttpFiltersAdapter(originalRequest) {
+            @Override
+            public HttpResponse clientToProxyRequest(HttpObject httpObject) {
+                HttpResponse answer = new DefaultHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.FOUND);
+                HttpHeaders.setHeader(answer, HttpHeaders.Names.LOCATION, REDIRECT_PATH);
+                HttpHeaders.setHeader(answer, Names.CONNECTION, Values.CLOSE);
+                return answer;
+            }
+        };
+    }
+
+    private HttpFiltersAdapter answerContentFilter(HttpRequest originalRequest) {
+        return new HttpFiltersAdapter(originalRequest) {
+            @Override
+            public HttpResponse clientToProxyRequest(HttpObject httpObject) {
+                ByteBuf buffer = Unpooled.wrappedBuffer("CONTENT".getBytes());
+                HttpResponse answer = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.OK, buffer);
+                HttpHeaders.setContentLength(answer, buffer.readableBytes());
+                return answer;
+            }
+        };
+    }
+
+    @Test
+    public void testRedirectWithWebDriver() throws Exception {
+        HttpProxyServer proxyServer = null;
+        WebDriver driver = null;
+        try {
+
+            HttpFiltersSourceAdapter filtersSource = new HttpFiltersSourceAdapter() {
+                @Override
+                public HttpFilters filterRequest(HttpRequest originalRequest) {
+                    if (REDIRECT_PATH.equals(originalRequest.getUri())) {
+                        return answerContentFilter(originalRequest);
+                    } else {
+                        return answerRedirectFilter(originalRequest);
+                    }
+                }
+
+            };
+            proxyServer = DefaultHttpProxyServer.bootstrap()//
+                    .withFiltersSource(filtersSource)//
+                    .withPort(0)//
+                    .start();
+
+            DesiredCapabilities capability = DesiredCapabilities.htmlUnit();
+            driver = new FirefoxDriver(capability);
+            driver.manage().timeouts().pageLoadTimeout(5, TimeUnit.SECONDS);
+
+            String urlStr = String.format("http://localhost:%d/somewhere", proxyServer.getListenAddress().getPort());
+            driver.get(urlStr);
+            String source = driver.getPageSource();
+
+            assertThat(source, containsString("CONTENT"));
+
+        } finally {
+            if (driver != null) {
+                driver.close();
+            }
+            if (proxyServer != null) {
+                proxyServer.abort();
+            }
+        }
+    }
+
+}