Add test for CreateTableAsSelect

trinodb · Jul 5, 2023 · 7e138ca · 7e138ca
1 parent dcee3ae
commit 7e138ca
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/core/trino-parser/src/test/java/io/trino/sql/TestSqlFormatter.java b/core/trino-parser/src/test/java/io/trino/sql/TestSqlFormatter.java
@@ -14,20 +14,26 @@
 package io.trino.sql;
 
 import com.google.common.collect.ImmutableList;
+import io.trino.sql.tree.AllColumns;
 import io.trino.sql.tree.ColumnDefinition;
 import io.trino.sql.tree.CreateTable;
+import io.trino.sql.tree.CreateTableAsSelect;
 import io.trino.sql.tree.ExecuteImmediate;
 import io.trino.sql.tree.GenericDataType;
 import io.trino.sql.tree.Identifier;
 import io.trino.sql.tree.LongLiteral;
 import io.trino.sql.tree.NodeLocation;
 import io.trino.sql.tree.QualifiedName;
+import io.trino.sql.tree.Query;
 import io.trino.sql.tree.StringLiteral;
 import org.junit.jupiter.api.Test;
 
 import java.util.Optional;
 import java.util.function.BiFunction;
 
+import static io.trino.sql.QueryUtil.selectList;
+import static io.trino.sql.QueryUtil.simpleQuery;
+import static io.trino.sql.QueryUtil.table;
 import static io.trino.sql.SqlFormatter.formatSql;
 import static io.trino.sql.tree.SaveMode.FAIL;
 import static org.assertj.core.api.Assertions.assertThat;
@@ -83,6 +89,28 @@ public void testCreateTable()
                 .isEqualTo(String.format(createTableSql, "\"exists\"", "\"exists\""));
     }
 
+    @Test
+    public void testCreateTableAsSelect()
+    {
+        BiFunction<String, String, CreateTableAsSelect> createTableAsSelect = (tableName, columnName) -> {
+            Query query = simpleQuery(selectList(new AllColumns()), table(QualifiedName.of("t")));
+            return new CreateTableAsSelect(
+                    QualifiedName.of(ImmutableList.of(new Identifier(tableName, false))),
+                    query,
+                    FAIL,
+                    ImmutableList.of(),
+                    true,
+                    Optional.of(ImmutableList.of(new Identifier(columnName, false))),
+                    Optional.empty());
+        };
+        String createTableSql = "CREATE TABLE %s( %s ) AS SELECT *\nFROM\n  t\n";
+
+        assertThat(formatSql(createTableAsSelect.apply("table_name", "column_name")))
+                .isEqualTo(String.format(createTableSql, "table_name", "column_name"));
+        assertThat(formatSql(createTableAsSelect.apply("exists", "exists")))
+                .isEqualTo(String.format(createTableSql, "\"exists\"", "\"exists\""));
+    }
+
     @Test
     public void testExecuteImmediate()
     {