Skip to content

Commit bf1a7ff

Browse files
committed
fix: add test cases
1 parent 8d8e2fb commit bf1a7ff

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

tests/sqlparser_bigquery.rs

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -315,18 +315,28 @@ fn test_select_wildcard_with_except() {
315315

316316
#[test]
317317
fn test_select_wildcard_with_replace() {
318+
let select = bigquery_and_generic()
319+
.verified_only_select(r#"SELECT * REPLACE 'widget' AS item_name FROM orders"#);
320+
let expected = SelectItem::Wildcard(WildcardAdditionalOptions {
321+
opt_replace: Some(ReplaceSelectItem::Single(Box::new(
322+
SelectItem::ExprWithAlias {
323+
expr: Expr::Value(Value::SingleQuotedString("widget".to_owned())),
324+
alias: Ident::new("item_name"),
325+
},
326+
))),
327+
..Default::default()
328+
});
329+
assert_eq!(expected, select.projection[0]);
330+
318331
let select = bigquery_and_generic()
319332
.verified_only_select(r#"SELECT * REPLACE (quantity / 2 AS quantity) FROM orders"#);
320333
let expected = SelectItem::Wildcard(WildcardAdditionalOptions {
321334
opt_replace: Some(ReplaceSelectItem::Multiple(vec![Box::new(
322335
SelectItem::ExprWithAlias {
323336
expr: Expr::BinaryOp {
324-
left: Box::new(sqlparser::ast::Expr::Identifier(Ident::new("quantity"))),
337+
left: Box::new(Expr::Identifier(Ident::new("quantity"))),
325338
op: BinaryOperator::Divide,
326-
right: Box::new(sqlparser::ast::Expr::Value(Value::Number(
327-
"2".to_string(),
328-
false,
329-
))),
339+
right: Box::new(Expr::Value(Value::Number("2".to_string(), false))),
330340
},
331341
alias: Ident::new("quantity"),
332342
},

0 commit comments

Comments
 (0)