ADD testIntervalSQL_OracleToPg test

ZhengguanLi · Jun 27, 2024 · 3b0e5d2 · 3b0e5d2
1 parent 3b47d4f
commit 3b0e5d2
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/core/src/test/java/com/alibaba/druid/sql/parser/PGIntervalSQLTest.java b/core/src/test/java/com/alibaba/druid/sql/parser/PGIntervalSQLTest.java
@@ -1,9 +1,13 @@
 package com.alibaba.druid.sql.parser;
 
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
 import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGSelectStatement;
 import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.test.TestUtils;
 import junit.framework.TestCase;
-import org.junit.Assert;
+
+import java.util.List;
 
 /**
  * Created by tianzhen.wtz on 2014/12/26 0026 20:44.
@@ -29,8 +33,17 @@ public void testIntervalSQL() {
     private void equal(String targetSql, String resultSql) {
         PGSQLStatementParser parser = new PGSQLStatementParser(targetSql);
         PGSelectStatement statement = parser.parseSelect();
-        Assert.assertTrue(statement.toString().equals(resultSql));
+        assertEquals(statement.toString(), resultSql);
 
     }
 
+    public void testIntervalSQL_OracleToPg() {
+        String sql = "SELECT (SYSTIMESTAMP - order_date) DAY(9) TO SECOND from orders WHERE order_id = 2458";
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        assertEquals(1, statementList.size());
+
+        String output = TestUtils.outputPg(statementList);
+        assertEquals("SELECT (SYSTIMESTAMP - order_date) DAY(9) TO SECOND\nFROM orders\nWHERE order_id = 2458", output);
+    }
 }
diff --git a/core/src/test/java/com/alibaba/druid/sql/test/TestUtils.java b/core/src/test/java/com/alibaba/druid/sql/test/TestUtils.java
@@ -24,6 +24,7 @@
 
 import com.alibaba.druid.sql.ast.SQLStatement;
 import com.alibaba.druid.sql.dialect.oracle.visitor.OracleOutputVisitor;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
 import com.alibaba.druid.sql.dialect.sqlserver.visitor.SQLServerOutputVisitor;
 import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
 
@@ -38,6 +39,17 @@ public static String outputOracle(List<SQLStatement> stmtList) {
 
         return out.toString();
     }
+
+    public static String outputPg(List<SQLStatement> stmtList) {
+        StringBuilder out = new StringBuilder();
+        PGOutputVisitor visitor = new PGOutputVisitor(out);
+
+        for (SQLStatement stmt : stmtList) {
+            stmt.accept(visitor);
+        }
+
+        return out.toString();
+    }
 
     public static String outputSqlServer(List<SQLStatement> stmtList) {
         StringBuilder out = new StringBuilder();