mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-09-03 20:50:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Use FETCH_FIRST_TWO_ROWS_ONLY in tests to reduce duplication

Browse source
This commit is contained in:
Nickolay Ponomarev 2019-06-17 00:49:25 +03:00
parent f87e8d5158
commit c1509b36ec
1 changed files with 13 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

77
tests/sqlparser_common.rs
View file

@ -2206,15 +2206,13 @@ fn parse_singular_row_offset() {
#[test] #[test]
fn parse_fetch() { fn parse_fetch() {
const FETCH_FIRST_TWO_ROWS_ONLY: Fetch = Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
};
let ast = verified_query("SELECT foo FROM bar FETCH FIRST 2 ROWS ONLY"); let ast = verified_query("SELECT foo FROM bar FETCH FIRST 2 ROWS ONLY");
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
let ast = verified_query("SELECT foo FROM bar FETCH FIRST ROWS ONLY"); let ast = verified_query("SELECT foo FROM bar FETCH FIRST ROWS ONLY");
assert_eq!( assert_eq!(
ast.fetch, ast.fetch,
@ -2225,23 +2223,9 @@ fn parse_fetch() {
}) })
); );
let ast = verified_query("SELECT foo FROM bar WHERE foo = 4 FETCH FIRST 2 ROWS ONLY"); let ast = verified_query("SELECT foo FROM bar WHERE foo = 4 FETCH FIRST 2 ROWS ONLY");
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
let ast = verified_query("SELECT foo FROM bar ORDER BY baz FETCH FIRST 2 ROWS ONLY"); let ast = verified_query("SELECT foo FROM bar ORDER BY baz FETCH FIRST 2 ROWS ONLY");
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
let ast = verified_query( let ast = verified_query(
"SELECT foo FROM bar WHERE foo = 4 ORDER BY baz FETCH FIRST 2 ROWS WITH TIES", "SELECT foo FROM bar WHERE foo = 4 ORDER BY baz FETCH FIRST 2 ROWS WITH TIES",
); );
@ -2266,36 +2250,15 @@ fn parse_fetch() {
"SELECT foo FROM bar WHERE foo = 4 ORDER BY baz OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY", "SELECT foo FROM bar WHERE foo = 4 ORDER BY baz OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY",
); );
assert_eq!(ast.offset, Some(ASTNode::SQLValue(Value::Long(2)))); assert_eq!(ast.offset, Some(ASTNode::SQLValue(Value::Long(2))));
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
let ast = verified_query( let ast = verified_query(
"SELECT foo FROM (SELECT * FROM bar FETCH FIRST 2 ROWS ONLY) FETCH FIRST 2 ROWS ONLY", "SELECT foo FROM (SELECT * FROM bar FETCH FIRST 2 ROWS ONLY) FETCH FIRST 2 ROWS ONLY",
); );
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
match ast.body { match ast.body {
SQLSetExpr::Select(s) => match only(s.from).relation { SQLSetExpr::Select(s) => match only(s.from).relation {
TableFactor::Derived { subquery, .. } => { TableFactor::Derived { subquery, .. } => {
assert_eq!( assert_eq!(subquery.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
subquery.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
} }
_ => panic!("Test broke"), _ => panic!("Test broke"),
}, },
@ -2303,26 +2266,12 @@ fn parse_fetch() {
} }
let ast = verified_query("SELECT foo FROM (SELECT * FROM bar OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY) OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY"); let ast = verified_query("SELECT foo FROM (SELECT * FROM bar OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY) OFFSET 2 ROWS FETCH FIRST 2 ROWS ONLY");
assert_eq!(ast.offset, Some(ASTNode::SQLValue(Value::Long(2)))); assert_eq!(ast.offset, Some(ASTNode::SQLValue(Value::Long(2))));
assert_eq!( assert_eq!(ast.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
ast.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
match ast.body { match ast.body {
SQLSetExpr::Select(s) => match only(s.from).relation { SQLSetExpr::Select(s) => match only(s.from).relation {
TableFactor::Derived { subquery, .. } => { TableFactor::Derived { subquery, .. } => {
assert_eq!(subquery.offset, Some(ASTNode::SQLValue(Value::Long(2)))); assert_eq!(subquery.offset, Some(ASTNode::SQLValue(Value::Long(2))));
assert_eq!( assert_eq!(subquery.fetch, Some(FETCH_FIRST_TWO_ROWS_ONLY));
subquery.fetch,
Some(Fetch {
with_ties: false,
percent: false,
quantity: Some(ASTNode::SQLValue(Value::Long(2))),
})
);
} }
_ => panic!("Test broke"), _ => panic!("Test broke"),
}, },