fix failing tests

Signed-off-by: Jeffrey Kinard <[email protected]>
apache · Dec 20, 2024 · 09f56e2 · 09f56e2
1 parent 295ae5a
commit 09f56e2
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 27 deletions.
diff --git a/...a/io/jdbc/src/main/java/org/apache/beam/sdk/io/jdbc/JdbcWriteSchemaTransformProvider.java b/...a/io/jdbc/src/main/java/org/apache/beam/sdk/io/jdbc/JdbcWriteSchemaTransformProvider.java
@@ -337,7 +337,7 @@ public abstract static class JdbcWriteSchemaTransformConfiguration implements Se
     public abstract String getWriteStatement();
 
     public void validate() {
-      validate("JDBC");
+      validate("");
     }
 
     public void validate(String jdbcType) throws IllegalArgumentException {
@@ -360,7 +360,8 @@ public void validate(String jdbcType) throws IllegalArgumentException {
       }
       if (jdbcTypePresent
           && !JDBC_DRIVER_MAP.containsKey(Objects.requireNonNull(jdbcType).toLowerCase())) {
-        throw new IllegalArgumentException("JDBC type must be one of " + JDBC_DRIVER_MAP.keySet());
+        throw new IllegalArgumentException(
+            "JDBC type must be one of " + JDBC_DRIVER_MAP.keySet() + " but was " + jdbcType);
       }
 
       boolean writeStatementPresent =

diff --git a/...o/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcReadSchemaTransformProviderTest.java b/...o/jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcReadSchemaTransformProviderTest.java
@@ -122,17 +122,6 @@ public void testInvalidReadSchemaOptions() {
               .build()
               .validate();
         });
-    assertThrows(
-        IllegalArgumentException.class,
-        () -> {
-          JdbcReadSchemaTransformProvider.JdbcReadSchemaTransformConfiguration.builder()
-              .setJdbcUrl("JdbcUrl")
-              .setLocation("Location")
-              .setDriverClassName("ClassName")
-              .setJdbcType((String) JDBC_DRIVER_MAP.keySet().toArray()[0])
-              .build()
-              .validate();
-        });
   }
 
   @Test

diff --git a/.../jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcWriteSchemaTransformProviderTest.java b/.../jdbc/src/test/java/org/apache/beam/sdk/io/jdbc/JdbcWriteSchemaTransformProviderTest.java
@@ -122,17 +122,6 @@ public void testInvalidWriteSchemaOptions() {
               .build()
               .validate();
         });
-    assertThrows(
-        IllegalArgumentException.class,
-        () -> {
-          JdbcWriteSchemaTransformProvider.JdbcWriteSchemaTransformConfiguration.builder()
-              .setJdbcUrl("JdbcUrl")
-              .setLocation("Location")
-              .setDriverClassName("ClassName")
-              .setJdbcType((String) JDBC_DRIVER_MAP.keySet().toArray()[0])
-              .build()
-              .validate();
-        });
   }
 
   @Test

diff --git a/sdks/python/apache_beam/yaml/generate_yaml_docs.py b/sdks/python/apache_beam/yaml/generate_yaml_docs.py
@@ -205,9 +205,9 @@ def add_transform_links(transform, description, provider_list):
 
   meanwhile,
   ```
-    - type: MyTransform
-      config:
-        ...
+  - type: MyTransform
+    config:
+      ...
   ```
   Would remain unchanged.