1717 */
1818public class WeaviateContainer extends GenericContainer <WeaviateContainer > {
1919
20+ private static final int HTTP_PORT = 8080 ;
21+
22+ private static final int GRPC_PORT = 50051 ;
23+
2024 private static final DockerImageName DEFAULT_WEAVIATE_IMAGE = DockerImageName .parse (
2125 "cr.weaviate.io/semitechnologies/weaviate"
2226 );
@@ -30,17 +34,25 @@ public WeaviateContainer(String dockerImageName) {
3034 public WeaviateContainer (DockerImageName dockerImageName ) {
3135 super (dockerImageName );
3236 dockerImageName .assertCompatibleWith (DEFAULT_WEAVIATE_IMAGE , DOCKER_HUB_WEAVIATE_IMAGE );
33- withExposedPorts (8080 , 50051 );
37+ withExposedPorts (HTTP_PORT , GRPC_PORT );
3438 withEnv ("AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED" , "true" );
3539 withEnv ("PERSISTENCE_DATA_PATH" , "/var/lib/weaviate" );
36- waitingFor (Wait .forHttp ("/v1/.well-known/ready" ).forPort (8080 ).forStatusCode (200 ));
40+ waitingFor (Wait .forHttp ("/v1/.well-known/ready" ).forPort (HTTP_PORT ).forStatusCode (200 ));
3741 }
3842
3943 public String getHttpHostAddress () {
40- return getHost () + ":" + getMappedPort (8080 );
44+ return getHost () + ":" + getHttpPort ();
45+ }
46+
47+ public Integer getHttpPort () {
48+ return getMappedPort (HTTP_PORT );
4149 }
4250
4351 public String getGrpcHostAddress () {
44- return getHost () + ":" + getMappedPort (50051 );
52+ return getHost () + ":" + getGrpcPort ();
53+ }
54+
55+ public Integer getGrpcPort () {
56+ return getMappedPort (GRPC_PORT );
4557 }
4658}
0 commit comments