add integration tests for postgresql, mysql and mariadb using testcontainers

commafeed-server/src/test/java/com/commafeed/CommaFeedDropwizardAppExtension.java

@@ -1,11 +1,21 @@
 package com.commafeed;
 
+import java.io.IOException;
+import java.io.InputStream;
 import java.sql.Connection;
 import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
 
 import javax.sql.DataSource;
 
 import org.mockserver.socket.PortFactory;
+import org.testcontainers.containers.JdbcDatabaseContainer;
+import org.testcontainers.containers.MariaDBContainer;
+import org.testcontainers.containers.MySQLContainer;
+import org.testcontainers.containers.PostgreSQLContainer;
 
 import com.codahale.metrics.MetricRegistry;
 
@@ -14,10 +24,57 @@ import io.dropwizard.testing.ResourceHelpers;
 import io.dropwizard.testing.junit5.DropwizardAppExtension;
 
 public class CommaFeedDropwizardAppExtension extends DropwizardAppExtension<CommaFeedConfiguration> {
+	private static final String TEST_DATABASE = System.getenv().getOrDefault("TEST_DATABASE", "h2");
+
+	private static final ConfigOverride[] CONFIG_OVERRIDES;
+	private static final List<String> DROP_ALL_STATEMENTS;
+	static {
+		List<ConfigOverride> overrides = new ArrayList<>();
+		overrides.add(ConfigOverride.config("server.applicationConnectors[0].port", String.valueOf(PortFactory.findFreePort())));
+
+		DatabaseConfiguration config = buildConfiguration(TEST_DATABASE);
+		JdbcDatabaseContainer<?> container = config.container();
+		if (container != null) {
+			container.withDatabaseName("commafeed");
+			container.withEnv("TZ", "UTC");
+			container.start();
+
+			overrides.add(ConfigOverride.config("database.url", container.getJdbcUrl()));
+			overrides.add(ConfigOverride.config("database.user", container.getUsername()));
+			overrides.add(ConfigOverride.config("database.password", container.getPassword()));
+			overrides.add(ConfigOverride.config("database.driverClass", container.getDriverClassName()));
+		}
+
+		CONFIG_OVERRIDES = overrides.toArray(new ConfigOverride[0]);
+		DROP_ALL_STATEMENTS = config.dropAllStatements();
+	}
 
 	public CommaFeedDropwizardAppExtension() {
-		super(CommaFeedApplication.class, ResourceHelpers.resourceFilePath("config.test.yml"),
-				ConfigOverride.config("server.applicationConnectors[0].port", String.valueOf(PortFactory.findFreePort())));
+		super(CommaFeedApplication.class, ResourceHelpers.resourceFilePath("config.test.yml"), CONFIG_OVERRIDES);
+	}
+
+	private static DatabaseConfiguration buildConfiguration(String databaseName) {
+		Properties properties = new Properties();
+		try (InputStream is = CommaFeedDropwizardAppExtension.class.getResourceAsStream("/docker-images.properties")) {
+			properties.load(is);
+		} catch (IOException e) {
+			throw new RuntimeException("could not read docker-images.properties", e);
+		}
+
+		String imageName = properties.getProperty(databaseName);
+		if ("postgresql".equals(databaseName)) {
+			JdbcDatabaseContainer<?> container = new PostgreSQLContainer<>(imageName).withTmpFs(Map.of("/var/lib/postgresql/data", "rw"));
+			return new DatabaseConfiguration(container, List.of("DROP SCHEMA public CASCADE", "CREATE SCHEMA public"));
+		} else if ("mysql".equals(databaseName)) {
+			JdbcDatabaseContainer<?> container = new MySQLContainer<>(imageName).withTmpFs(Map.of("/var/lib/mysql", "rw"));
+			return new DatabaseConfiguration(container, List.of("DROP DATABASE IF EXISTS commafeed", " CREATE DATABASE commafeed"));
+		} else if ("mariadb".equals(databaseName)) {
+			JdbcDatabaseContainer<?> container = new MariaDBContainer<>(imageName).withTmpFs(Map.of("/var/lib/mysql", "rw"));
+			return new DatabaseConfiguration(container, List.of("DROP DATABASE IF EXISTS commafeed", " CREATE DATABASE commafeed"));
+		} else {
+			// h2
+			return new DatabaseConfiguration(null, List.of("DROP ALL OBJECTS"));
+		}
 	}
 
 	@Override
@@ -27,10 +84,15 @@ public class CommaFeedDropwizardAppExtension extends DropwizardAppExtension<Comm
 		// clean database after each test
 		DataSource dataSource = getConfiguration().getDataSourceFactory().build(new MetricRegistry(), "cleanup");
 		try (Connection connection = dataSource.getConnection()) {
-			connection.prepareStatement("DROP ALL OBJECTS").executeUpdate();
+			for (String statement : DROP_ALL_STATEMENTS) {
+				connection.prepareStatement(statement).executeUpdate();
+			}
 		} catch (SQLException e) {
 			throw new RuntimeException("could not cleanup database", e);
 		}
 	}
 
+	private record DatabaseConfiguration(JdbcDatabaseContainer<?> container, List<String> dropAllStatements) {
+	}
+
 }