Implement ASSERT statement (#226)

As supported by PostgreSQL and BigQuery (with some differences between them)
This commit is contained in:
Daniël Heres 2020-07-16 17:28:03 +02:00 committed by GitHub
parent 5cab18963e
commit c24b0e01db
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 78 additions and 0 deletions

View file

@ -1156,6 +1156,46 @@ fn parse_create_table_with_multiple_on_delete_fails() {
.expect_err("should have failed");
}
#[test]
fn parse_assert() {
let sql = "ASSERT (SELECT COUNT(*) FROM table) > 0";
let ast = one_statement_parses_to(sql, "ASSERT (SELECT COUNT(*) FROM table) > 0");
match ast {
Statement::Assert {
condition: _condition,
separator,
message,
} => {
assert_eq!(message, None);
assert_eq!(separator, "");
}
_ => unreachable!(),
}
}
#[test]
fn parse_assert_message() {
let sql = "ASSERT (SELECT COUNT(*) FROM table) > 0 AS 'No rows in table'";
let ast = one_statement_parses_to(
sql,
"ASSERT (SELECT COUNT(*) FROM table) > 0 AS 'No rows in table'",
);
match ast {
Statement::Assert {
condition: _condition,
message: Some(message),
separator,
} => {
assert_eq!(separator, "AS");
match message {
Expr::Value(Value::SingleQuotedString(s)) => assert_eq!(s, "No rows in table"),
_ => unreachable!(),
};
}
_ => unreachable!(),
}
}
#[test]
fn parse_create_schema() {
let sql = "CREATE SCHEMA X";