From 5d97fb9c316ef485fbfb3d80dd1c3ab1b5137185 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edd=C3=BA=20Mel=C3=A9ndez?= Date: Thu, 16 Apr 2026 00:51:36 -0500 Subject: [PATCH] Add getHttpPort and getGrpcPort methods in WeaviateContainer --- .../weaviate/WeaviateContainer.java | 20 +++++++++++++++---- .../weaviate/WeaviateContainerTest.java | 8 ++++---- 2 files changed, 20 insertions(+), 8 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 index 86aff57ae5d..dbc18a49a22 100644 --- a/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java +++ b/modules/weaviate/src/main/java/org/testcontainers/weaviate/WeaviateContainer.java @@ -17,6 +17,10 @@ */ public class WeaviateContainer extends GenericContainer { + private static final int HTTP_PORT = 8080; + + private static final int GRPC_PORT = 50051; + private static final DockerImageName DEFAULT_WEAVIATE_IMAGE = DockerImageName.parse( "cr.weaviate.io/semitechnologies/weaviate" ); @@ -30,17 +34,25 @@ public WeaviateContainer(String dockerImageName) { public WeaviateContainer(DockerImageName dockerImageName) { super(dockerImageName); dockerImageName.assertCompatibleWith(DEFAULT_WEAVIATE_IMAGE, DOCKER_HUB_WEAVIATE_IMAGE); - withExposedPorts(8080, 50051); + withExposedPorts(HTTP_PORT, GRPC_PORT); withEnv("AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED", "true"); withEnv("PERSISTENCE_DATA_PATH", "/var/lib/weaviate"); - waitingFor(Wait.forHttp("/v1/.well-known/ready").forPort(8080).forStatusCode(200)); + waitingFor(Wait.forHttp("/v1/.well-known/ready").forPort(HTTP_PORT).forStatusCode(200)); } public String getHttpHostAddress() { - return getHost() + ":" + getMappedPort(8080); + return getHost() + ":" + getHttpPort(); + } + + public Integer getHttpPort() { + return getMappedPort(HTTP_PORT); } public String getGrpcHostAddress() { - return getHost() + ":" + getMappedPort(50051); + return getHost() + ":" + getGrpcPort(); + } + + public Integer getGrpcPort() { + return getMappedPort(GRPC_PORT); } } diff --git a/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java b/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java index d103970053c..098a1561913 100644 --- a/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java +++ b/modules/weaviate/src/test/java/org/testcontainers/weaviate/WeaviateContainerTest.java @@ -26,9 +26,9 @@ void testWeaviate() throws Exception { return conn .scheme("http") .httpHost(weaviate.getHost()) - .httpPort(weaviate.getMappedPort(8080)) + .httpPort(weaviate.getHttpPort()) .grpcHost(weaviate.getHost()) - .grpcPort(weaviate.getMappedPort(50051)); + .grpcPort(weaviate.getGrpcPort()); }) ) { InstanceMetadata meta = client.meta(); @@ -56,9 +56,9 @@ void testWeaviateWithModules() throws Exception { return conn .scheme("http") .httpHost(weaviate.getHost()) - .httpPort(weaviate.getMappedPort(8080)) + .httpPort(weaviate.getHttpPort()) .grpcHost(weaviate.getHost()) - .grpcPort(weaviate.getMappedPort(50051)); + .grpcPort(weaviate.getGrpcPort()); }) ) { InstanceMetadata meta = client.meta();