Add SQL formatting tests for ALTER TABLE and DROP TABLE statements

Author: xterao

src/test/kotlin/org/domaframework/doma/intellij/formatter/SqlFormatterTest.kt

@@ -278,6 +278,22 @@ class SqlFormatterTest : BasePlatformTestCase() {
         formatSqlFile("OrderByGroupWithConditionDirective.sql", "OrderByGroupWithConditionDirective$formatDataPrefix.sql")
     }
 
+    fun testAlterTableAddColumnFormatter() {
+        formatSqlFile("AlterTableAddColumn.sql", "AlterTableAddColumn$formatDataPrefix.sql")
+    }
+
+    fun testAlterTableAddForeignFormatter() {
+        formatSqlFile("AlterTableAddForeign.sql", "AlterTableAddForeign$formatDataPrefix.sql")
+    }
+
+    fun testDropTableFormatter() {
+        formatSqlFile("DropTable.sql", "DropTable$formatDataPrefix.sql")
+    }
+
+    fun testDropTableIfExistsFormatter() {
+        formatSqlFile("DropTableIfExists.sql", "DropTableIfExists$formatDataPrefix.sql")
+    }
+
     private fun formatSqlFile(
         beforeFile: String,
         afterFile: String,

src/test/testData/sql/formatter/AlterTableAddColumn.sql

@@ -0,0 +1,9 @@
+ ALTER TABLE /*# tableName */
+       /*%for column : columns */
+       ADD COLUMN 
+           IF NOT EXISTS /*# column.name */ /*# column.type */
+           /*%if column_has_next */
+            ,
+           /*%end */
+       /*%end */
+DROP COLUMN /*# column2.name */ /*# column2.type */

src/test/testData/sql/formatter/AlterTableAddColumn_format.sql

@@ -0,0 +1,8 @@
+ ALTER TABLE /*# tableName */
+         /*%for column : columns */
+         ADD COLUMN IF NOT EXISTS /*# column.name */ /*# column.type */
+           /*%if column_has_next */
+           ,
+           /*%end */
+         /*%end */
+         DROP COLUMN /*# column2.name */ /*# column2.type */

src/test/testData/sql/formatter/AlterTableAddForeign.sql

@@ -0,0 +1,4 @@
+ALTER TABLE employee
+        ADD FOREIGN KEY (employee_id) REFERENCES users(user_iad)
+ON 
+UPDATE RESTRICT

src/test/testData/sql/formatter/AlterTableAddForeign_format.sql

@@ -0,0 +1,4 @@
+ALTER TABLE employee
+        ADD FOREIGN KEY (employee_id)
+        REFERENCES users(user_iad)
+        ON UPDATE RESTRICT

src/test/testData/sql/formatter/DropTable.sql

@@ -0,0 +1,2 @@
+Drop TABLE
+employees

src/test/testData/sql/formatter/DropTableIfExists.sql

@@ -0,0 +1,2 @@
+DROP TABLE 
+IF EXISTS employees ;

src/test/testData/sql/formatter/DropTableIfExists_format.sql

@@ -0,0 +1 @@
+DROP TABLE IF EXISTS employees ;

src/test/testData/sql/formatter/DropTable_format.sql

@@ -0,0 +1 @@
+DROP TABLE employees