added TableFunction alias suppurt

added TableFunction unit tests

Author: tfedkiv

src/main/java/net/sf/jsqlparser/statement/select/TableFunction.java

@@ -28,6 +28,7 @@ public class TableFunction implements FromItem {
 
     private String name;
     private ExpressionList parameters;
+    private Alias alias;
 
     /**
      * The name of he procedure, i.e. "UNWIND_TOP"
@@ -62,22 +63,21 @@ public void accept(FromItemVisitor fromItemVisitor) {
 
     @Override
     public Alias getAlias() {
-        throw new UnsupportedOperationException();
+        return alias;
     }
 
     @Override
     public void setAlias(Alias alias) {
-        throw new UnsupportedOperationException();
+        this.alias = alias;
     }
 
     @Override
     public Pivot getPivot() {
-        throw new UnsupportedOperationException();
+        return null;
     }
 
     @Override
     public void setPivot(Pivot pivot) {
-        throw new UnsupportedOperationException();
     }
 
     @Override
@@ -90,9 +90,11 @@ public String toString() {
             params = "()";
         }
 
-        String ans = name + "" + params + "";
-
-        return ans;
+        String string = name + "" + params + "";
+        if (alias != null) {
+            string += alias.toString();
+        }
+        return string;
     }
 
 }

src/test/java/net/sf/jsqlparser/test/select/SelectTest.java

@@ -2005,5 +2005,60 @@ public void testOracleHintExpression() throws JSQLParserException {
         PlainSelect plainSelect = (PlainSelect) ((Select) parsed).getSelectBody();
         assertExpressionCanBeDeparsedAs(plainSelect.getOracleHint(), "--+ HINT\n");
     }
-        
+
+    public void testTableFunctionWithNoParams() throws Exception {
+        final String statement = "SELECT f2 FROM SOME_FUNCTION()";
+        Select select = (Select) parserManager.parse(new StringReader(statement));
+        PlainSelect plainSelect = (PlainSelect) select.getSelectBody();
+
+        assertTrue(plainSelect.getFromItem() instanceof TableFunction);
+        TableFunction fromItem = (TableFunction) plainSelect.getFromItem();
+        assertEquals("SOME_FUNCTION", fromItem.getName());
+        assertNull(fromItem.getParameters());
+        assertNull(fromItem.getAlias());
+        assertStatementCanBeDeparsedAs(select, statement);
+    }
+
+    public void testTableFunctionWithParams() throws Exception {
+        final String statement = "SELECT f2 FROM SOME_FUNCTION(1, 'val')";
+        Select select = (Select) parserManager.parse(new StringReader(statement));
+        PlainSelect plainSelect = (PlainSelect) select.getSelectBody();
+
+        assertTrue(plainSelect.getFromItem() instanceof TableFunction);
+        TableFunction fromItem = (TableFunction) plainSelect.getFromItem();
+        assertEquals("SOME_FUNCTION", fromItem.getName());
+
+        // verify params
+        assertNotNull(fromItem.getParameters());
+        List<Expression> expressions = fromItem.getParameters().getExpressions();
+        assertEquals(2, expressions.size());
+
+        Expression firstParam = expressions.get(0);
+        assertNotNull(firstParam);
+        assertTrue(firstParam instanceof LongValue);
+        assertEquals(1l, ((LongValue) firstParam).getValue());
+
+        Expression secondParam = expressions.get(1);
+        assertNotNull(secondParam);
+        assertTrue(secondParam instanceof StringValue);
+        assertEquals("val", ((StringValue) secondParam).getValue());
+
+        assertNull(fromItem.getAlias());
+        assertStatementCanBeDeparsedAs(select, statement);
+    }
+
+    public void testTableFunctionWithAlias() throws Exception {
+        final String statement = "SELECT f2 FROM SOME_FUNCTION() AS z";
+        Select select = (Select) parserManager.parse(new StringReader(statement));
+        PlainSelect plainSelect = (PlainSelect) select.getSelectBody();
+
+        assertTrue(plainSelect.getFromItem() instanceof TableFunction);
+        TableFunction fromItem = (TableFunction) plainSelect.getFromItem();
+
+        assertEquals("SOME_FUNCTION", fromItem.getName());
+        assertNull(fromItem.getParameters());
+        assertNotNull(fromItem.getAlias());
+        assertEquals("z", fromItem.getAlias().getName());
+        assertStatementCanBeDeparsedAs(select, statement);
+    }
 }