From 852d85cfab5aa5ecde0e851c12ed89d1e2602d94 Mon Sep 17 00:00:00 2001
From: yuxiqian <34335406+yuxiqian@users.noreply.github.com>
Date: Mon, 24 Jul 2023 14:12:24 +0800
Subject: [PATCH] [mongodb] [tests] Loosen E2E timeout limitations

 This closes #2334.

(cherry picked from commit 883b6846add4c24001879479b52dafb6933c4a4b)
---
 .../com/ververica/cdc/connectors/tests/MongoE2eITCase.java  | 5 +++--
 .../cdc/connectors/mongodb/utils/MongoDBContainer.java      | 6 ++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/flink-cdc-e2e-tests/src/test/java/com/ververica/cdc/connectors/tests/MongoE2eITCase.java b/flink-cdc-e2e-tests/src/test/java/com/ververica/cdc/connectors/tests/MongoE2eITCase.java
index 4f299d699e8..81703f2b9e0 100644
--- a/flink-cdc-e2e-tests/src/test/java/com/ververica/cdc/connectors/tests/MongoE2eITCase.java
+++ b/flink-cdc-e2e-tests/src/test/java/com/ververica/cdc/connectors/tests/MongoE2eITCase.java
@@ -100,7 +100,8 @@ public void before() {
                 new MongoDBContainer(NETWORK, MongoDBContainer.ShardingClusterRole.ROUTER)
                         .dependsOn(shard)
                         .withNetworkAliases(INTER_CONTAINER_MONGO_ALIAS)
-                        .withLogConsumer(new Slf4jLogConsumer(LOG));
+                        .withLogConsumer(new Slf4jLogConsumer(LOG))
+                        .withStartupTimeout(Duration.ofSeconds(120));
 
         Startables.deepStart(Stream.of(config)).join();
         Startables.deepStart(Stream.of(shard)).join();
@@ -237,7 +238,7 @@ public void testMongoDbCDC() throws Exception {
                 expectResult,
                 "mongodb_products_sink",
                 new String[] {"id", "name", "description", "weight"},
-                60000L);
+                150000L);
     }
 
     private Document productDocOf(String id, String name, String description, Double weight) {
diff --git a/flink-connector-mongodb-cdc/src/test/java/com/ververica/cdc/connectors/mongodb/utils/MongoDBContainer.java b/flink-connector-mongodb-cdc/src/test/java/com/ververica/cdc/connectors/mongodb/utils/MongoDBContainer.java
index 14e7055b6c7..6c957ef0d83 100644
--- a/flink-connector-mongodb-cdc/src/test/java/com/ververica/cdc/connectors/mongodb/utils/MongoDBContainer.java
+++ b/flink-connector-mongodb-cdc/src/test/java/com/ververica/cdc/connectors/mongodb/utils/MongoDBContainer.java
@@ -30,6 +30,7 @@
 import java.net.URL;
 import java.nio.file.Files;
 import java.nio.file.Paths;
+import java.time.Duration;
 import java.util.Random;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -101,6 +102,11 @@ public String getHostAndPort() {
         return String.format("%s:%s", getContainerIpAddress(), getMappedPort(MONGODB_PORT));
     }
 
+    @Override
+    public MongoDBContainer withStartupTimeout(Duration timeout) {
+        return (MongoDBContainer) super.withStartupTimeout(timeout);
+    }
+
     public void executeCommand(String command) {
         try {
             LOG.info("Executing mongo command: {}", command);