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

Encapsulate Insert and Delete into specific structs (#1224)

Browse source 
Signed-off-by: tison <wander4096@gmail.com>
This commit is contained in:
tison 2024-04-21 21:13:18 +08:00 • committed by GitHub
parent d2c2b15f9e
commit bf89b7d808
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 187 additions and 147 deletions
Download patch file Download diff file Expand all files Collapse all files

40
tests/sqlparser_postgres.rs
View file

@ -1482,12 +1482,12 @@ fn parse_prepare() {
_ => unreachable!(),
};
match sub_stmt.as_ref() {
Statement::Insert {
Statement::Insert(Insert {
table_name,
columns,
source: Some(source),
..
} => {
}) => {
assert_eq!(table_name.to_string(), "customers");
assert!(columns.is_empty());
@ -1539,14 +1539,14 @@ fn parse_pg_on_conflict() {
DO UPDATE SET dname = EXCLUDED.dname",
);
match stmt {
Statement::Insert {
Statement::Insert(Insert {
on:
Some(OnInsert::OnConflict(OnConflict {
conflict_target: Some(ConflictTarget::Columns(cols)),
action,
})),
..
} => {
}) => {
assert_eq!(vec![Ident::from("did")], cols);
assert_eq!(
OnConflictAction::DoUpdate(DoUpdate {
@ -1569,14 +1569,14 @@ fn parse_pg_on_conflict() {
DO UPDATE SET dname = EXCLUDED.dname, area = EXCLUDED.area",
);
match stmt {
Statement::Insert {
Statement::Insert(Insert {
on:
Some(OnInsert::OnConflict(OnConflict {
conflict_target: Some(ConflictTarget::Columns(cols)),
action,
})),
..
} => {
}) => {
assert_eq!(vec![Ident::from("did"), Ident::from("area"),], cols);
assert_eq!(
OnConflictAction::DoUpdate(DoUpdate {
@ -1607,14 +1607,14 @@ fn parse_pg_on_conflict() {
ON CONFLICT DO NOTHING",
);
match stmt {
Statement::Insert {
Statement::Insert(Insert {
on:
Some(OnInsert::OnConflict(OnConflict {
conflict_target: None,
action,
})),
..
} => {
}) => {
assert_eq!(OnConflictAction::DoNothing, action);
}
_ => unreachable!(),
@ -1627,14 +1627,14 @@ fn parse_pg_on_conflict() {
DO UPDATE SET dname = $1 WHERE dsize > $2",
);
match stmt {
Statement::Insert {
Statement::Insert(Insert {
on:
Some(OnInsert::OnConflict(OnConflict {
conflict_target: Some(ConflictTarget::Columns(cols)),
action,
})),
..
} => {
}) => {
assert_eq!(vec![Ident::from("did")], cols);
assert_eq!(
OnConflictAction::DoUpdate(DoUpdate {
@ -1664,14 +1664,14 @@ fn parse_pg_on_conflict() {
DO UPDATE SET dname = $1 WHERE dsize > $2",
);
match stmt {
Statement::Insert {
Statement::Insert(Insert {
on:
Some(OnInsert::OnConflict(OnConflict {
conflict_target: Some(ConflictTarget::OnConstraint(cname)),
action,
})),
..
} => {
}) => {
assert_eq!(vec![Ident::from("distributors_did_pkey")], cname.0);
assert_eq!(
OnConflictAction::DoUpdate(DoUpdate {
@ -1701,7 +1701,7 @@ fn parse_pg_returning() {
"INSERT INTO distributors (did, dname) VALUES (DEFAULT, 'XYZ Widgets') RETURNING did",
);
match stmt {
Statement::Insert { returning, .. } => {
Statement::Insert(Insert { returning, .. }) => {
assert_eq!(
Some(vec![SelectItem::UnnamedExpr(Expr::Identifier(
"did".into()
@ -1739,7 +1739,7 @@ fn parse_pg_returning() {
let stmt =
pg_and_generic().verified_stmt("DELETE FROM tasks WHERE status = 'DONE' RETURNING *");
match stmt {
Statement::Delete { returning, .. } => {
Statement::Delete(Delete { returning, .. }) => {
assert_eq!(
Some(vec![SelectItem::Wildcard(
WildcardAdditionalOptions::default()
@ -3570,7 +3570,7 @@ fn test_simple_postgres_insert_with_alias() {
assert_eq!(
statement,
Statement::Insert {
Statement::Insert(Insert {
or: None,
ignore: false,
into: true,
@ -3621,7 +3621,7 @@ fn test_simple_postgres_insert_with_alias() {
replace_into: false,
priority: None,
insert_alias: None
}
})
)
}
@ -3634,7 +3634,7 @@ fn test_simple_postgres_insert_with_alias() {
assert_eq!(
statement,
Statement::Insert {
Statement::Insert(Insert {
or: None,
ignore: false,
into: true,
@ -3688,7 +3688,7 @@ fn test_simple_postgres_insert_with_alias() {
replace_into: false,
priority: None,
insert_alias: None
}
})
)
}
@ -3700,7 +3700,7 @@ fn test_simple_insert_with_quoted_alias() {
assert_eq!(
statement,
Statement::Insert {
Statement::Insert(Insert {
or: None,
ignore: false,
into: true,
@ -3751,7 +3751,7 @@ fn test_simple_insert_with_quoted_alias() {
replace_into: false,
priority: None,
insert_alias: None,
}
})
)
}