1 files changed, 78 insertions, 0 deletions
diff --git a/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDriver.java b/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDriver.java
new file mode 100644
index 00000000000..49928a5679e
--- /dev/null
+++ b/container-core/src/test/java/com/yahoo/jdisc/http/server/jetty/TestDriver.java
@@ -0,0 +1,78 @@
+// 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.Module;
+import com.yahoo.jdisc.application.ContainerBuilder;
+import com.yahoo.jdisc.handler.RequestHandler;
+import com.yahoo.jdisc.http.ConnectorConfig;
+import com.yahoo.security.SslContextBuilder;
+
+import javax.net.ssl.SSLContext;
+import java.nio.file.Paths;
+
+import static com.yahoo.yolean.Exceptions.uncheck;
+
+/**
+ * Provides functionality for setting up a jdisc container with an HTTP server and handlers.
+ *
+ * @author Simon Thoresen Hult
+ * @author bakksjo
+ */
+public class TestDriver {
+
+    private final com.yahoo.jdisc.test.TestDriver driver;
+    private final JettyHttpServer server;
+    private final SimpleHttpClient client;
+
+    private TestDriver(com.yahoo.jdisc.test.TestDriver driver, JettyHttpServer server, SimpleHttpClient client) {
+        this.driver = driver;
+        this.server = server;
+        this.client = client;
+    }
+
+    public static TestDriver newInstance(Class<? extends JettyHttpServer> serverClass,
+                                         RequestHandler requestHandler,
+                                         Module testConfig) {
+        com.yahoo.jdisc.test.TestDriver driver =
+                com.yahoo.jdisc.test.TestDriver.newSimpleApplicationInstance(testConfig);
+        ContainerBuilder builder = driver.newContainerBuilder();
+        JettyHttpServer server = builder.getInstance(serverClass);
+        builder.serverProviders().install(server);
+        builder.serverBindings().bind("http://*/*", requestHandler);
+        driver.activateContainer(builder);
+        server.start();
+
+        SimpleHttpClient client = new SimpleHttpClient(newSslContext(builder), server.getListenPort(), false);
+        return new TestDriver(driver, server, client);
+    }
+
+    public boolean close() {
+        server.close();
+        server.release();
+        uncheck(client::close);
+        return driver.close();
+    }
+
+    public JettyHttpServer server() { return server; }
+
+    public SimpleHttpClient client() { return client; }
+
+    public SimpleHttpClient newClient(final boolean useCompression) {
+        return new SimpleHttpClient(newSslContext(), server.getListenPort(), useCompression);
+    }
+
+    public SSLContext newSslContext() {
+        return newSslContext(driver.newContainerBuilder());
+    }
+
+    private static SSLContext newSslContext(ContainerBuilder builder) {
+        ConnectorConfig.Ssl sslConfig = builder.getInstance(ConnectorConfig.class).ssl();
+        if (!sslConfig.enabled()) return null;
+
+        return new SslContextBuilder()
+                .withKeyStore(Paths.get(sslConfig.privateKeyFile()), Paths.get(sslConfig.certificateFile()))
+                .withTrustStore(Paths.get(sslConfig.caCertificateFile()))
+                .build();
+    }
+
+}