1 files changed, 94 insertions, 0 deletions
diff --git a/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDrivers.java b/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDrivers.java
new file mode 100644
index 00000000000..7d7530c32e0
--- /dev/null
+++ b/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDrivers.java
@@ -0,0 +1,94 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.jdisc.http.server.jetty;
+
+import com.google.inject.AbstractModule;
+import com.google.inject.Module;
+import com.google.inject.util.Modules;
+import com.yahoo.container.logging.ConnectionLog;
+import com.yahoo.container.logging.RequestLog;
+import com.yahoo.jdisc.handler.RequestHandler;
+import com.yahoo.jdisc.http.ConnectorConfig;
+import com.yahoo.jdisc.http.ServerConfig;
+import com.yahoo.jdisc.http.ServletPathsConfig;
+import com.yahoo.jdisc.http.guiceModules.ConnectorFactoryRegistryModule;
+import com.yahoo.jdisc.http.guiceModules.ServletModule;
+
+import java.nio.file.Path;
+
+/**
+ * @author Simon Thoresen Hult
+ * @author bjorncs
+ */
+public class TestDrivers {
+
+    public static TestDriver newConfiguredInstance(RequestHandler requestHandler,
+                                                   ServerConfig.Builder serverConfig,
+                                                   ConnectorConfig.Builder connectorConfig,
+                                                   Module... guiceModules) {
+        return TestDriver.newInstance(
+                JettyHttpServer.class,
+                requestHandler,
+                newConfigModule(serverConfig, connectorConfig, guiceModules));
+    }
+
+    public static TestDriver newInstance(RequestHandler requestHandler, Module... guiceModules) {
+        return TestDriver.newInstance(
+                JettyHttpServer.class,
+                requestHandler,
+                newConfigModule(
+                        new ServerConfig.Builder(),
+                        new ConnectorConfig.Builder(),
+                        guiceModules
+                ));
+    }
+
+    public enum TlsClientAuth { NEED, WANT }
+
+    public static TestDriver newInstanceWithSsl(RequestHandler requestHandler,
+                                                Path certificateFile,
+                                                Path privateKeyFile,
+                                                TlsClientAuth tlsClientAuth,
+                                                Module... guiceModules) {
+        return TestDriver.newInstance(
+                JettyHttpServer.class,
+                requestHandler,
+                        newConfigModule(
+                                new ServerConfig.Builder().connectionLog(new ServerConfig.ConnectionLog.Builder().enabled(true)),
+                                new ConnectorConfig.Builder()
+                                        .tlsClientAuthEnforcer(
+                                                new ConnectorConfig.TlsClientAuthEnforcer.Builder()
+                                                        .enable(true)
+                                                        .pathWhitelist("/status.html"))
+                                        .ssl(new ConnectorConfig.Ssl.Builder()
+                                                .enabled(true)
+                                                .clientAuth(tlsClientAuth == TlsClientAuth.NEED
+                                                        ? ConnectorConfig.Ssl.ClientAuth.Enum.NEED_AUTH
+                                                        : ConnectorConfig.Ssl.ClientAuth.Enum.WANT_AUTH)
+                                                .privateKeyFile(privateKeyFile.toString())
+                                                .certificateFile(certificateFile.toString())
+                                                .caCertificateFile(certificateFile.toString())),
+                                guiceModules));
+    }
+
+    private static Module newConfigModule(ServerConfig.Builder serverConfig,
+                                          ConnectorConfig.Builder connectorConfigBuilder,
+                                          Module... guiceModules) {
+        return Modules.override(
+                Modules.combine(
+                    new AbstractModule() {
+                        @Override
+                        protected void configure() {
+                            bind(ServletPathsConfig.class).toInstance(new ServletPathsConfig(new ServletPathsConfig.Builder()));
+                            bind(ServerConfig.class).toInstance(new ServerConfig(serverConfig));
+                            bind(ConnectorConfig.class).toInstance(new ConnectorConfig(connectorConfigBuilder));
+                            bind(FilterBindings.class).toInstance(new FilterBindings.Builder().build());
+                            bind(ConnectionLog.class).toInstance(new VoidConnectionLog());
+                            bind(RequestLog.class).toInstance(new VoidRequestLog());
+                        }
+                    },
+                    new ConnectorFactoryRegistryModule(connectorConfigBuilder),
+                    new ServletModule()))
+                .with(guiceModules);
+    }
+
+}