This is an automated email from the ASF dual-hosted git repository.

jackie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-pinot.git


The following commit(s) were added to refs/heads/master by this push:
     new 8900ebb  Improve test coverage for TransformFunctionType (#5979)
8900ebb is described below

commit 8900ebb992f2872d379770f7196e245a6ffa37be
Author: Braden Groom <[email protected]>
AuthorDate: Fri Sep 4 12:00:47 2020 -0700

    Improve test coverage for TransformFunctionType (#5979)
---
 .../pinot/common/function/TransformFunctionType.java      |  2 +-
 .../pinot/common/function/TransformFunctionTypeTest.java  | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git 
a/pinot-common/src/main/java/org/apache/pinot/common/function/TransformFunctionType.java
 
b/pinot-common/src/main/java/org/apache/pinot/common/function/TransformFunctionType.java
index 0893916..76f6446 100644
--- 
a/pinot-common/src/main/java/org/apache/pinot/common/function/TransformFunctionType.java
+++ 
b/pinot-common/src/main/java/org/apache/pinot/common/function/TransformFunctionType.java
@@ -93,7 +93,7 @@ public enum TransformFunctionType {
     String upperCaseFunctionName = functionName.toUpperCase();
     try {
       return TransformFunctionType.valueOf(upperCaseFunctionName);
-    } catch (Exception e) {
+    } catch (IllegalArgumentException e) {
       if (FunctionRegistry.containsFunction(functionName)) {
         return SCALAR;
       }
diff --git 
a/pinot-common/src/test/java/org/apache/pinot/common/function/TransformFunctionTypeTest.java
 
b/pinot-common/src/test/java/org/apache/pinot/common/function/TransformFunctionTypeTest.java
index 64fe7e3..fef0f38 100644
--- 
a/pinot-common/src/test/java/org/apache/pinot/common/function/TransformFunctionTypeTest.java
+++ 
b/pinot-common/src/test/java/org/apache/pinot/common/function/TransformFunctionTypeTest.java
@@ -26,7 +26,22 @@ import static org.testng.Assert.assertEquals;
 public class TransformFunctionTypeTest {
 
   @Test
+  public void testSimpleLookupFunction() {
+    assertEquals(TransformFunctionType.getTransformFunctionType("add"), 
TransformFunctionType.ADD);
+  }
+
+  @Test
   public void testScalarFunction() {
     
assertEquals(TransformFunctionType.getTransformFunctionType("toEpochSeconds"), 
TransformFunctionType.SCALAR);
   }
+
+  @Test
+  public void testSnakeCaseLookupFunction() {
+    
assertEquals(TransformFunctionType.getTransformFunctionType("json_extract_scalar"),
 TransformFunctionType.JSONEXTRACTSCALAR);
+  }
+
+  @Test(expectedExceptions = IllegalArgumentException.class)
+  public void invalidLookup() {
+    TransformFunctionType.getTransformFunctionType("foo_bar");
+  }
 }


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to