Add method for getting Weaviate's gRPC port (#8431)

testcontainers · Mar 9, 2024 · 5bb9607 · 5bb9607
1 parent d3b0df2
commit 5bb9607
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/modules/weaviate/build.gradle b/modules/weaviate/build.gradle
@@ -4,5 +4,5 @@ dependencies {
     api project(':testcontainers')
 
     testImplementation 'org.assertj:assertj-core:3.25.1'
-    testImplementation 'io.weaviate:client:4.5.1'
+    testImplementation 'io.weaviate:client:4.6.0'
 }
diff --git a/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java b/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java
@@ -33,4 +33,8 @@ public WeaviateContainer(DockerImageName dockerImageName) {
     public String getHttpHostAddress() {
         return getHost() + ":" + getMappedPort(8080);
     }
+
+    public String getGrpcHostAddress() {
+        return getHost() + ":" + getMappedPort(50051);
+    }
 }
diff --git a/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java b/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java
@@ -13,13 +13,15 @@ public class WeaviateContainerTest {
     @Test
     public void test() {
         try ( // container {
-            WeaviateContainer weaviate = new WeaviateContainer("semitechnologies/weaviate:1.22.4")
+            WeaviateContainer weaviate = new WeaviateContainer("semitechnologies/weaviate:1.24.1")
             // }
         ) {
             weaviate.start();
-            WeaviateClient client = new WeaviateClient(new Config("http", weaviate.getHttpHostAddress()));
+            Config config = new Config("http", weaviate.getHttpHostAddress());
+            config.setGRPCHost(weaviate.getGrpcHostAddress());
+            WeaviateClient client = new WeaviateClient(config);
             Result<Meta> meta = client.misc().metaGetter().run();
-            assertThat(meta.getResult().getVersion()).isEqualTo("1.22.4");
+            assertThat(meta.getResult().getVersion()).isEqualTo("1.24.1");
         }
     }
 }