Allow establishing docker container startup dependency

trinodb · Sep 30, 2020 · 6e9014a · 6e9014a
1 parent 58f6742
commit 6e9014a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...uct-tests-launcher/src/main/java/io/prestosql/tests/product/launcher/env/Environment.java b/...uct-tests-launcher/src/main/java/io/prestosql/tests/product/launcher/env/Environment.java
@@ -320,6 +320,15 @@ public String getEnvironmentName()
             return name;
         }
 
+        public Builder containerDependsOn(String container, String dependencyContainer)
+        {
+            checkState(containers.containsKey(container), "Container with name %s is not registered", name);
+            checkState(containers.containsKey(dependencyContainer), "Dependency container with name %s is not registered", dependencyContainer);
+            containers.get(container).dependsOn(containers.get(dependencyContainer));
+
+            return this;
+        }
+
         public Builder addContainers(DockerContainer... containers)
         {
             Arrays.stream(containers)

diff --git a/...ct-tests-launcher/src/main/java/io/prestosql/tests/product/launcher/env/common/Kafka.java b/...ct-tests-launcher/src/main/java/io/prestosql/tests/product/launcher/env/common/Kafka.java
@@ -40,7 +40,8 @@ public Kafka(PortBinder portBinder)
     @Override
     public void extendEnvironment(Environment.Builder builder)
     {
-        builder.addContainers(createZookeeper(), createKafka());
+        builder.addContainers(createZookeeper(), createKafka())
+                .containerDependsOn("kafka", "zookeeper");
     }
 
     @SuppressWarnings("resource")