Update TemplateDefinitionsTest

GoogleCloudPlatform · May 23, 2024 · b7e5c3d · b7e5c3d
1 parent f710ec3
commit b7e5c3d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/...-plugin/src/test/java/com/google/cloud/teleport/plugin/model/TemplateDefinitionsTest.java b/...-plugin/src/test/java/com/google/cloud/teleport/plugin/model/TemplateDefinitionsTest.java
@@ -66,6 +66,9 @@ public void testSampleAtoBOk() {
     ImageSpecParameter inputKafkaTopic = metadata.getParameter("inputKafkaTopic").get();
     assertEquals(ImageSpecParameterType.KAFKA_TOPIC, inputKafkaTopic.getParamType());
 
+    ImageSpecParameter javascriptUdfFile = metadata.getParameter("javascriptUdfFile").get();
+    assertEquals(ImageSpecParameterType.JAVASCRIPT_UDF_FILE, javascriptUdfFile.getParamType());
+
     ImageSpecParameter logical = metadata.getParameter("logical").get();
     assertEquals(ImageSpecParameterType.BOOLEAN, logical.getParamType());
     assertEquals("^(true|false)$", logical.getRegexes().get(0));

diff --git a/plugins/core-plugin/src/test/java/com/google/cloud/teleport/plugin/sample/AtoBOk.java b/plugins/core-plugin/src/test/java/com/google/cloud/teleport/plugin/sample/AtoBOk.java
@@ -98,6 +98,13 @@ public interface AtoBOptions {
     @Default.Boolean(false)
     Boolean getParamWithGroupName();
 
+    @TemplateParameter.JavascriptUdfFile(
+        order = 16,
+        description = "JavaScript UDF in Cloud Storage",
+        helpText = "JavaScript UDF in Cloud Storage")
+    @Default.String("")
+    String javascriptUdfFile();
+
     @TemplateParameter.Text(
         order = 9,
         parentName = "paramWithGroupName",