fix: Support SET ROLE with TO/=

cube-js · Sep 2, 2024 · 74b7bad · 74b7bad
1 parent 3225ad3
commit 74b7bad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/parser.rs b/src/parser.rs
@@ -3629,6 +3629,7 @@ impl<'a> Parser<'a> {
         ]);
 
         if modifier != Some(Keyword::HIVEVAR) && self.parse_keyword(Keyword::ROLE) {
+            let _ = self.consume_token(&Token::Eq) || self.parse_keyword(Keyword::TO);
             let role_name = if self.parse_keyword(Keyword::NONE) {
                 None
             } else {

diff --git a/tests/sqlparser_postgres.rs b/tests/sqlparser_postgres.rs
@@ -960,7 +960,10 @@ fn parse_set_role() {
         }
     );
 
-    let stmt = pg_and_generic().verified_stmt("SET LOCAL ROLE \"rolename\"");
+    let stmt = pg_and_generic().one_statement_parses_to(
+        "SET LOCAL ROLE = \"rolename\"",
+        "SET LOCAL ROLE \"rolename\"",
+    );
     assert_eq!(
         stmt,
         Statement::SetRole {
@@ -973,7 +976,8 @@ fn parse_set_role() {
         }
     );
 
-    let stmt = pg_and_generic().verified_stmt("SET ROLE 'rolename'");
+    let stmt =
+        pg_and_generic().one_statement_parses_to("SET ROLE TO 'rolename'", "SET ROLE 'rolename'");
     assert_eq!(
         stmt,
         Statement::SetRole {