Support init script at JdbcDatabaseContainer level (#575)

Currently init script can be applied only through ContainerDatabaseDriver which doesn't support private Docker registries. This limitation also denies generic container approach of usage. 
More information about the problem is in the corresponding issue: #532

So I've implemented an init script support for all Jdbc containers. 

Resolve #532

Author: AnkBurov

modules/database-commons/src/main/java/org/testcontainers/ext/ScriptUtils.java

@@ -16,12 +16,16 @@
 
 package org.testcontainers.ext;
 
+import org.apache.commons.io.IOUtils;
 import org.apache.commons.lang.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.testcontainers.delegate.DatabaseDelegate;
 
 import javax.script.ScriptException;
+import java.io.IOException;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
 import java.util.LinkedList;
 import java.util.List;
 
@@ -210,6 +214,30 @@ public static boolean containsSqlScriptDelimiters(String script, String delim) {
 		return false;
 	}
 
+	/**
+	 * Load script from classpath and apply it to the given database
+	 *
+	 * @param databaseDelegate database delegate for script execution
+	 * @param initScriptPath   the resource to load the init script from
+	 */
+	public static void runInitScript(DatabaseDelegate databaseDelegate, String initScriptPath) {
+		try {
+			URL resource = ScriptUtils.class.getClassLoader().getResource(initScriptPath);
+			if (resource == null) {
+				LOGGER.warn("Could not load classpath init script: {}", initScriptPath);
+				throw new ScriptLoadException("Could not load classpath init script: " + initScriptPath + ". Resource not found.");
+			}
+			String scripts = IOUtils.toString(resource, StandardCharsets.UTF_8);
+			executeDatabaseScript(databaseDelegate, initScriptPath, scripts);
+		} catch (IOException e) {
+			LOGGER.warn("Could not load classpath init script: {}", initScriptPath);
+			throw new ScriptLoadException("Could not load classpath init script: " + initScriptPath, e);
+		} catch (ScriptException e) {
+			LOGGER.error("Error while executing init script: {}", initScriptPath, e);
+			throw new UncategorizedScriptException("Error while executing init script: " + initScriptPath, e);
+		}
+	}
+
     public static void executeDatabaseScript(DatabaseDelegate databaseDelegate, String scriptPath, String script) throws ScriptException {
         executeDatabaseScript(databaseDelegate, scriptPath, script, false, false, DEFAULT_COMMENT_PREFIX, DEFAULT_STATEMENT_SEPARATOR, DEFAULT_BLOCK_COMMENT_START_DELIMITER, DEFAULT_BLOCK_COMMENT_END_DELIMITER);
     }

modules/jdbc-test/src/test/java/org/testcontainers/junit/SimpleMySQLTest.java

@@ -132,6 +132,20 @@ public void testMySQL8() throws SQLException {
         }
     }
 
+    @Test
+    public void testExplicitInitScript() throws SQLException {
+        try (MySQLContainer container = (MySQLContainer) new MySQLContainer()
+            .withInitScript("somepath/init_mysql.sql")
+            .withLogConsumer(new Slf4jLogConsumer(logger))) {
+            container.start();
+
+            ResultSet resultSet = performQuery(container, "SELECT foo FROM bar");
+            String firstColumnValue = resultSet.getString(1);
+
+            assertEquals("Value from init script should equal real value", "hello world", firstColumnValue);
+        }
+     }
+
     @Test
     public void testEmptyPasswordWithNonRootUser() {
 

modules/jdbc-test/src/test/java/org/testcontainers/junit/SimplePostgreSQLTest.java

@@ -2,8 +2,8 @@
 
 import com.zaxxer.hikari.HikariConfig;
 import com.zaxxer.hikari.HikariDataSource;
-import org.junit.Rule;
 import org.junit.Test;
+import org.testcontainers.containers.JdbcDatabaseContainer;
 import org.testcontainers.containers.PostgreSQLContainer;
 
 import java.sql.ResultSet;
@@ -17,23 +17,42 @@
  */
 public class SimplePostgreSQLTest {
 
-    @Rule
-    public PostgreSQLContainer postgres = new PostgreSQLContainer();
-
     @Test
     public void testSimple() throws SQLException {
+        try (PostgreSQLContainer postgres = new PostgreSQLContainer<>()) {
+            postgres.start();
+
+            ResultSet resultSet = performQuery(postgres, "SELECT 1");
+
+            int resultSetInt = resultSet.getInt(1);
+            assertEquals("A basic SELECT query succeeds", 1, resultSetInt);
+        }
+    }
+
+    @Test
+    public void testExplicitInitScript() throws SQLException {
+        try (PostgreSQLContainer postgres = new PostgreSQLContainer<>()
+                .withInitScript("somepath/init_postgresql.sql")) {
+            postgres.start();
+
+            ResultSet resultSet = performQuery(postgres, "SELECT foo FROM bar");
+
+            String firstColumnValue = resultSet.getString(1);
+            assertEquals("Value from init script should equal real value", "hello world", firstColumnValue);
+        }
+    }
+
+    private ResultSet performQuery(JdbcDatabaseContainer container, String sql) throws SQLException {
         HikariConfig hikariConfig = new HikariConfig();
-        hikariConfig.setJdbcUrl(postgres.getJdbcUrl());
-        hikariConfig.setUsername(postgres.getUsername());
-        hikariConfig.setPassword(postgres.getPassword());
+        hikariConfig.setJdbcUrl(container.getJdbcUrl());
+        hikariConfig.setUsername(container.getUsername());
+        hikariConfig.setPassword(container.getPassword());
 
         HikariDataSource ds = new HikariDataSource(hikariConfig);
         Statement statement = ds.getConnection().createStatement();
-        statement.execute("SELECT 1");
+        statement.execute(sql);
         ResultSet resultSet = statement.getResultSet();
-
         resultSet.next();
-        int resultSetInt = resultSet.getInt(1);
-        assertEquals("A basic SELECT query succeeds", 1, resultSetInt);
+        return resultSet;
     }
 }

modules/jdbc-test/src/test/resources/somepath/init_postgresql.sql

@@ -0,0 +1,5 @@
+CREATE TABLE bar (
+  foo VARCHAR(255)
+);
+
+INSERT INTO bar (foo) VALUES ('hello world');

modules/jdbc/src/main/java/org/testcontainers/containers/JdbcDatabaseContainer.java

@@ -1,11 +1,15 @@
 package org.testcontainers.containers;
 
 import lombok.NonNull;
+import com.github.dockerjava.api.command.InspectContainerResponse;
 import org.jetbrains.annotations.NotNull;
 import org.rnorth.ducttape.ratelimits.RateLimiter;
 import org.rnorth.ducttape.ratelimits.RateLimiterBuilder;
 import org.rnorth.ducttape.unreliables.Unreliables;
 import org.testcontainers.containers.traits.LinkableContainer;
+import org.testcontainers.delegate.DatabaseDelegate;
+import org.testcontainers.ext.ScriptUtils;
+import org.testcontainers.jdbc.JdbcDatabaseDelegate;
 import org.testcontainers.utility.MountableFile;
 
 import java.sql.Connection;
@@ -26,6 +30,7 @@ public abstract class JdbcDatabaseContainer<SELF extends JdbcDatabaseContainer<S
 
     private static final Object DRIVER_LOAD_MUTEX = new Object();
     private Driver driver;
+    private String initScriptPath;
     protected Map<String, String> parameters = new HashMap<>();
 
     private static final RateLimiter DB_CONNECT_RATE_LIMIT = RateLimiterBuilder.newBuilder()
@@ -111,6 +116,11 @@ public SELF withConnectTimeoutSeconds(int connectTimeoutSeconds) {
         return self();
     }
 
+    public SELF withInitScript(String initScriptPath) {
+        this.initScriptPath = initScriptPath;
+        return self();
+    }
+
     @Override
     protected void waitUntilContainerStarted() {
         // Repeatedly try and open a connection to the DB and execute a test query
@@ -135,6 +145,11 @@ protected void waitUntilContainerStarted() {
         });
     }
 
+    @Override
+    protected void containerIsStarted(InspectContainerResponse containerInfo) {
+        runInitScriptIfRequired();
+    }
+
     /**
      * Obtain an instance of the correct JDBC driver for this particular database container type
      *
@@ -202,6 +217,15 @@ protected void optionallyMapResourceParameterAsVolume(@NotNull String paramName,
         }
     }
 
+    /**
+     * Load init script content and apply it to the database if initScriptPath is set
+     */
+    protected void runInitScriptIfRequired() {
+        if (initScriptPath != null) {
+            ScriptUtils.runInitScript(getDatabaseDelegate(), initScriptPath);
+        }
+    }
+
     public void setParameters(Map<String, String> parameters) {
         this.parameters = parameters;
     }
@@ -228,4 +252,8 @@ protected int getStartupTimeoutSeconds() {
     protected int getConnectTimeoutSeconds() {
         return connectTimeoutSeconds;
     }
+
+    protected DatabaseDelegate getDatabaseDelegate() {
+        return new JdbcDatabaseDelegate(this, "");
+    }
 }