Testing of embedded examples

Signed-off-by: Joakim Erdfelt <joakim.erdfelt@gmail.com>

examples/embedded/src/main/java/org/eclipse/jetty/embedded/ManyHandlers.java

@@ -30,6 +30,8 @@ import org.eclipse.jetty.server.Handler;
 import org.eclipse.jetty.server.Request;
 import org.eclipse.jetty.server.Server;
 import org.eclipse.jetty.server.handler.AbstractHandler;
+import org.eclipse.jetty.server.handler.ContextHandler;
+import org.eclipse.jetty.server.handler.ContextHandlerCollection;
 import org.eclipse.jetty.server.handler.DefaultHandler;
 import org.eclipse.jetty.server.handler.HandlerCollection;
 import org.eclipse.jetty.server.handler.HandlerList;
@@ -99,20 +101,23 @@ public class ManyHandlers
                            HttpServletResponse response) throws IOException,
             ServletException
         {
-            request.setAttribute("welcome", "Hello");
+            response.setHeader("X-Welcome", "Greetings from WelcomeWrapHandler");
             super.handle(target, baseRequest, request, response);
         }
     }
 
-    public static void main(String[] args) throws Exception
+    public static Server createServer(int port) throws IOException
     {
-        Server server = new Server(8080);
+        Server server = new Server(port);
 
         // create the handlers
         Handler param = new ParamHandler();
         HandlerWrapper wrapper = new WelcomeWrapHandler();
         Handler hello = new HelloHandler();
-        Handler dft = new DefaultHandler();
+        GzipHandler gzipHandler = new GzipHandler();
+        gzipHandler.setMinGzipSize(10);
+        gzipHandler.addIncludedMimeTypes("text/plain");
+        gzipHandler.addIncludedMimeTypes("text/html");
 
         // configure request logging
         File requestLogFile = File.createTempFile("demo", "log");
@@ -120,16 +125,46 @@ public class ManyHandlers
         server.setRequestLog(ncsaLog);
 
         // create the handler collections
-        HandlerCollection handlers = new HandlerCollection();
-        HandlerList list = new HandlerList();
+        HandlerList handlers = new HandlerList();
 
-        // link them all together
+        // wrap contexts around specific handlers
         wrapper.setHandler(hello);
-        list.setHandlers(new Handler[]{param, new GzipHandler()});
-        handlers.setHandlers(new Handler[]{list, dft});
+        ContextHandler helloContext = new ContextHandler("/hello");
+        helloContext.setHandler(wrapper);
 
+        ContextHandler paramContext = new ContextHandler("/params");
+        paramContext.setHandler(param);
+
+        ContextHandlerCollection contexts = new ContextHandlerCollection(helloContext, paramContext);
+
+        // Wrap Contexts with GZIP
+        gzipHandler.setHandler(contexts);
+
+        // Set the top level Handler List
+        handlers.addHandler(gzipHandler);
+        handlers.addHandler(new DefaultHandler());
         server.setHandler(handlers);
 
+        /* At this point you have the following handler hierarchy.
+         *
+         * Server.handler:
+         * HandlerList
+         *    \- GzipHandler
+         *    |   \- ContextHandlerCollection
+         *    |       \- ContextHandler ("/hello")
+         *    |       |   \- WelcomeWrapHandler
+         *    |       |       \- HelloHandler
+         *    |       \- ContextHandler ("/params")
+         *    |           \- ParamHandler
+         *    \- DefaultHandler
+         */
+
+        return server;
+    }
+
+    public static void main(String[] args) throws Exception
+    {
+        Server server = createServer(8080);
         server.start();
         server.join();
     }

examples/embedded/src/test/java/org/eclipse/jetty/embedded/ManyHandlersTest.java

@@ -0,0 +1,132 @@
+//
+//  ========================================================================
+//  Copyright (c) 1995-2019 Mort Bay Consulting Pty. Ltd.
+//  ------------------------------------------------------------------------
+//  All rights reserved. This program and the accompanying materials
+//  are made available under the terms of the Eclipse Public License v1.0
+//  and Apache License v2.0 which accompanies this distribution.
+//
+//      The Eclipse Public License is available at
+//      http://www.eclipse.org/legal/epl-v10.html
+//
+//      The Apache License v2.0 is available at
+//      http://www.opensource.org/licenses/apache2.0.php
+//
+//  You may elect to redistribute this code under either of these licenses.
+//  ========================================================================
+//
+
+package org.eclipse.jetty.embedded;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.HttpURLConnection;
+import java.net.URI;
+import java.util.Map;
+import java.util.zip.GZIPInputStream;
+
+import org.eclipse.jetty.server.Server;
+import org.eclipse.jetty.util.IO;
+import org.eclipse.jetty.util.ajax.JSON;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.containsString;
+import static org.hamcrest.Matchers.is;
+
+public class ManyHandlersTest
+{
+    private Server server;
+
+    @BeforeEach
+    public void startServer() throws Exception
+    {
+        server = ManyHandlers.createServer(0);
+        server.start();
+    }
+
+    @AfterEach
+    public void stopServer() throws Exception
+    {
+        server.stop();
+    }
+
+    @Test
+    public void testGetParams() throws IOException
+    {
+        URI uri = server.getURI().resolve("/params?a=b&foo=bar");
+        HttpURLConnection http = (HttpURLConnection)uri.toURL().openConnection();
+        http.setRequestProperty("Accept-Encoding", "gzip");
+        assertThat("HTTP Response Status", http.getResponseCode(), is(HttpURLConnection.HTTP_OK));
+
+        // dumpResponseHeaders(http);
+
+        // test gzip
+        assertGzippedResponse(http);
+
+        // test response content
+        String responseBody = getResponseBody(http);
+        Object jsonObj = JSON.parse(responseBody);
+        Map jsonMap = (Map)jsonObj;
+        assertThat("Response JSON keys.size", jsonMap.keySet().size(), is(2));
+    }
+
+    @Test
+    public void testGetHello() throws IOException
+    {
+        URI uri = server.getURI().resolve("/hello");
+        HttpURLConnection http = (HttpURLConnection)uri.toURL().openConnection();
+        http.setRequestProperty("Accept-Encoding", "gzip");
+        assertThat("HTTP Response Status", http.getResponseCode(), is(HttpURLConnection.HTTP_OK));
+
+        // dumpResponseHeaders(http);
+
+        // test gzip
+        assertGzippedResponse(http);
+
+        // test expected header from wrapper
+        String welcome = http.getHeaderField("X-Welcome");
+        assertThat("X-Welcome header", welcome, containsString("Greetings from WelcomeWrapHandler"));
+
+        // test response content
+        String responseBody = getResponseBody(http);
+        assertThat("Response Content", responseBody, containsString("Hello"));
+    }
+
+    private void assertGzippedResponse(HttpURLConnection http)
+    {
+        String value = http.getHeaderField("Content-Encoding");
+        assertThat("Content-Encoding", value, containsString("gzip"));
+    }
+
+    private String getResponseBody(HttpURLConnection http) throws IOException
+    {
+        try (InputStream in = http.getInputStream();
+             GZIPInputStream gzipInputStream = new GZIPInputStream(in))
+        {
+            return IO.toString(gzipInputStream, UTF_8);
+        }
+    }
+
+    @SuppressWarnings("unused")
+    private void dumpResponseHeaders(HttpURLConnection http)
+    {
+        int i = 0;
+        while (true)
+        {
+            String field = http.getHeaderField(i);
+            if (field == null)
+                return;
+            String key = http.getHeaderFieldKey(i);
+            if (key != null)
+            {
+                System.out.printf("%s: ", key);
+            }
+            System.out.println(field);
+            i++;
+        }
+    }
+}