fix: cte (#263)

Author: psteinroe

crates/pgt_statement_splitter/src/lib.rs

@@ -142,6 +142,12 @@ mod tests {
             .expect_statements(vec!["insert into tbl (id) select 1", "select 3"]);
     }
 
+    #[test]
+    fn with_cte() {
+        Tester::from("with test as (select 1 as id) select * from test;")
+            .expect_statements(vec!["with test as (select 1 as id) select * from test;"]);
+    }
+
     #[test]
     fn case() {
         Tester::from("select case when select 2 then 1 else 0 end")

crates/pgt_statement_splitter/src/parser/dml.rs

@@ -2,7 +2,7 @@ use pgt_lexer::SyntaxKind;
 
 use super::{
     Parser,
-    common::{parenthesis, statement, unknown},
+    common::{parenthesis, unknown},
 };
 
 pub(crate) fn cte(p: &mut Parser) {
@@ -18,7 +18,16 @@ pub(crate) fn cte(p: &mut Parser) {
         }
     }
 
-    statement(p);
+    unknown(
+        p,
+        &[
+            SyntaxKind::Select,
+            SyntaxKind::Insert,
+            SyntaxKind::Update,
+            SyntaxKind::DeleteP,
+            SyntaxKind::Merge,
+        ],
+    );
 }
 
 pub(crate) fn select(p: &mut Parser) {