fix: create trigger split

Author: psteinroe

crates/pglt_statement_splitter/src/lib.rs

@@ -142,6 +142,23 @@ mod tests {
             .expect_statements(vec!["select case when select 2 then 1 else 0 end"]);
     }
 
+    #[test]
+    fn create_trigger() {
+        Tester::from("alter table appointment_status add constraint valid_key check (private.strip_special_chars(key) = key and length(key) > 0 and length(key) < 60);
+
+create trigger default_key before insert on appointment_type for each row when (new.key is null) execute procedure default_key ();
+
+create trigger default_key before insert on appointment_status for each row when (new.key is null) execute procedure default_key ();
+
+alter table deal_type add column key text not null;
+")
+            .expect_statements(vec!["alter table appointment_status add constraint valid_key check (private.strip_special_chars(key) = key and length(key) > 0 and length(key) < 60);",
+                "create trigger default_key before insert on appointment_type for each row when (new.key is null) execute procedure default_key ();",
+                "create trigger default_key before insert on appointment_status for each row when (new.key is null) execute procedure default_key ();",
+                "alter table deal_type add column key text not null;",
+            ]);
+    }
+
     #[test]
     #[timeout(1000)]
     fn simple_select() {

crates/pglt_statement_splitter/src/parser/ddl.rs

@@ -5,7 +5,15 @@ use super::{common::unknown, Parser};
 pub(crate) fn create(p: &mut Parser) {
     p.expect(SyntaxKind::Create);
 
-    unknown(p, &[]);
+    unknown(
+        p,
+        &[
+            SyntaxKind::Insert,
+            SyntaxKind::Update,
+            SyntaxKind::DeleteP,
+            SyntaxKind::Select,
+        ],
+    );
 }
 
 pub(crate) fn alter(p: &mut Parser) {