mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-08-04 22:38:01 +00:00
Postgres: support for OWNER TO
clause (#1314)
Co-authored-by: Andrew Lamb <andrew@nerdnetworks.org>
This commit is contained in:
parent
07278952f9
commit
32b8276b32
5 changed files with 123 additions and 2 deletions
|
@ -713,6 +713,78 @@ fn parse_alter_table_add_columns() {
|
|||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn parse_alter_table_owner_to() {
|
||||
struct TestCase {
|
||||
sql: &'static str,
|
||||
expected_owner: Owner,
|
||||
}
|
||||
|
||||
let test_cases = vec![
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO new_owner",
|
||||
expected_owner: Owner::Ident(Ident::new("new_owner".to_string())),
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO postgres",
|
||||
expected_owner: Owner::Ident(Ident::new("postgres".to_string())),
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO CREATE", // treats CREATE as an identifier
|
||||
expected_owner: Owner::Ident(Ident::new("CREATE".to_string())),
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO \"new_owner\"",
|
||||
expected_owner: Owner::Ident(Ident::with_quote('\"', "new_owner".to_string())),
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO CURRENT_USER",
|
||||
expected_owner: Owner::CurrentUser,
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO CURRENT_ROLE",
|
||||
expected_owner: Owner::CurrentRole,
|
||||
},
|
||||
TestCase {
|
||||
sql: "ALTER TABLE tab OWNER TO SESSION_USER",
|
||||
expected_owner: Owner::SessionUser,
|
||||
},
|
||||
];
|
||||
|
||||
for case in test_cases {
|
||||
match pg_and_generic().verified_stmt(case.sql) {
|
||||
Statement::AlterTable {
|
||||
name,
|
||||
if_exists: _,
|
||||
only: _,
|
||||
operations,
|
||||
location: _,
|
||||
} => {
|
||||
assert_eq!(name.to_string(), "tab");
|
||||
assert_eq!(
|
||||
operations,
|
||||
vec![AlterTableOperation::OwnerTo {
|
||||
new_owner: case.expected_owner.clone()
|
||||
}]
|
||||
);
|
||||
}
|
||||
_ => unreachable!("Expected an AlterTable statement"),
|
||||
}
|
||||
}
|
||||
|
||||
let res = pg().parse_sql_statements("ALTER TABLE tab OWNER TO CREATE FOO");
|
||||
assert_eq!(
|
||||
ParserError::ParserError("Expected: end of statement, found: FOO".to_string()),
|
||||
res.unwrap_err()
|
||||
);
|
||||
|
||||
let res = pg().parse_sql_statements("ALTER TABLE tab OWNER TO 4");
|
||||
assert_eq!(
|
||||
ParserError::ParserError("Expected: CURRENT_USER, CURRENT_ROLE, SESSION_USER or identifier after OWNER TO. sql parser error: Expected: identifier, found: 4".to_string()),
|
||||
res.unwrap_err()
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn parse_create_table_if_not_exists() {
|
||||
let sql = "CREATE TABLE IF NOT EXISTS uk_cities ()";
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue