Add Presto container

Author: findepi

modules/jdbc-test/build.gradle

@@ -7,6 +7,7 @@ repositories {
 dependencies {
     compile project(':mysql')
     compile project(':postgresql')
+    compile project(':presto')
     compile project(':cockroachdb')
     compile project(':mariadb')
     compile project(':oracle-xe')
@@ -16,6 +17,7 @@ dependencies {
 
     testCompile 'com.google.guava:guava:18.0'
     testCompile 'org.postgresql:postgresql:42.0.0'
+    testCompile 'io.prestosql:presto-jdbc:326'
     testCompile 'mysql:mysql-connector-java:8.0.14'
     testCompile 'org.mariadb.jdbc:mariadb-java-client:1.4.6'
     testCompile 'com.oracle:ojdbc6:12.1.0.1-atlassian-hosted'

modules/jdbc-test/src/test/java/org/testcontainers/jdbc/JDBCDriverTest.java

@@ -53,6 +53,7 @@ public static Iterable<Object[]> data() {
                 {"jdbc:tc:postgresql:9.6.8://hostname/databasename?user=someuser&password=somepwd", EnumSet.of(Options.JDBCParams)},
                 {"jdbc:tc:postgis://hostname/databasename?user=someuser&password=somepwd", EnumSet.of(Options.JDBCParams)},
                 {"jdbc:tc:postgis:9.6://hostname/databasename?user=someuser&password=somepwd", EnumSet.of(Options.JDBCParams)},
+                {"jdbc:tc:presto:326://hostname/", EnumSet.of(Options.PmdKnownBroken)},
                 {"jdbc:tc:mysql:5.6://hostname/databasename?TC_MY_CNF=somepath/mysql_conf_override", EnumSet.of(Options.CustomIniFile)},
                 {"jdbc:tc:mariadb://hostname/databasename", EnumSet.noneOf(Options.class)},
                 {"jdbc:tc:mariadb://hostname/databasename?user=someuser&TC_INITSCRIPT=somepath/init_mariadb.sql", EnumSet.of(Options.ScriptedSchema, Options.JDBCParams)},

modules/presto/build.gradle

@@ -0,0 +1,8 @@
+description = "Testcontainers :: JDBC :: Presto"
+
+dependencies {
+    compile project(':jdbc')
+
+    testCompile 'io.prestosql:presto-jdbc:326'
+    testCompile 'commons-dbutils:commons-dbutils:1.7'
+}

modules/presto/src/main/java/org/testcontainers/containers/PrestoContainer.java

@@ -0,0 +1,119 @@
+package org.testcontainers.containers;
+
+import com.google.common.base.Strings;
+import org.jetbrains.annotations.NotNull;
+import org.testcontainers.containers.wait.LogMessageWaitStrategy;
+
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.time.Duration;
+import java.util.HashSet;
+import java.util.Set;
+
+import static com.google.common.base.Strings.isNullOrEmpty;
+import static com.google.common.base.Strings.nullToEmpty;
+import static java.lang.String.format;
+import static java.time.temporal.ChronoUnit.SECONDS;
+
+public class PrestoContainer<SELF extends PrestoContainer<SELF>> extends JdbcDatabaseContainer<SELF> {
+    public static final String NAME = "presto";
+    public static final String IMAGE = "prestosql/presto";
+    public static final String DEFAULT_TAG = "326";
+
+    public static final Integer PRESTO_PORT = 8080;
+
+    private String username = "test";
+    private String password = "";
+    private String catalog = null;
+
+    public PrestoContainer() {
+        this(IMAGE + ":" + DEFAULT_TAG);
+    }
+
+    public PrestoContainer(final String dockerImageName) {
+        super(dockerImageName);
+        this.waitStrategy = new LogMessageWaitStrategy()
+            .withRegEx(".*io.prestosql.server.PrestoServer\\s+======== SERVER STARTED ========.*")
+            .withStartupTimeout(Duration.of(60, SECONDS));
+    }
+
+    @NotNull
+    @Override
+    protected Set<Integer> getLivenessCheckPorts() {
+        return new HashSet<>(getMappedPort(PRESTO_PORT));
+    }
+
+    @Override
+    protected void configure() {
+        addExposedPort(PRESTO_PORT);
+    }
+
+    @Override
+    public String getDriverClassName() {
+        return "io.prestosql.jdbc.PrestoDriver";
+    }
+
+    @Override
+    public String getJdbcUrl() {
+        return format("jdbc:presto://%s:%s/%s", getContainerIpAddress(), getMappedPort(PRESTO_PORT), nullToEmpty(catalog));
+    }
+
+    @Override
+    public String getUsername() {
+        return username;
+    }
+
+    @Override
+    public String getPassword() {
+        return password;
+    }
+
+    @Override
+    public String getDatabaseName() {
+        return catalog;
+    }
+
+    @Override
+    public String getTestQueryString() {
+        return "SELECT 1";
+    }
+
+    @Override
+    public SELF withUsername(final String username) {
+        this.username = username;
+        return self();
+    }
+
+    @Override
+    public SELF withPassword(final String password) {
+        // ignored, Presto does not support password authentication without TLS
+        return self();
+    }
+
+    @Override
+    public SELF withDatabaseName(String dbName) {
+        this.catalog = dbName;
+        return self();
+    }
+
+    @Override
+    protected void waitUntilContainerStarted() {
+        getWaitStrategy().waitUntilReady(this);
+    }
+
+    public Connection createConnection() throws SQLException, NoDriverFoundException {
+        return createConnection("");
+    }
+
+    @Override
+    protected void runInitScriptIfRequired() {
+        try {
+            // TODO remove when upgrading to 327
+            Thread.sleep(5000);
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt();
+            throw new RuntimeException("Interrupted");
+        }
+        super.runInitScriptIfRequired();
+    }
+}

modules/presto/src/main/java/org/testcontainers/containers/PrestoContainerProvider.java

@@ -0,0 +1,35 @@
+package org.testcontainers.containers;
+
+import org.testcontainers.jdbc.ConnectionUrl;
+
+import java.util.Objects;
+
+/**
+ * Factory for Presto containers.
+ */
+public class PrestoContainerProvider extends JdbcDatabaseContainerProvider {
+
+    public static final String USER_PARAM = "user";
+    public static final String PASSWORD_PARAM = "password";
+
+    @Override
+    public boolean supports(String databaseType) {
+        return databaseType.equals(PrestoContainer.NAME);
+    }
+
+    @Override
+    public JdbcDatabaseContainer newInstance() {
+        return newInstance(PrestoContainer.DEFAULT_TAG);
+    }
+
+    @Override
+    public JdbcDatabaseContainer newInstance(String tag) {
+        return new PrestoContainer(PrestoContainer.IMAGE + ":" + tag);
+    }
+
+    @Override
+    public JdbcDatabaseContainer newInstance(ConnectionUrl connectionUrl) {
+        return newInstanceFromConnectionUrl(connectionUrl, USER_PARAM, PASSWORD_PARAM);
+    }
+
+}

modules/presto/src/main/resources/META-INF/services/org.testcontainers.containers.JdbcDatabaseContainerProvider

@@ -0,0 +1 @@
+org.testcontainers.containers.PrestoContainerProvider

modules/presto/src/test/java/org/testcontainers/containers/PrestoContainerTest.java

@@ -0,0 +1,72 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.testcontainers.containers;
+
+import org.junit.Test;
+
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.Statement;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author findepi
+ */
+public class PrestoContainerTest {
+    @Test
+    public void testSimple() throws Exception {
+        try (PrestoContainer<?> prestoContainer = new PrestoContainer<>()) {
+            prestoContainer.start();
+            try (Connection connection = prestoContainer.createConnection();
+                 Statement statement = connection.createStatement();
+                 ResultSet resultSet = statement.executeQuery("SELECT DISTINCT node_version FROM system.runtime.nodes")) {
+                assertTrue("No result", resultSet.next());
+                assertEquals("Presto version", PrestoContainer.DEFAULT_TAG, resultSet.getString("node_version"));
+            }
+        }
+    }
+
+    @Test
+    public void testSpecificVersion() throws Exception {
+        String prestoVersion = "325";
+        try (PrestoContainer<?> prestoContainer = new PrestoContainer<>("prestosql/presto:" + prestoVersion)) {
+            prestoContainer.start();
+            try (Connection connection = prestoContainer.createConnection();
+                 Statement statement = connection.createStatement();
+                 ResultSet resultSet = statement.executeQuery("SELECT DISTINCT node_version FROM system.runtime.nodes")) {
+                assertTrue("No result", resultSet.next());
+                assertEquals("Presto version", "325", resultSet.getString("node_version"));
+            }
+        }
+    }
+
+
+    @Test
+    public void testInitScript() throws Exception {
+        try (PrestoContainer<?> prestoContainer = new PrestoContainer<>()) {
+            prestoContainer.withInitScript("initial.sql");
+            prestoContainer.start();
+            try (Connection connection = prestoContainer.createConnection();
+                 Statement statement = connection.createStatement();
+                 ResultSet resultSet = statement.executeQuery("SELECT a FROM memory.default.test_table")) {
+                assertTrue("No result", resultSet.next());
+                assertEquals("Value", 12345678909324L, resultSet.getObject("a"));
+                assertFalse("Too many result", resultSet.next());
+            }
+        }
+    }
+}

modules/presto/src/test/resources/initial.sql

@@ -0,0 +1,2 @@
+CREATE TABLE memory.default.test_table(a bigint);
+INSERT INTO memory.default.test_table(a) VALUES (12345678909324);