Change the Docker Image used by the Presto module to prestodb/presto

Author: linghengqian

docs/modules/databases/presto.md

@@ -1,8 +1,5 @@
 # Presto Module
 
-!!! note
-    This module is deprecated, use Trino module.
-
 See [Database containers](./index.md) for documentation and usage that is common to all database container types.
 
 ## Usage example
@@ -84,6 +81,6 @@ Add the following dependency to your `pom.xml`/`build.gradle` file:
 !!! hint
     Adding this Testcontainers library JAR will not automatically add the Presto JDBC driver JAR to your project.
     You should ensure that your project has the Presto JDBC driver as a dependency, if you plan on using it.
-    Refer to [Presto project download page](https://prestosql.io/download.html) for instructions.
+    Refer to [Presto project download page](https://prestodb.io/getting-started/) for instructions.
 
 

modules/presto/build.gradle

@@ -4,6 +4,6 @@ dependencies {
     api project(':jdbc')
 
     testImplementation project(':jdbc-test')
-    testRuntimeOnly 'io.prestosql:presto-jdbc:350'
+    testRuntimeOnly 'com.facebook.presto:presto-jdbc:0.288'
     compileOnly 'org.jetbrains:annotations:24.1.0'
 }

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

@@ -3,7 +3,9 @@
 import com.google.common.base.Strings;
 import org.jetbrains.annotations.NotNull;
 import org.testcontainers.containers.wait.strategy.LogMessageWaitStrategy;
+import org.testcontainers.images.builder.Transferable;
 import org.testcontainers.utility.DockerImageName;
+import org.testcontainers.utility.MountableFile;
 
 import java.sql.Connection;
 import java.sql.SQLException;
@@ -12,18 +14,21 @@
 import java.util.Set;
 
 /**
- * @deprecated Use {@code TrinoContainer} instead.
+ * Testcontainers implementation for Presto.
+ * <p>
+ * Supported image: {@code prestodb/presto}
+ * <p>
+ * Exposed ports: 8080
  */
-@Deprecated
 public class PrestoContainer<SELF extends PrestoContainer<SELF>> extends JdbcDatabaseContainer<SELF> {
 
     public static final String NAME = "presto";
 
-    private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse("ghcr.io/trinodb/presto");
+    private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse("prestodb/presto");
 
-    public static final String IMAGE = "ghcr.io/trinodb/presto";
+    public static final String IMAGE = "prestodb/presto";
 
-    public static final String DEFAULT_TAG = "344";
+    public static final String DEFAULT_TAG = "0.288";
 
     public static final Integer PRESTO_PORT = 8080;
 
@@ -53,6 +58,10 @@ public PrestoContainer(final DockerImageName dockerImageName) {
                 .withStartupTimeout(Duration.of(60, ChronoUnit.SECONDS));
 
         addExposedPort(PRESTO_PORT);
+        withCopyFileToContainer(
+            MountableFile.forClasspathResource("default", Transferable.DEFAULT_DIR_MODE),
+            "/opt/presto-server/etc"
+        );
     }
 
     /**
@@ -68,7 +77,7 @@ protected Set<Integer> getLivenessCheckPorts() {
 
     @Override
     public String getDriverClassName() {
-        return "io.prestosql.jdbc.PrestoDriver";
+        return "com.facebook.presto.jdbc.PrestoDriver";
     }
 
     @Override

modules/presto/src/main/resources/default/catalog/memory.properties

@@ -0,0 +1 @@
+connector.name=memory

modules/presto/src/main/resources/default/catalog/tpch.properties

@@ -0,0 +1 @@
+connector.name=tpch

modules/presto/src/main/resources/default/config.properties

@@ -0,0 +1,6 @@
+coordinator=true
+node-scheduler.include-coordinator=true
+http-server.http.port=8080
+discovery-server.enabled=true
+discovery.uri=http://localhost:8080
+offset-clause-enabled=true

modules/presto/src/main/resources/default/jvm.config

@@ -0,0 +1,9 @@
+-server
+-Xmx1G
+-XX:+UseG1GC
+-XX:G1HeapRegionSize=32M
+-XX:+UseGCOverheadLimit
+-XX:+ExplicitGCInvokesConcurrent
+-XX:+HeapDumpOnOutOfMemoryError
+-XX:+ExitOnOutOfMemoryError
+-Djdk.attach.allowAttachSelf=true

modules/presto/src/test/java/org/testcontainers/PrestoTestImages.java

@@ -3,6 +3,6 @@
 import org.testcontainers.utility.DockerImageName;
 
 public interface PrestoTestImages {
-    DockerImageName PRESTO_TEST_IMAGE = DockerImageName.parse("ghcr.io/trinodb/presto:344");
-    DockerImageName PRESTO_PREVIOUS_VERSION_TEST_IMAGE = DockerImageName.parse("ghcr.io/trinodb/presto:343");
+    DockerImageName PRESTO_TEST_IMAGE = DockerImageName.parse("prestodb/presto:0.288");
+    DockerImageName PRESTO_PREVIOUS_VERSION_TEST_IMAGE = DockerImageName.parse("prestodb/presto:0.287");
 }

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

@@ -28,7 +28,7 @@ public void testSimple() throws Exception {
                 assertThat(resultSet.next()).as("has result").isTrue();
                 assertThat(resultSet.getString("node_version"))
                     .as("Presto version")
-                    .isEqualTo(PrestoContainer.DEFAULT_TAG);
+                    .startsWith(PrestoContainer.DEFAULT_TAG);
                 assertHasCorrectExposedAndLivenessCheckPorts(prestoSql);
             }
         }
@@ -48,7 +48,7 @@ public void testSpecificVersion() throws Exception {
                 assertThat(resultSet.next()).as("has result").isTrue();
                 assertThat(resultSet.getString("node_version"))
                     .as("Presto version")
-                    .isEqualTo(PrestoTestImages.PRESTO_PREVIOUS_VERSION_TEST_IMAGE.getVersionPart());
+                    .startsWith(PrestoTestImages.PRESTO_PREVIOUS_VERSION_TEST_IMAGE.getVersionPart());
             }
         }
     }
@@ -73,15 +73,15 @@ public void testQueryMemoryAndTpch() throws SQLException {
                         "SELECT nationkey, element " +
                         "FROM tpch.tiny.nation " +
                         "JOIN memory.default.table_with_array twa ON nationkey = twa.id " +
-                        "LEFT JOIN UNNEST(my_array) a(element) ON true " +
-                        "ORDER BY element OFFSET 1 FETCH NEXT 3 ROWS WITH TIES "
+                        "CROSS JOIN UNNEST(my_array) a(element) " +
+                        "ORDER BY element OFFSET 1 FETCH FIRST 3 ROWS ONLY "
                     )
                 ) {
                     List<Integer> actualElements = new ArrayList<>();
                     while (resultSet.next()) {
                         actualElements.add(resultSet.getInt("element"));
                     }
-                    assertThat(actualElements).isEqualTo(Arrays.asList(2, 4, 42, 42, 42));
+                    assertThat(actualElements).isEqualTo(Arrays.asList(2, 4, 42));
                 }
             }
         }
@@ -112,8 +112,7 @@ public void testTcJdbcUri() throws Exception {
             )
         ) {
             // Verify metadata with tc: JDBC connection URI
-            assertThat(Integer.parseInt(PrestoContainer.DEFAULT_TAG))
-                .isEqualTo(connection.getMetaData().getDatabaseMajorVersion());
+            assertThat(0).isEqualTo(connection.getMetaData().getDatabaseMajorVersion());
 
             // Verify transactions with tc: JDBC connection URI
             assertThat(connection.getAutoCommit()).as("Is autocommit").isTrue();
@@ -129,19 +128,18 @@ public void testTcJdbcUri() throws Exception {
                     .isEqualTo(0);
                 try (
                     ResultSet resultSet = statement.executeQuery(
-                        "SELECT sum(cast(node_version AS bigint)) AS v FROM system.runtime.nodes"
+                        "SELECT node_version AS v FROM system.runtime.nodes"
                     )
                 ) {
                     assertThat(resultSet.next()).isTrue();
-                    assertThat(resultSet.getString("v")).isEqualTo(PrestoContainer.DEFAULT_TAG);
+                    assertThat(resultSet.getString("v")).startsWith(PrestoContainer.DEFAULT_TAG);
                     assertThat(resultSet.next()).isFalse();
                 }
                 connection.commit();
             } finally {
                 connection.rollback();
             }
-            connection.setAutoCommit(true);
-            assertThat(connection.getAutoCommit()).as("Is autocommit").isTrue();
+            assertThat(connection.getAutoCommit()).as("Is autocommit").isFalse();
             assertThat(connection.getTransactionIsolation())
                 .as("Transaction isolation should be retained")
                 .isEqualTo(Connection.TRANSACTION_READ_UNCOMMITTED);

modules/presto/src/test/java/org/testcontainers/jdbc/presto/PrestoJDBCDriverTest.java

@@ -14,7 +14,7 @@ public class PrestoJDBCDriverTest extends AbstractJDBCDriverTest {
     public static Iterable<Object[]> data() {
         return Arrays.asList(
             new Object[][] { //
-                { "jdbc:tc:presto:344://hostname/", EnumSet.of(Options.PmdKnownBroken) },
+                { "jdbc:tc:presto:0.288://hostname/", EnumSet.of(Options.PmdKnownBroken) },
             }
         );
     }