set testContainer StartupTimeout

apache · Mar 14, 2024 · 5ed0a7d · 5ed0a7d
1 parent 0deb7a7
commit 5ed0a7d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/...mmon/src/test/java/com/datastrato/gravitino/integration/test/container/BaseContainer.java b/...mmon/src/test/java/com/datastrato/gravitino/integration/test/container/BaseContainer.java
@@ -97,6 +97,10 @@ protected void withLogConsumer(Consumer<OutputFrame> logConsumer) {
     container.withLogConsumer(logConsumer);
   }
 
+  protected void withStartupTimeout(Duration duration) {
+    container.withStartupTimeout(duration);
+  }
+
   // This method is used to get the expose port number of the container.
   public Integer getMappedPort(int exposedPort) {
     return container.getMappedPort(exposedPort);

diff --git a/...mon/src/test/java/com/datastrato/gravitino/integration/test/container/DorisContainer.java b/...mon/src/test/java/com/datastrato/gravitino/integration/test/container/DorisContainer.java
@@ -11,6 +11,7 @@
 import java.sql.DriverManager;
 import java.sql.ResultSet;
 import java.sql.Statement;
+import java.time.Duration;
 import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
@@ -48,6 +49,7 @@ protected DorisContainer(
   protected void setupContainer() {
     super.setupContainer();
     withLogConsumer(new PrintingContainerLog(format("%-14s| ", "DorisContainer")));
+    withStartupTimeout(Duration.ofMinutes(5));
   }
 
   @Override