Merge pull request #116 from cvng/feat/txn-stmt-begin

Author: psteinroe

.cargo/config.toml

@@ -1,2 +1,3 @@
 [alias]
+snap = "insta test"
 xtask = "run --package xtask --"

crates/codegen/src/get_node_properties.rs

@@ -486,6 +486,7 @@ fn custom_handlers(node: &Node) -> TokenStream {
         },
         "TransactionStmt" => quote! {
             match n.kind() {
+                protobuf::TransactionStmtKind::TransStmtBegin => tokens.push(TokenProperty::from(Token::BeginP)),
                 protobuf::TransactionStmtKind::TransStmtCommit => tokens.push(TokenProperty::from(Token::Commit)),
                 _ => panic!("Unknown TransactionStmt {:#?}", n.kind())
             }

crates/parser/tests/skipped.txt

@@ -1,4 +1,3 @@
-advisory_lock
 aggregates
 alter_generic
 alter_operator

crates/parser/tests/snapshots/statements/valid/[email protected]

@@ -0,0 +1,32 @@
+---
+source: crates/parser/tests/statement_parser_test.rs
+description: "--\n-- ADVISORY LOCKS\n--\n\nBEGIN;"
+---
+Parse {
+    cst: SourceFile@0..31
+      SqlComment@0..2 "--"
+      Newline@2..3 "\n"
+      SqlComment@3..20 "-- ADVISORY LOCKS"
+      Newline@20..21 "\n"
+      SqlComment@21..23 "--"
+      Newline@23..25 "\n\n"
+      TransactionStmt@25..31
+        BeginP@25..30 "BEGIN"
+        Ascii59@30..31 ";"
+    ,
+    errors: [],
+    stmts: [
+        RawStmt {
+            stmt: TransactionStmt(
+                TransactionStmt {
+                    kind: TransStmtBegin,
+                    options: [],
+                    savepoint_name: "",
+                    gid: "",
+                    chain: false,
+                },
+            ),
+            range: 24..30,
+        },
+    ],
+}

crates/parser/tests/snapshots/statements/valid/[email protected]

@@ -0,0 +1,12 @@
+---
+source: crates/parser/tests/statement_parser_test.rs
+description: ROLLBACK;
+---
+Parse {
+    cst: SourceFile@0..9
+      Rollback@0..8 "ROLLBACK"
+      Ascii59@8..9 ";"
+    ,
+    errors: [],
+    stmts: [],
+}