add additional assertion for parse_set_transaction test

Signed-off-by: poonai <[email protected]>

Author: poonai

src/ast/mod.rs

@@ -786,7 +786,11 @@ pub enum Statement {
     /// `{ BEGIN [ TRANSACTION | WORK ] | START TRANSACTION } ...`
     StartTransaction { modes: Vec<TransactionMode> },
     /// `SET TRANSACTION ...`
-    SetTransaction { modes: Vec<TransactionMode>, snapshot: Option<Value>, session: bool },
+    SetTransaction {
+        modes: Vec<TransactionMode>,
+        snapshot: Option<Value>,
+        session: bool,
+    },
     /// `COMMIT [ TRANSACTION | WORK ] [ AND [ NO ] CHAIN ]`
     Commit { chain: bool },
     /// `ROLLBACK [ TRANSACTION | WORK ] [ AND [ NO ] CHAIN ]`
@@ -1369,7 +1373,11 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
-            Statement::SetTransaction { modes, snapshot, session } => {
+            Statement::SetTransaction {
+                modes,
+                snapshot,
+                session,
+            } => {
                 if *session {
                     write!(f, "SET SESSION CHARACTERISTICS AS TRANSACTION")?;
                 } else {
@@ -1378,7 +1386,7 @@ impl fmt::Display for Statement {
                 if !modes.is_empty() {
                     write!(f, " {}", display_comma_separated(modes))?;
                 }
-                if let Some(snapshot_id) = snapshot{
+                if let Some(snapshot_id) = snapshot {
                     write!(f, " SNAPSHOT {}", snapshot_id)?;
                 }
                 Ok(())

src/parser.rs

@@ -2683,13 +2683,13 @@ impl<'a> Parser<'a> {
                 return Ok(Statement::SetTransaction {
                     modes: vec![],
                     snapshot: Some(snaphot_id),
-                    session: false
+                    session: false,
                 });
             }
             Ok(Statement::SetTransaction {
                 modes: self.parse_transaction_modes()?,
                 snapshot: None,
-                session: false
+                session: false,
             })
         } else {
             self.expected("equals sign or TO", self.peek_token())

tests/sqlparser_common.rs

@@ -3586,14 +3586,22 @@ fn parse_set_transaction() {
     // TRANSACTION, so no need to duplicate the tests here. We just do a quick
     // sanity check.
     match verified_stmt("SET TRANSACTION READ ONLY, READ WRITE, ISOLATION LEVEL SERIALIZABLE") {
-        Statement::SetTransaction { modes, .. } => assert_eq!(
+        Statement::SetTransaction {
             modes,
-            vec![
-                TransactionMode::AccessMode(TransactionAccessMode::ReadOnly),
-                TransactionMode::AccessMode(TransactionAccessMode::ReadWrite),
-                TransactionMode::IsolationLevel(TransactionIsolationLevel::Serializable),
-            ]
-        ),
+            session,
+            snapshot,
+        } => {
+            assert_eq!(
+                modes,
+                vec![
+                    TransactionMode::AccessMode(TransactionAccessMode::ReadOnly),
+                    TransactionMode::AccessMode(TransactionAccessMode::ReadWrite),
+                    TransactionMode::IsolationLevel(TransactionIsolationLevel::Serializable),
+                ]
+            );
+            assert_eq!(session, false);
+            assert_eq!(snapshot, None);
+        }
         _ => unreachable!(),
     }
 }