Add ChromaDB module (#8336)

Author: eddumelendez

.github/ISSUE_TEMPLATE/bug_report.yaml

@@ -17,6 +17,7 @@ body:
         - ActiveMQ
         - Azure
         - Cassandra
+        - ChromaDB
         - Clickhouse
         - CockroachDB
         - Consul

.github/ISSUE_TEMPLATE/enhancement.yaml

@@ -17,6 +17,7 @@ body:
         - ActiveMQ
         - Azure
         - Cassandra
+        - ChromaDB
         - Clickhouse
         - CockroachDB
         - Consul

.github/ISSUE_TEMPLATE/feature.yaml

@@ -17,6 +17,7 @@ body:
         - ActiveMQ
         - Azure
         - Cassandra
+        - ChromaDB
         - Clickhouse
         - CockroachDB
         - CrateDB

.github/dependabot.yml

@@ -51,6 +51,11 @@ updates:
     ignore:
       - dependency-name: "io.dropwizard.metrics:metrics-core"
         update-types: [ "version-update:semver-major" ]
+  - package-ecosystem: "gradle"
+    directory: "/modules/chromadb"
+    schedule:
+      interval: "weekly"
+    open-pull-requests-limit: 10
   - package-ecosystem: "gradle"
     directory: "/modules/clickhouse"
     schedule:

.github/labeler.yml

@@ -27,6 +27,10 @@
   - changed-files:
     - any-glob-to-any-file:
       - modules/cassandra/**/*
+"modules/chromadb":
+  - changed-files:
+    - any-glob-to-any-file:
+      - modules/chromadb/**/*
 "modules/clickhouse":
   - changed-files:
     - any-glob-to-any-file:

.github/settings.yml

@@ -115,6 +115,9 @@ labels:
   - name: modules/cassandra
     color: '#006b75'
 
+  - name: modules/chromadb
+    color: '#006b75'
+
   - name: modules/clickhouse
     color: '#006b75'
 

docs/modules/chromadb.md

@@ -0,0 +1,30 @@
+# ChromaDB
+
+Testcontainers module for [ChromaDB](https://registry.hub.docker.com/r/chromadb/chroma)
+
+## ChromaDB's usage examples
+
+You can start a ChromaDB container instance from any Java application by using:
+
+<!--codeinclude-->
+[Default ChromaDB container](../../modules/chromadb/src/test/java/org/testcontainers/chromadb/ChromaDBContainerTest.java) inside_block:container
+<!--/codeinclude-->
+
+## Adding this module to your project dependencies
+
+Add the following dependency to your `pom.xml`/`build.gradle` file:
+
+=== "Gradle"
+```groovy
+testImplementation "org.testcontainers:chromadb:{{latest_version}}"
+```
+
+=== "Maven"
+```xml
+<dependency>
+<groupId>org.testcontainers</groupId>
+<artifactId>chromadb</artifactId>
+<version>{{latest_version}}</version>
+<scope>test</scope>
+</dependency>
+```

mkdocs.yml

@@ -75,6 +75,7 @@ nav:
                 - modules/databases/yugabytedb.md
           - modules/activemq.md
           - modules/azure.md
+          - modules/chromadb.md
           - modules/consul.md
           - modules/docker_compose.md
           - modules/elasticsearch.md

modules/chromadb/build.gradle

@@ -0,0 +1,8 @@
+description = "Testcontainers :: ChromaDB"
+
+dependencies {
+    api project(':testcontainers')
+
+    testImplementation 'org.assertj:assertj-core:3.25.1'
+    testImplementation 'io.rest-assured:rest-assured:5.4.0'
+}

modules/chromadb/src/main/java/org/testcontainers/chromadb/ChromaDBContainer.java

@@ -0,0 +1,34 @@
+package org.testcontainers.chromadb;
+
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.wait.strategy.Wait;
+import org.testcontainers.utility.DockerImageName;
+
+/**
+ * Testcontainers implementation of ChromaDB.
+ * <p>
+ * Supported images: {@code chromadb/chroma}, {@code ghcr.io/chroma-core/chroma}
+ * <p>
+ * Exposed ports: 8000
+ */
+public class ChromaDBContainer extends GenericContainer<ChromaDBContainer> {
+
+    private static final DockerImageName DEFAULT_DOCKER_IMAGE = DockerImageName.parse("chromadb/chroma");
+
+    private static final DockerImageName GHCR_DOCKER_IMAGE = DockerImageName.parse("ghcr.io/chroma-core/chroma");
+
+    public ChromaDBContainer(String dockerImageName) {
+        this(DockerImageName.parse(dockerImageName));
+    }
+
+    public ChromaDBContainer(DockerImageName dockerImageName) {
+        super(dockerImageName);
+        dockerImageName.assertCompatibleWith(DEFAULT_DOCKER_IMAGE, GHCR_DOCKER_IMAGE);
+        withExposedPorts(8000);
+        waitingFor(Wait.forHttp("/api/v1/heartbeat"));
+    }
+
+    public String getEndpoint() {
+        return "http://" + getHost() + ":" + getFirstMappedPort();
+    }
+}