mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-07-08 01:15:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

SET statements: scope modifier for multiple assignments (#1772)

Browse source
This commit is contained in:
Mohamed Abdeen 2025-03-22 07:38:00 +02:00 • committed by GitHub
parent 939fbdd4f6
commit 3a8a3bb7a5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 133 additions and 94 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/sqlparser_postgres.rs
View file

@ -1432,7 +1432,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a")]),
values: vec![Expr::Identifier(Ident {
@ -1447,7 +1447,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a")]),
values: vec![Expr::Value(
@ -1460,7 +1460,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a")]),
values: vec![Expr::value(number("0"))],
@ -1471,7 +1471,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a")]),
values: vec![Expr::Identifier(Ident::new("DEFAULT"))],
@ -1482,7 +1482,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::Local,
scope: Some(ContextModifier::Local),
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a")]),
values: vec![Expr::Identifier("b".into())],
@ -1493,7 +1493,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![Ident::new("a"), Ident::new("b"), Ident::new("c")]),
values: vec![Expr::Identifier(Ident {
@ -1511,7 +1511,7 @@ fn parse_set() {
assert_eq!(
stmt,
Statement::Set(Set::SingleAssignment {
scope: ContextModifier::None,
scope: None,
hivevar: false,
variable: ObjectName::from(vec![
Ident::new("hive"),
@ -1555,7 +1555,7 @@ fn parse_set_role() {
assert_eq!(
stmt,
Statement::Set(Set::SetRole {
context_modifier: ContextModifier::Session,
context_modifier: Some(ContextModifier::Session),
role_name: None,
})
);
@ -1566,7 +1566,7 @@ fn parse_set_role() {
assert_eq!(
stmt,
Statement::Set(Set::SetRole {
context_modifier: ContextModifier::Local,
context_modifier: Some(ContextModifier::Local),
role_name: Some(Ident {
value: "rolename".to_string(),
quote_style: Some('\"'),
@ -1581,7 +1581,7 @@ fn parse_set_role() {
assert_eq!(
stmt,
Statement::Set(Set::SetRole {
context_modifier: ContextModifier::None,
context_modifier: None,
role_name: Some(Ident {
value: "rolename".to_string(),
quote_style: Some('\''),