fix: ut bug

apache · Oct 10, 2024 · 92699c0 · 92699c0
1 parent ae19f54
commit 92699c0
Show file tree

Hide file tree

Showing 3 changed files with 72 additions and 0 deletions.
diff --git a/...est/java/org/apache/inlong/sdk/transform/process/function/encryption/TestMd5Function.java b/...est/java/org/apache/inlong/sdk/transform/process/function/encryption/TestMd5Function.java
@@ -19,16 +19,40 @@
 
 import org.apache.inlong.sdk.transform.decode.SourceDecoderFactory;
 import org.apache.inlong.sdk.transform.encode.SinkEncoderFactory;
+import org.apache.inlong.sdk.transform.pojo.CsvSourceInfo;
+import org.apache.inlong.sdk.transform.pojo.FieldInfo;
+import org.apache.inlong.sdk.transform.pojo.KvSinkInfo;
 import org.apache.inlong.sdk.transform.pojo.TransformConfig;
 import org.apache.inlong.sdk.transform.process.TransformProcessor;
+import org.apache.inlong.sdk.transform.process.converter.BooleanConverter;
 
 import org.junit.Assert;
 import org.junit.Test;
 
+import java.util.ArrayList;
 import java.util.List;
 
 public class TestMd5Function extends AbstractFunctionEncryptionTestBase {
 
+    private static final List<FieldInfo> srcFields = new ArrayList<>();
+    private static final List<FieldInfo> dstFields = new ArrayList<>();
+    private static final CsvSourceInfo csvSource;
+    private static final KvSinkInfo kvSink;
+
+    static {
+        for (int i = 1; i <= 5; i++) {
+            FieldInfo field = new FieldInfo();
+            field.setName("numeric" + i);
+            srcFields.add(field);
+        }
+        srcFields.add(new FieldInfo("booleanVal", new BooleanConverter()));
+        FieldInfo field = new FieldInfo();
+        field.setName("result");
+        dstFields.add(field);
+        csvSource = new CsvSourceInfo("UTF-8", '|', '\\', srcFields);
+        kvSink = new KvSinkInfo("UTF-8", dstFields);
+    }
+
     @Test
     public void testMd5Function() throws Exception {
         String transformSql = "select md5(numeric1) from source";

diff --git a/...st/java/org/apache/inlong/sdk/transform/process/function/encryption/TestSha2Function.java b/...st/java/org/apache/inlong/sdk/transform/process/function/encryption/TestSha2Function.java
@@ -19,17 +19,41 @@
 
 import org.apache.inlong.sdk.transform.decode.SourceDecoderFactory;
 import org.apache.inlong.sdk.transform.encode.SinkEncoderFactory;
+import org.apache.inlong.sdk.transform.pojo.CsvSourceInfo;
+import org.apache.inlong.sdk.transform.pojo.FieldInfo;
+import org.apache.inlong.sdk.transform.pojo.KvSinkInfo;
 import org.apache.inlong.sdk.transform.pojo.TransformConfig;
 import org.apache.inlong.sdk.transform.process.TransformProcessor;
+import org.apache.inlong.sdk.transform.process.converter.BooleanConverter;
 
 import org.junit.Assert;
 import org.junit.Test;
 
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 
 public class TestSha2Function extends AbstractFunctionEncryptionTestBase {
 
+    private static final List<FieldInfo> srcFields = new ArrayList<>();
+    private static final List<FieldInfo> dstFields = new ArrayList<>();
+    private static final CsvSourceInfo csvSource;
+    private static final KvSinkInfo kvSink;
+
+    static {
+        for (int i = 1; i <= 5; i++) {
+            FieldInfo field = new FieldInfo();
+            field.setName("numeric" + i);
+            srcFields.add(field);
+        }
+        srcFields.add(new FieldInfo("booleanVal", new BooleanConverter()));
+        FieldInfo field = new FieldInfo();
+        field.setName("result");
+        dstFields.add(field);
+        csvSource = new CsvSourceInfo("UTF-8", '|', '\\', srcFields);
+        kvSink = new KvSinkInfo("UTF-8", dstFields);
+    }
+
     @Test
     public void testSha2Function() throws Exception {
         String transformSql = null, data = null;

diff --git a/...est/java/org/apache/inlong/sdk/transform/process/function/encryption/TestShaFunction.java b/...est/java/org/apache/inlong/sdk/transform/process/function/encryption/TestShaFunction.java
@@ -19,17 +19,41 @@
 
 import org.apache.inlong.sdk.transform.decode.SourceDecoderFactory;
 import org.apache.inlong.sdk.transform.encode.SinkEncoderFactory;
+import org.apache.inlong.sdk.transform.pojo.CsvSourceInfo;
+import org.apache.inlong.sdk.transform.pojo.FieldInfo;
+import org.apache.inlong.sdk.transform.pojo.KvSinkInfo;
 import org.apache.inlong.sdk.transform.pojo.TransformConfig;
 import org.apache.inlong.sdk.transform.process.TransformProcessor;
+import org.apache.inlong.sdk.transform.process.converter.BooleanConverter;
 
 import org.junit.Assert;
 import org.junit.Test;
 
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 
 public class TestShaFunction extends AbstractFunctionEncryptionTestBase {
 
+    private static final List<FieldInfo> srcFields = new ArrayList<>();
+    private static final List<FieldInfo> dstFields = new ArrayList<>();
+    private static final CsvSourceInfo csvSource;
+    private static final KvSinkInfo kvSink;
+
+    static {
+        for (int i = 1; i <= 5; i++) {
+            FieldInfo field = new FieldInfo();
+            field.setName("numeric" + i);
+            srcFields.add(field);
+        }
+        srcFields.add(new FieldInfo("booleanVal", new BooleanConverter()));
+        FieldInfo field = new FieldInfo();
+        field.setName("result");
+        dstFields.add(field);
+        csvSource = new CsvSourceInfo("UTF-8", '|', '\\', srcFields);
+        kvSink = new KvSinkInfo("UTF-8", dstFields);
+    }
+
     @Test
     public void testShaFunction() throws Exception {
         String transformSql = null, data = null;