fix alter database collation

sbdchd · Nov 7, 2023 · 05c26ce · 05c26ce
1 parent a4b5d41
commit 05c26ce
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/parser/src/ast.rs b/parser/src/ast.rs
@@ -945,6 +945,7 @@ pub enum Stmt {
     ViewStmt(Value),
     LoadStmt(Value),
     CreatedbStmt(Value),
+    AlterDatabaseRefreshCollStmt(Value),
     AlterDatabaseStmt(Value),
     AlterDatabaseSetStmt(Value),
     DropdbStmt(Value),

diff --git a/parser/src/parse.rs b/parser/src/parse.rs
@@ -965,6 +965,15 @@ ALTER DATABASE name RESET ALL;
         assert_debug_snapshot!(res);
     }
 
+    #[test]
+    fn test_alter_database_collation() {
+        let sql = r#"
+ALTER DATABASE pipelines REFRESH COLLATION VERSION;
+"#;
+        let res = parse_sql_query(sql);
+        assert_debug_snapshot!(res);
+    }
+
     #[test]
     fn test_drop_database_stmt() {
         let sql = r#"

diff --git a/parser/src/snapshots/squawk_parser__parse__tests__alter_database_collation.snap b/parser/src/snapshots/squawk_parser__parse__tests__alter_database_collation.snap
@@ -0,0 +1,21 @@
+---
+source: parser/src/parse.rs
+expression: res
+---
+Ok(
+    [
+        RawStmt {
+            stmt: AlterDatabaseRefreshCollStmt(
+                Object({
+                    "dbname": String(
+                        "pipelines",
+                    ),
+                }),
+            ),
+            stmt_location: 0,
+            stmt_len: Some(
+                51,
+            ),
+        },
+    ],
+)