Weaviate: Support new container registry (#8512)

Fixes gh-8511 Signed-off-by: Thomas Vitale <ThomasVitale@users.noreply.github.com>
testcontainers · Apr 5, 2024 · 5969b0f · 5969b0f
1 parent 9d6e765
commit 5969b0f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java b/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java
@@ -6,7 +6,7 @@
 /**
  * Testcontainers implementation of Weaviate.
  * <p>
- * Supported image: {@code semitechnologies/weaviate}
+ * Supported images: {@code cr.weaviate.io/semitechnologies/weaviate}, {@code semitechnologies/weaviate}
  * <p>
  * Exposed ports:
  * <ul>
@@ -16,15 +16,19 @@
  */
 public class WeaviateContainer extends GenericContainer<WeaviateContainer> {
 
-    private static final String WEAVIATE_IMAGE = "semitechnologies/weaviate";
+    private static final DockerImageName DEFAULT_WEAVIATE_IMAGE = DockerImageName.parse(
+        "cr.weaviate.io/semitechnologies/weaviate"
+    );
+
+    private static final DockerImageName DOCKER_HUB_WEAVIATE_IMAGE = DockerImageName.parse("semitechnologies/weaviate");
 
     public WeaviateContainer(String dockerImageName) {
         this(DockerImageName.parse(dockerImageName));
     }
 
     public WeaviateContainer(DockerImageName dockerImageName) {
         super(dockerImageName);
-        dockerImageName.assertCompatibleWith(DockerImageName.parse(WEAVIATE_IMAGE));
+        dockerImageName.assertCompatibleWith(DEFAULT_WEAVIATE_IMAGE, DOCKER_HUB_WEAVIATE_IMAGE);
         withExposedPorts(8080, 50051);
         withEnv("AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED", "true");
         withEnv("PERSISTENCE_DATA_PATH", "/var/lib/weaviate");

diff --git a/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java b/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java
@@ -19,7 +19,7 @@ public class WeaviateContainerTest {
     @Test
     public void testWeaviate() {
         try ( // container {
-            WeaviateContainer weaviate = new WeaviateContainer("semitechnologies/weaviate:1.24.5")
+            WeaviateContainer weaviate = new WeaviateContainer("cr.weaviate.io/semitechnologies/weaviate:1.24.5")
             // }
         ) {
             weaviate.start();