Skip to content

Commit 05fd8d9

Browse files
committed
fix: add test cases
1 parent 5176273 commit 05fd8d9

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
@@ -297,18 +297,28 @@ fn test_select_wildcard_with_except() {
297297

298298
#[test]
299299
fn test_select_wildcard_with_replace() {
300+
let select = bigquery_and_generic()
301+
.verified_only_select(r#"SELECT * REPLACE 'widget' AS item_name FROM orders"#);
302+
let expected = SelectItem::Wildcard(WildcardAdditionalOptions {
303+
opt_replace: Some(ReplaceSelectItem::Single(Box::new(
304+
SelectItem::ExprWithAlias {
305+
expr: Expr::Value(Value::SingleQuotedString("widget".to_owned())),
306+
alias: Ident::new("item_name"),
307+
},
308+
))),
309+
..Default::default()
310+
});
311+
assert_eq!(expected, select.projection[0]);
312+
300313
let select = bigquery_and_generic()
301314
.verified_only_select(r#"SELECT * REPLACE (quantity / 2 AS quantity) FROM orders"#);
302315
let expected = SelectItem::Wildcard(WildcardAdditionalOptions {
303316
opt_replace: Some(ReplaceSelectItem::Multiple(vec![Box::new(
304317
SelectItem::ExprWithAlias {
305318
expr: Expr::BinaryOp {
306-
left: Box::new(sqlparser::ast::Expr::Identifier(Ident::new("quantity"))),
319+
left: Box::new(Expr::Identifier(Ident::new("quantity"))),
307320
op: BinaryOperator::Divide,
308-
right: Box::new(sqlparser::ast::Expr::Value(Value::Number(
309-
"2".to_string(),
310-
false,
311-
))),
321+
right: Box::new(Expr::Value(Value::Number("2".to_string(), false))),
312322
},
313323
alias: Ident::new("quantity"),
314324
},

0 commit comments

Comments
 (0)