Fix DebeziumIO testWrongHost trying to connect real external address (#…

…31365)
apache · May 22, 2024 · 611676d · 611676d
1 parent 98b5d5b
commit 611676d
Showing 1 changed file with 13 additions and 20 deletions.
diff --git a/...o/debezium/src/test/java/org/apache/beam/io/debezium/DebeziumReadSchemaTransformTest.java b/...o/debezium/src/test/java/org/apache/beam/io/debezium/DebeziumReadSchemaTransformTest.java
@@ -188,26 +188,19 @@ public void testWrongPort() {
   }
 
   @Test
-  public void testWrongHost() throws Exception {
+  public void testWrongHost() {
     Pipeline readPipeline = Pipeline.create();
-    Exception ex =
-        assertThrows(
-            Exception.class,
-            () -> {
-              PCollectionRowTuple.empty(readPipeline)
-                  .apply(
-                      makePtransform(
-                          userName,
-                          password,
-                          database,
-                          databaseContainer.getMappedPort(port),
-                          "23.128.129.130"))
-                  .get("output");
-            });
-    Throwable lowestCause = ex.getCause();
-    while (lowestCause.getCause() != null) {
-      lowestCause = lowestCause.getCause();
-    }
-    assertThat(lowestCause.getMessage(), Matchers.containsString("Connection refused"));
+    assertThrows(
+        Exception.class,
+        () ->
+            PCollectionRowTuple.empty(readPipeline)
+                .apply(
+                    makePtransform(
+                        userName,
+                        password,
+                        database,
+                        databaseContainer.getMappedPort(port),
+                        "169.254.254.254"))
+                .get("output"));
   }
 }