Merge branch 'main' into main

Author: rajucomp

docs/modules/solr.md

@@ -9,7 +9,7 @@ Note that it's based on the [official Docker image](https://hub.docker.com/_/sol
 You can start a solr container instance from any Java application by using:
 
 <!--codeinclude-->
-[Using a Solr container](../../modules/solr/src/test/java/org/testcontainers/containers/SolrContainerTest.java) inside_block:solrContainerUsage
+[Using a Solr container](../../modules/solr/src/test/java/org/testcontainers/solr/SolrContainerTest.java) inside_block:solrContainerUsage
 <!--/codeinclude-->
 
 ## Adding this module to your project dependencies

modules/cassandra/src/main/java/org/testcontainers/cassandra/CassandraContainer.java

@@ -1,8 +1,6 @@
 package org.testcontainers.cassandra;
 
 import com.github.dockerjava.api.command.InspectContainerResponse;
-import org.testcontainers.cassandra.delegate.CassandraDatabaseDelegate;
-import org.testcontainers.cassandra.wait.CassandraQueryWaitStrategy;
 import org.testcontainers.containers.GenericContainer;
 import org.testcontainers.ext.ScriptUtils;
 import org.testcontainers.ext.ScriptUtils.ScriptLoadException;

…/delegate/CassandraDatabaseDelegate.java …cassandra/CassandraDatabaseDelegate.javamodules/cassandra/src/main/java/org/testcontainers/cassandra/delegate/CassandraDatabaseDelegate.java renamed to modules/cassandra/src/main/java/org/testcontainers/cassandra/CassandraDatabaseDelegate.java modules/cassandra/src/main/java/org/testcontainers/cassandra/delegate/CassandraDatabaseDelegate.java renamed to modules/cassandra/src/main/java/org/testcontainers/cassandra/CassandraDatabaseDelegate.java

@@ -1,10 +1,9 @@
-package org.testcontainers.cassandra.delegate;
+package org.testcontainers.cassandra;
 
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ArrayUtils;
 import org.apache.commons.lang3.StringUtils;
-import org.testcontainers.cassandra.CassandraContainer;
 import org.testcontainers.containers.Container;
 import org.testcontainers.containers.ContainerState;
 import org.testcontainers.containers.ExecConfig;

…dra/wait/CassandraQueryWaitStrategy.java …assandra/CassandraQueryWaitStrategy.javamodules/cassandra/src/main/java/org/testcontainers/cassandra/wait/CassandraQueryWaitStrategy.java renamed to modules/cassandra/src/main/java/org/testcontainers/cassandra/CassandraQueryWaitStrategy.java modules/cassandra/src/main/java/org/testcontainers/cassandra/wait/CassandraQueryWaitStrategy.java renamed to modules/cassandra/src/main/java/org/testcontainers/cassandra/CassandraQueryWaitStrategy.java

@@ -1,7 +1,6 @@
-package org.testcontainers.cassandra.wait;
+package org.testcontainers.cassandra;
 
 import org.rnorth.ducttape.TimeoutException;
-import org.testcontainers.cassandra.delegate.CassandraDatabaseDelegate;
 import org.testcontainers.containers.ContainerLaunchException;
 import org.testcontainers.containers.wait.strategy.AbstractWaitStrategy;
 import org.testcontainers.delegate.DatabaseDelegate;

modules/cassandra/src/main/java/org/testcontainers/containers/delegate/CassandraDatabaseDelegate.java

@@ -14,7 +14,7 @@
 /**
  * Cassandra database delegate
  *
- * @deprecated use {@link org.testcontainers.cassandra.delegate.CassandraDatabaseDelegate} instead.
+ * @deprecated use {@link org.testcontainers.cassandra.CassandraDatabaseDelegate} instead.
  */
 @Slf4j
 @RequiredArgsConstructor

modules/cassandra/src/main/java/org/testcontainers/containers/wait/CassandraQueryWaitStrategy.java

@@ -13,7 +13,7 @@
 /**
  * Waits until Cassandra returns its version
  *
- * @deprecated use {@link org.testcontainers.cassandra.wait.CassandraQueryWaitStrategy} instead.
+ * @deprecated use {@link org.testcontainers.cassandra.CassandraQueryWaitStrategy} instead.
  */
 @Deprecated
 public class CassandraQueryWaitStrategy extends AbstractWaitStrategy {

modules/kafka/src/main/java/org/testcontainers/kafka/ConfluentKafkaContainer.java

@@ -6,7 +6,7 @@
 import org.testcontainers.utility.DockerImageName;
 
 import java.util.ArrayList;
-import java.util.HashSet;
+import java.util.LinkedHashSet;
 import java.util.List;
 import java.util.Set;
 import java.util.function.Supplier;
@@ -22,9 +22,9 @@ public class ConfluentKafkaContainer extends GenericContainer<ConfluentKafkaCont
 
     private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse("confluentinc/cp-kafka");
 
-    private final Set<String> listeners = new HashSet<>();
+    private final Set<String> listeners = new LinkedHashSet<>();
 
-    private final Set<Supplier<String>> advertisedListeners = new HashSet<>();
+    private final Set<Supplier<String>> advertisedListeners = new LinkedHashSet<>();
 
     public ConfluentKafkaContainer(String imageName) {
         this(DockerImageName.parse(imageName));

modules/kafka/src/main/java/org/testcontainers/kafka/KafkaContainer.java

@@ -6,7 +6,7 @@
 import org.testcontainers.utility.DockerImageName;
 
 import java.util.ArrayList;
-import java.util.HashSet;
+import java.util.LinkedHashSet;
 import java.util.List;
 import java.util.Set;
 import java.util.function.Supplier;
@@ -28,9 +28,9 @@ public class KafkaContainer extends GenericContainer<KafkaContainer> {
 
     private static final String STARTER_SCRIPT = "/tmp/testcontainers_start.sh";
 
-    private final Set<String> listeners = new HashSet<>();
+    private final Set<String> listeners = new LinkedHashSet<>();
 
-    private final Set<Supplier<String>> advertisedListeners = new HashSet<>();
+    private final Set<Supplier<String>> advertisedListeners = new LinkedHashSet<>();
 
     public KafkaContainer(String imageName) {
         this(DockerImageName.parse(imageName));

modules/solr/src/main/java/org/testcontainers/containers/SolrContainer.java

@@ -23,6 +23,8 @@
  *     <li>Solr: 8983</li>
  *     <li>Zookeeper: 9983</li>
  * </ul>
+ *
+ * @deprecated use {@link org.testcontainers.solr.SolrContainer} instead.
  */
 public class SolrContainer extends GenericContainer<SolrContainer> {
 

modules/solr/src/main/java/org/testcontainers/solr/SolrContainer.java

@@ -0,0 +1,156 @@
+package org.testcontainers.solr;
+
+import com.github.dockerjava.api.command.InspectContainerResponse;
+import lombok.SneakyThrows;
+import org.apache.commons.lang3.StringUtils;
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.SolrClientUtils;
+import org.testcontainers.containers.SolrContainerConfiguration;
+import org.testcontainers.containers.wait.strategy.Wait;
+import org.testcontainers.utility.ComparableVersion;
+import org.testcontainers.utility.DockerImageName;
+
+import java.net.URL;
+import java.time.Duration;
+import java.util.HashSet;
+import java.util.Set;
+
+/**
+ * Testcontainers implementation for Solr.
+ * <p>
+ * Supported image: {@code solr}
+ * <p>
+ * Exposed ports:
+ * <ul>
+ *     <li>Solr: 8983</li>
+ *     <li>Zookeeper: 9983</li>
+ * </ul>
+ */
+public class SolrContainer extends GenericContainer<SolrContainer> {
+
+    private static final DockerImageName DEFAULT_IMAGE_NAME = DockerImageName.parse("solr");
+
+    public static final Integer ZOOKEEPER_PORT = 9983;
+
+    public static final Integer SOLR_PORT = 8983;
+
+    private SolrContainerConfiguration configuration;
+
+    private final ComparableVersion imageVersion;
+
+    public SolrContainer(final String dockerImageName) {
+        this(DockerImageName.parse(dockerImageName));
+    }
+
+    public SolrContainer(final DockerImageName dockerImageName) {
+        super(dockerImageName);
+        dockerImageName.assertCompatibleWith(DEFAULT_IMAGE_NAME);
+
+        waitingFor(
+            Wait.forLogMessage(".*o\\.e\\.j\\.s\\.Server Started.*", 1).withStartupTimeout(Duration.ofMinutes(1))
+        );
+        this.configuration = new SolrContainerConfiguration();
+        this.imageVersion = new ComparableVersion(dockerImageName.getVersionPart());
+    }
+
+    public SolrContainer withZookeeper(boolean zookeeper) {
+        configuration.setZookeeper(zookeeper);
+        return self();
+    }
+
+    public SolrContainer withCollection(String collection) {
+        if (StringUtils.isEmpty(collection)) {
+            throw new IllegalArgumentException("Collection name must not be empty");
+        }
+        configuration.setCollectionName(collection);
+        return self();
+    }
+
+    public SolrContainer withConfiguration(String name, URL solrConfig) {
+        if (StringUtils.isEmpty(name) || solrConfig == null) {
+            throw new IllegalArgumentException();
+        }
+        configuration.setConfigurationName(name);
+        configuration.setSolrConfiguration(solrConfig);
+        return self();
+    }
+
+    public SolrContainer withSchema(URL schema) {
+        configuration.setSolrSchema(schema);
+        return self();
+    }
+
+    public int getSolrPort() {
+        return getMappedPort(SOLR_PORT);
+    }
+
+    public int getZookeeperPort() {
+        return getMappedPort(ZOOKEEPER_PORT);
+    }
+
+    @Override
+    @SneakyThrows
+    protected void configure() {
+        if (configuration.getSolrSchema() != null && configuration.getSolrConfiguration() == null) {
+            throw new IllegalStateException("Solr needs to have a configuration if you want to use a schema");
+        }
+        // Generate Command Builder
+        String command = "solr start -f";
+        // Add Default Ports
+        addExposedPort(SOLR_PORT);
+
+        // Configure Zookeeper
+        if (configuration.isZookeeper()) {
+            addExposedPort(ZOOKEEPER_PORT);
+            if (this.imageVersion.isGreaterThanOrEqualTo("9.7.0")) {
+                command = "-DzkRun --host localhost";
+            } else {
+                command = "-DzkRun -h localhost";
+            }
+        }
+
+        // Apply generated Command
+        setCommand(command);
+    }
+
+    @Override
+    public Set<Integer> getLivenessCheckPortNumbers() {
+        return new HashSet<>(getSolrPort());
+    }
+
+    @Override
+    protected void waitUntilContainerStarted() {
+        getWaitStrategy().waitUntilReady(this);
+    }
+
+    @Override
+    @SneakyThrows
+    protected void containerIsStarted(InspectContainerResponse containerInfo) {
+        if (!configuration.isZookeeper()) {
+            ExecResult result = execInContainer("solr", "create", "-c", configuration.getCollectionName());
+            if (result.getExitCode() != 0) {
+                throw new IllegalStateException(
+                    "Unable to create solr core:\nStdout: " + result.getStdout() + "\nStderr:" + result.getStderr()
+                );
+            }
+            return;
+        }
+
+        if (StringUtils.isNotEmpty(configuration.getConfigurationName())) {
+            SolrClientUtils.uploadConfiguration(
+                getHost(),
+                getSolrPort(),
+                configuration.getConfigurationName(),
+                configuration.getSolrConfiguration(),
+                configuration.getSolrSchema()
+            );
+        }
+
+        SolrClientUtils.createCollection(
+            getHost(),
+            getSolrPort(),
+            configuration.getCollectionName(),
+            configuration.getConfigurationName()
+        );
+    }
+}