mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-07-08 01:15:00 +00:00
7cbbd9188e
* Support for empty array literals * Added support for AT TIME ZONE Co-authored-by: Chris Allen <chrisa@indeed.com>
1566 lines
51 KiB
Rust
1566 lines
51 KiB
Rust
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#![warn(clippy::all)]
|
|
//! Test SQL syntax specific to PostgreSQL. The parser based on the
|
|
//! generic dialect is also tested (on the inputs it can handle).
|
|
|
|
#[macro_use]
|
|
mod test_utils;
|
|
use test_utils::*;
|
|
|
|
use sqlparser::ast::Value::Boolean;
|
|
use sqlparser::ast::*;
|
|
use sqlparser::dialect::{GenericDialect, PostgreSqlDialect};
|
|
use sqlparser::parser::ParserError;
|
|
|
|
#[test]
|
|
fn parse_create_table_with_defaults() {
|
|
let sql = "CREATE TABLE public.customer (
|
|
customer_id integer DEFAULT nextval(public.customer_customer_id_seq),
|
|
store_id smallint NOT NULL,
|
|
first_name character varying(45) NOT NULL,
|
|
last_name character varying(45) COLLATE \"es_ES\" NOT NULL,
|
|
email character varying(50),
|
|
address_id smallint NOT NULL,
|
|
activebool boolean DEFAULT true NOT NULL,
|
|
create_date date DEFAULT now()::text NOT NULL,
|
|
last_update timestamp without time zone DEFAULT now() NOT NULL,
|
|
active int NOT NULL
|
|
) WITH (fillfactor = 20, user_catalog_table = true, autovacuum_vacuum_threshold = 100)";
|
|
match pg_and_generic().one_statement_parses_to(sql, "") {
|
|
Statement::CreateTable {
|
|
name,
|
|
columns,
|
|
constraints,
|
|
with_options,
|
|
if_not_exists: false,
|
|
external: false,
|
|
file_format: None,
|
|
location: None,
|
|
..
|
|
} => {
|
|
assert_eq!("public.customer", name.to_string());
|
|
assert_eq!(
|
|
columns,
|
|
vec![
|
|
ColumnDef {
|
|
name: "customer_id".into(),
|
|
data_type: DataType::Integer(None),
|
|
collation: None,
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::Default(
|
|
pg().verified_expr("nextval(public.customer_customer_id_seq)")
|
|
)
|
|
}],
|
|
},
|
|
ColumnDef {
|
|
name: "store_id".into(),
|
|
data_type: DataType::SmallInt(None),
|
|
collation: None,
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}],
|
|
},
|
|
ColumnDef {
|
|
name: "first_name".into(),
|
|
data_type: DataType::Varchar(Some(45)),
|
|
collation: None,
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}],
|
|
},
|
|
ColumnDef {
|
|
name: "last_name".into(),
|
|
data_type: DataType::Varchar(Some(45)),
|
|
collation: Some(ObjectName(vec![Ident::with_quote('"', "es_ES")])),
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}],
|
|
},
|
|
ColumnDef {
|
|
name: "email".into(),
|
|
data_type: DataType::Varchar(Some(50)),
|
|
collation: None,
|
|
options: vec![],
|
|
},
|
|
ColumnDef {
|
|
name: "address_id".into(),
|
|
data_type: DataType::SmallInt(None),
|
|
collation: None,
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull
|
|
}],
|
|
},
|
|
ColumnDef {
|
|
name: "activebool".into(),
|
|
data_type: DataType::Boolean,
|
|
collation: None,
|
|
options: vec![
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::Default(Expr::Value(Value::Boolean(true))),
|
|
},
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}
|
|
],
|
|
},
|
|
ColumnDef {
|
|
name: "create_date".into(),
|
|
data_type: DataType::Date,
|
|
collation: None,
|
|
options: vec![
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::Default(
|
|
pg().verified_expr("CAST(now() AS TEXT)")
|
|
)
|
|
},
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}
|
|
],
|
|
},
|
|
ColumnDef {
|
|
name: "last_update".into(),
|
|
data_type: DataType::Timestamp,
|
|
collation: None,
|
|
options: vec![
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::Default(pg().verified_expr("now()")),
|
|
},
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull,
|
|
}
|
|
],
|
|
},
|
|
ColumnDef {
|
|
name: "active".into(),
|
|
data_type: DataType::Int(None),
|
|
collation: None,
|
|
options: vec![ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::NotNull
|
|
}],
|
|
},
|
|
]
|
|
);
|
|
assert!(constraints.is_empty());
|
|
assert_eq!(
|
|
with_options,
|
|
vec![
|
|
SqlOption {
|
|
name: "fillfactor".into(),
|
|
value: number("20")
|
|
},
|
|
SqlOption {
|
|
name: "user_catalog_table".into(),
|
|
value: Value::Boolean(true)
|
|
},
|
|
SqlOption {
|
|
name: "autovacuum_vacuum_threshold".into(),
|
|
value: number("100")
|
|
},
|
|
]
|
|
);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_from_pg_dump() {
|
|
let sql = "CREATE TABLE public.customer (
|
|
customer_id integer DEFAULT nextval('public.customer_customer_id_seq'::regclass) NOT NULL,
|
|
store_id smallint NOT NULL,
|
|
first_name character varying(45) NOT NULL,
|
|
last_name character varying(45) NOT NULL,
|
|
info text[],
|
|
address_id smallint NOT NULL,
|
|
activebool boolean DEFAULT true NOT NULL,
|
|
create_date date DEFAULT now()::DATE NOT NULL,
|
|
create_date1 date DEFAULT 'now'::TEXT::date NOT NULL,
|
|
last_update timestamp without time zone DEFAULT now(),
|
|
release_year public.year,
|
|
active int
|
|
)";
|
|
pg().one_statement_parses_to(sql, "CREATE TABLE public.customer (\
|
|
customer_id INTEGER DEFAULT nextval(CAST('public.customer_customer_id_seq' AS REGCLASS)) NOT NULL, \
|
|
store_id SMALLINT NOT NULL, \
|
|
first_name CHARACTER VARYING(45) NOT NULL, \
|
|
last_name CHARACTER VARYING(45) NOT NULL, \
|
|
info TEXT[], \
|
|
address_id SMALLINT NOT NULL, \
|
|
activebool BOOLEAN DEFAULT true NOT NULL, \
|
|
create_date DATE DEFAULT CAST(now() AS DATE) NOT NULL, \
|
|
create_date1 DATE DEFAULT CAST(CAST('now' AS TEXT) AS DATE) NOT NULL, \
|
|
last_update TIMESTAMP DEFAULT now(), \
|
|
release_year public.year, \
|
|
active INT\
|
|
)");
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_with_inherit() {
|
|
let sql = "\
|
|
CREATE TABLE bazaar.settings (\
|
|
settings_id UUID PRIMARY KEY DEFAULT uuid_generate_v4() NOT NULL, \
|
|
user_id UUID UNIQUE, \
|
|
value TEXT[], \
|
|
use_metric BOOLEAN DEFAULT true\
|
|
)";
|
|
pg().verified_stmt(sql);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_empty() {
|
|
// Zero-column tables are weird, but supported by at least PostgreSQL.
|
|
// <https://github.com/sqlparser-rs/sqlparser-rs/pull/94>
|
|
let _ = pg_and_generic().verified_stmt("CREATE TABLE t ()");
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_constraints_only() {
|
|
// Zero-column tables can also have constraints in PostgreSQL
|
|
let sql = "CREATE TABLE t (CONSTRAINT positive CHECK (2 > 1))";
|
|
let ast = pg_and_generic().verified_stmt(sql);
|
|
match ast {
|
|
Statement::CreateTable {
|
|
name,
|
|
columns,
|
|
constraints,
|
|
..
|
|
} => {
|
|
assert_eq!("t", name.to_string());
|
|
assert!(columns.is_empty());
|
|
assert_eq!(
|
|
only(constraints).to_string(),
|
|
"CONSTRAINT positive CHECK (2 > 1)"
|
|
);
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
}
|
|
|
|
#[test]
|
|
fn parse_alter_table_constraints_rename() {
|
|
match pg().verified_stmt("ALTER TABLE tab RENAME CONSTRAINT old_name TO new_name") {
|
|
Statement::AlterTable {
|
|
name,
|
|
operation: AlterTableOperation::RenameConstraint { old_name, new_name },
|
|
} => {
|
|
assert_eq!("tab", name.to_string());
|
|
assert_eq!(old_name.to_string(), "old_name");
|
|
assert_eq!(new_name.to_string(), "new_name");
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_alter_table_alter_column() {
|
|
pg().one_statement_parses_to(
|
|
"ALTER TABLE tab ALTER COLUMN is_active TYPE TEXT USING 'text'",
|
|
"ALTER TABLE tab ALTER COLUMN is_active SET DATA TYPE TEXT USING 'text'",
|
|
);
|
|
|
|
match pg()
|
|
.verified_stmt("ALTER TABLE tab ALTER COLUMN is_active SET DATA TYPE TEXT USING 'text'")
|
|
{
|
|
Statement::AlterTable {
|
|
name,
|
|
operation: AlterTableOperation::AlterColumn { column_name, op },
|
|
} => {
|
|
assert_eq!("tab", name.to_string());
|
|
assert_eq!("is_active", column_name.to_string());
|
|
let using_expr = Expr::Value(Value::SingleQuotedString("text".to_string()));
|
|
assert_eq!(
|
|
op,
|
|
AlterColumnOperation::SetDataType {
|
|
data_type: DataType::Text,
|
|
using: Some(using_expr),
|
|
}
|
|
);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_if_not_exists() {
|
|
let sql = "CREATE TABLE IF NOT EXISTS uk_cities ()";
|
|
let ast = pg_and_generic().verified_stmt(sql);
|
|
match ast {
|
|
Statement::CreateTable {
|
|
name,
|
|
if_not_exists: true,
|
|
..
|
|
} => {
|
|
assert_eq!("uk_cities", name.to_string());
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_bad_if_not_exists() {
|
|
let res = pg().parse_sql_statements("CREATE TABLE NOT EXISTS uk_cities ()");
|
|
assert_eq!(
|
|
ParserError::ParserError("Expected end of statement, found: EXISTS".to_string()),
|
|
res.unwrap_err()
|
|
);
|
|
|
|
let res = pg().parse_sql_statements("CREATE TABLE IF EXISTS uk_cities ()");
|
|
assert_eq!(
|
|
ParserError::ParserError("Expected end of statement, found: EXISTS".to_string()),
|
|
res.unwrap_err()
|
|
);
|
|
|
|
let res = pg().parse_sql_statements("CREATE TABLE IF uk_cities ()");
|
|
assert_eq!(
|
|
ParserError::ParserError("Expected end of statement, found: uk_cities".to_string()),
|
|
res.unwrap_err()
|
|
);
|
|
|
|
let res = pg().parse_sql_statements("CREATE TABLE IF NOT uk_cities ()");
|
|
assert_eq!(
|
|
ParserError::ParserError("Expected end of statement, found: NOT".to_string()),
|
|
res.unwrap_err()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_schema_if_not_exists() {
|
|
let sql = "CREATE SCHEMA IF NOT EXISTS schema_name";
|
|
let ast = pg_and_generic().verified_stmt(sql);
|
|
match ast {
|
|
Statement::CreateSchema {
|
|
if_not_exists: true,
|
|
schema_name,
|
|
} => assert_eq!("schema_name", schema_name.to_string()),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_drop_schema_if_exists() {
|
|
let sql = "DROP SCHEMA IF EXISTS schema_name";
|
|
let ast = pg().verified_stmt(sql);
|
|
match ast {
|
|
Statement::Drop {
|
|
object_type,
|
|
if_exists: true,
|
|
..
|
|
} => assert_eq!(object_type, ObjectType::Schema),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_copy_from_stdin() {
|
|
let sql = r#"COPY public.actor (actor_id, first_name, last_name, last_update, value) FROM stdin;
|
|
1 PENELOPE GUINESS 2006-02-15 09:34:33 0.11111
|
|
2 NICK WAHLBERG 2006-02-15 09:34:33 0.22222
|
|
3 ED CHASE 2006-02-15 09:34:33 0.312323
|
|
4 JENNIFER DAVIS 2006-02-15 09:34:33 0.3232
|
|
5 JOHNNY LOLLOBRIGIDA 2006-02-15 09:34:33 1.343
|
|
6 BETTE NICHOLSON 2006-02-15 09:34:33 5.0
|
|
7 GRACE MOSTEL 2006-02-15 09:34:33 6.0
|
|
8 MATTHEW JOHANSSON 2006-02-15 09:34:33 7.0
|
|
9 JOE SWANK 2006-02-15 09:34:33 8.0
|
|
10 CHRISTIAN GABLE 2006-02-15 09:34:33 9.1
|
|
11 ZERO CAGE 2006-02-15 09:34:33 10.001
|
|
12 KARL BERRY 2017-11-02 19:15:42.308637+08 11.001
|
|
A Fateful Reflection of a Waitress And a Boat who must Discover a Sumo Wrestler in Ancient China
|
|
Kwara & Kogi
|
|
{"Deleted Scenes","Behind the Scenes"}
|
|
'awe':5 'awe-inspir':4 'barbarella':1 'cat':13 'conquer':16 'dog':18 'feminist':10 'inspir':6 'monasteri':21 'must':15 'stori':7 'streetcar':2
|
|
PHP ₱ USD $
|
|
\N Some other value
|
|
\\."#;
|
|
let ast = pg_and_generic().one_statement_parses_to(sql, "");
|
|
println!("{:#?}", ast);
|
|
//assert_eq!(sql, ast.to_string());
|
|
}
|
|
|
|
#[test]
|
|
fn parse_update_set_from() {
|
|
let sql = "UPDATE t1 SET name = t2.name FROM (SELECT name, id FROM t1 GROUP BY id) AS t2 WHERE t1.id = t2.id";
|
|
let stmt = pg().verified_stmt(sql);
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Update {
|
|
table: TableWithJoins {
|
|
relation: TableFactor::Table {
|
|
name: ObjectName(vec![Ident::new("t1")]),
|
|
alias: None,
|
|
args: None,
|
|
with_hints: vec![],
|
|
},
|
|
joins: vec![],
|
|
},
|
|
assignments: vec![Assignment {
|
|
id: vec![Ident::new("name")],
|
|
value: Expr::CompoundIdentifier(vec![Ident::new("t2"), Ident::new("name")])
|
|
}],
|
|
from: Some(TableWithJoins {
|
|
relation: TableFactor::Derived {
|
|
lateral: false,
|
|
subquery: Box::new(Query {
|
|
with: None,
|
|
body: Box::new(SetExpr::Select(Box::new(Select {
|
|
distinct: false,
|
|
top: None,
|
|
projection: vec![
|
|
SelectItem::UnnamedExpr(Expr::Identifier(Ident::new("name"))),
|
|
SelectItem::UnnamedExpr(Expr::Identifier(Ident::new("id"))),
|
|
],
|
|
into: None,
|
|
from: vec![TableWithJoins {
|
|
relation: TableFactor::Table {
|
|
name: ObjectName(vec![Ident::new("t1")]),
|
|
alias: None,
|
|
args: None,
|
|
with_hints: vec![],
|
|
},
|
|
joins: vec![],
|
|
}],
|
|
lateral_views: vec![],
|
|
selection: None,
|
|
group_by: vec![Expr::Identifier(Ident::new("id"))],
|
|
cluster_by: vec![],
|
|
distribute_by: vec![],
|
|
sort_by: vec![],
|
|
having: None,
|
|
qualify: None
|
|
}))),
|
|
order_by: vec![],
|
|
limit: None,
|
|
offset: None,
|
|
fetch: None,
|
|
lock: None,
|
|
}),
|
|
alias: Some(TableAlias {
|
|
name: Ident::new("t2"),
|
|
columns: vec![],
|
|
})
|
|
},
|
|
joins: vec![],
|
|
}),
|
|
selection: Some(Expr::BinaryOp {
|
|
left: Box::new(Expr::CompoundIdentifier(vec![
|
|
Ident::new("t1"),
|
|
Ident::new("id")
|
|
])),
|
|
op: BinaryOperator::Eq,
|
|
right: Box::new(Expr::CompoundIdentifier(vec![
|
|
Ident::new("t2"),
|
|
Ident::new("id")
|
|
])),
|
|
}),
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_copy_from() {
|
|
let stmt = pg().verified_stmt("COPY users FROM 'data.csv'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg().verified_stmt("COPY users FROM 'data.csv' DELIMITER ','");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![CopyLegacyOption::Delimiter(',')],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg().verified_stmt("COPY users FROM 'data.csv' DELIMITER ',' CSV HEADER");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![
|
|
CopyLegacyOption::Delimiter(','),
|
|
CopyLegacyOption::Csv(vec![CopyLegacyCsvOption::Header,])
|
|
],
|
|
values: vec![],
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_copy_to() {
|
|
let stmt = pg().verified_stmt("COPY users TO 'data.csv'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg().verified_stmt("COPY users TO 'data.csv' DELIMITER ','");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![CopyLegacyOption::Delimiter(',')],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg().verified_stmt("COPY users TO 'data.csv' DELIMITER ',' CSV HEADER");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![
|
|
CopyLegacyOption::Delimiter(','),
|
|
CopyLegacyOption::Csv(vec![CopyLegacyCsvOption::Header,])
|
|
],
|
|
values: vec![],
|
|
}
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn parse_copy_from() {
|
|
let sql = "COPY table (a, b) FROM 'file.csv' WITH
|
|
(
|
|
FORMAT CSV,
|
|
FREEZE,
|
|
FREEZE TRUE,
|
|
FREEZE FALSE,
|
|
DELIMITER ',',
|
|
NULL '',
|
|
HEADER,
|
|
HEADER TRUE,
|
|
HEADER FALSE,
|
|
QUOTE '\"',
|
|
ESCAPE '\\',
|
|
FORCE_QUOTE (a, b),
|
|
FORCE_NOT_NULL (a),
|
|
FORCE_NULL (b),
|
|
ENCODING 'utf8'
|
|
)";
|
|
assert_eq!(
|
|
pg_and_generic().one_statement_parses_to(sql, ""),
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["table".into()]),
|
|
columns: vec!["a".into(), "b".into()],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "file.csv".into()
|
|
},
|
|
options: vec![
|
|
CopyOption::Format("CSV".into()),
|
|
CopyOption::Freeze(true),
|
|
CopyOption::Freeze(true),
|
|
CopyOption::Freeze(false),
|
|
CopyOption::Delimiter(','),
|
|
CopyOption::Null("".into()),
|
|
CopyOption::Header(true),
|
|
CopyOption::Header(true),
|
|
CopyOption::Header(false),
|
|
CopyOption::Quote('"'),
|
|
CopyOption::Escape('\\'),
|
|
CopyOption::ForceQuote(vec!["a".into(), "b".into()]),
|
|
CopyOption::ForceNotNull(vec!["a".into()]),
|
|
CopyOption::ForceNull(vec!["b".into()]),
|
|
CopyOption::Encoding("utf8".into()),
|
|
],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_copy_to() {
|
|
let stmt = pg().verified_stmt("COPY users TO 'data.csv'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg().verified_stmt("COPY country TO STDOUT (DELIMITER '|')");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["country".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::Stdout,
|
|
options: vec![CopyOption::Delimiter('|')],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt =
|
|
pg().verified_stmt("COPY country TO PROGRAM 'gzip > /usr1/proj/bray/sql/country_data.gz'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["country".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::Program {
|
|
command: "gzip > /usr1/proj/bray/sql/country_data.gz".into(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![],
|
|
values: vec![],
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_copy_from_before_v9_0() {
|
|
let stmt = pg().verified_stmt("COPY users FROM 'data.csv' BINARY DELIMITER ',' NULL 'null' CSV HEADER QUOTE '\"' ESCAPE '\\' FORCE NOT NULL column");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![
|
|
CopyLegacyOption::Binary,
|
|
CopyLegacyOption::Delimiter(','),
|
|
CopyLegacyOption::Null("null".into()),
|
|
CopyLegacyOption::Csv(vec![
|
|
CopyLegacyCsvOption::Header,
|
|
CopyLegacyCsvOption::Quote('\"'),
|
|
CopyLegacyCsvOption::Escape('\\'),
|
|
CopyLegacyCsvOption::ForceNotNull(vec!["column".into()]),
|
|
]),
|
|
],
|
|
values: vec![],
|
|
}
|
|
);
|
|
|
|
// test 'AS' keyword
|
|
let sql = "COPY users FROM 'data.csv' DELIMITER AS ',' NULL AS 'null' CSV QUOTE AS '\"' ESCAPE AS '\\'";
|
|
assert_eq!(
|
|
pg_and_generic().one_statement_parses_to(sql, ""),
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: false,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![
|
|
CopyLegacyOption::Delimiter(','),
|
|
CopyLegacyOption::Null("null".into()),
|
|
CopyLegacyOption::Csv(vec![
|
|
CopyLegacyCsvOption::Quote('\"'),
|
|
CopyLegacyCsvOption::Escape('\\'),
|
|
]),
|
|
],
|
|
values: vec![],
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_copy_to_before_v9_0() {
|
|
let stmt = pg().verified_stmt("COPY users TO 'data.csv' BINARY DELIMITER ',' NULL 'null' CSV HEADER QUOTE '\"' ESCAPE '\\' FORCE QUOTE column");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Copy {
|
|
table_name: ObjectName(vec!["users".into()]),
|
|
columns: vec![],
|
|
to: true,
|
|
target: CopyTarget::File {
|
|
filename: "data.csv".to_string(),
|
|
},
|
|
options: vec![],
|
|
legacy_options: vec![
|
|
CopyLegacyOption::Binary,
|
|
CopyLegacyOption::Delimiter(','),
|
|
CopyLegacyOption::Null("null".into()),
|
|
CopyLegacyOption::Csv(vec![
|
|
CopyLegacyCsvOption::Header,
|
|
CopyLegacyCsvOption::Quote('\"'),
|
|
CopyLegacyCsvOption::Escape('\\'),
|
|
CopyLegacyCsvOption::ForceQuote(vec!["column".into()]),
|
|
]),
|
|
],
|
|
values: vec![],
|
|
}
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn parse_set() {
|
|
let stmt = pg_and_generic().verified_stmt("SET a = b");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a")]),
|
|
value: vec![SetVariableValue::Ident("b".into())],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET a = 'b'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a")]),
|
|
value: vec![SetVariableValue::Literal(Value::SingleQuotedString(
|
|
"b".into()
|
|
))],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET a = 0");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a")]),
|
|
value: vec![SetVariableValue::Literal(number("0"))],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET a = DEFAULT");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a")]),
|
|
value: vec![SetVariableValue::Ident("DEFAULT".into())],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET LOCAL a = b");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: true,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a")]),
|
|
value: vec![SetVariableValue::Ident("b".into())],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET a.b.c = b");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![Ident::new("a"), Ident::new("b"), Ident::new("c")]),
|
|
value: vec![SetVariableValue::Ident("b".into())],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().one_statement_parses_to(
|
|
"SET hive.tez.auto.reducer.parallelism=false",
|
|
"SET hive.tez.auto.reducer.parallelism = false",
|
|
);
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetVariable {
|
|
local: false,
|
|
hivevar: false,
|
|
variable: ObjectName(vec![
|
|
Ident::new("hive"),
|
|
Ident::new("tez"),
|
|
Ident::new("auto"),
|
|
Ident::new("reducer"),
|
|
Ident::new("parallelism")
|
|
]),
|
|
value: vec![SetVariableValue::Literal(Boolean(false))],
|
|
}
|
|
);
|
|
|
|
pg_and_generic().one_statement_parses_to("SET a TO b", "SET a = b");
|
|
pg_and_generic().one_statement_parses_to("SET SESSION a = b", "SET a = b");
|
|
|
|
assert_eq!(
|
|
pg_and_generic().parse_sql_statements("SET"),
|
|
Err(ParserError::ParserError(
|
|
"Expected identifier, found: EOF".to_string()
|
|
)),
|
|
);
|
|
|
|
assert_eq!(
|
|
pg_and_generic().parse_sql_statements("SET a b"),
|
|
Err(ParserError::ParserError(
|
|
"Expected equals sign or TO, found: b".to_string()
|
|
)),
|
|
);
|
|
|
|
assert_eq!(
|
|
pg_and_generic().parse_sql_statements("SET a ="),
|
|
Err(ParserError::ParserError(
|
|
"Expected variable value, found: EOF".to_string()
|
|
)),
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_set_role() {
|
|
let stmt = pg_and_generic().verified_stmt("SET SESSION ROLE NONE");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetRole {
|
|
local: false,
|
|
session: true,
|
|
role_name: None,
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET LOCAL ROLE \"rolename\"");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetRole {
|
|
local: true,
|
|
session: false,
|
|
role_name: Some(Ident {
|
|
value: "rolename".to_string(),
|
|
quote_style: Some('\"'),
|
|
}),
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SET ROLE 'rolename'");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::SetRole {
|
|
local: false,
|
|
session: false,
|
|
role_name: Some(Ident {
|
|
value: "rolename".to_string(),
|
|
quote_style: Some('\''),
|
|
}),
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_show() {
|
|
let stmt = pg_and_generic().verified_stmt("SHOW a a");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::ShowVariable {
|
|
variable: vec!["a".into(), "a".into()]
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("SHOW ALL ALL");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::ShowVariable {
|
|
variable: vec!["ALL".into(), "ALL".into()]
|
|
}
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn parse_deallocate() {
|
|
let stmt = pg_and_generic().verified_stmt("DEALLOCATE a");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Deallocate {
|
|
name: "a".into(),
|
|
prepare: false,
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("DEALLOCATE ALL");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Deallocate {
|
|
name: "ALL".into(),
|
|
prepare: false,
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("DEALLOCATE PREPARE a");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Deallocate {
|
|
name: "a".into(),
|
|
prepare: true,
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("DEALLOCATE PREPARE ALL");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Deallocate {
|
|
name: "ALL".into(),
|
|
prepare: true,
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_execute() {
|
|
let stmt = pg_and_generic().verified_stmt("EXECUTE a");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Execute {
|
|
name: "a".into(),
|
|
parameters: vec![],
|
|
}
|
|
);
|
|
|
|
let stmt = pg_and_generic().verified_stmt("EXECUTE a(1, 't')");
|
|
assert_eq!(
|
|
stmt,
|
|
Statement::Execute {
|
|
name: "a".into(),
|
|
parameters: vec![
|
|
Expr::Value(number("1")),
|
|
Expr::Value(Value::SingleQuotedString("t".to_string()))
|
|
],
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_prepare() {
|
|
let stmt =
|
|
pg_and_generic().verified_stmt("PREPARE a AS INSERT INTO customers VALUES (a1, a2, a3)");
|
|
let sub_stmt = match stmt {
|
|
Statement::Prepare {
|
|
name,
|
|
data_types,
|
|
statement,
|
|
..
|
|
} => {
|
|
assert_eq!(name, "a".into());
|
|
assert!(data_types.is_empty());
|
|
|
|
statement
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
match sub_stmt.as_ref() {
|
|
Statement::Insert {
|
|
table_name,
|
|
columns,
|
|
source,
|
|
..
|
|
} => {
|
|
assert_eq!(table_name.to_string(), "customers");
|
|
assert!(columns.is_empty());
|
|
|
|
let expected_values = [vec![
|
|
Expr::Identifier("a1".into()),
|
|
Expr::Identifier("a2".into()),
|
|
Expr::Identifier("a3".into()),
|
|
]];
|
|
match &*source.body {
|
|
SetExpr::Values(Values(values)) => assert_eq!(values.as_slice(), &expected_values),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
|
|
let stmt = pg_and_generic()
|
|
.verified_stmt("PREPARE a (INT, TEXT) AS SELECT * FROM customers WHERE customers.id = a1");
|
|
let sub_stmt = match stmt {
|
|
Statement::Prepare {
|
|
name,
|
|
data_types,
|
|
statement,
|
|
..
|
|
} => {
|
|
assert_eq!(name, "a".into());
|
|
assert_eq!(data_types, vec![DataType::Int(None), DataType::Text]);
|
|
|
|
statement
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
assert_eq!(
|
|
sub_stmt,
|
|
Box::new(Statement::Query(Box::new(pg_and_generic().verified_query(
|
|
"SELECT * FROM customers WHERE customers.id = a1"
|
|
))))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_pg_bitwise_binary_ops() {
|
|
let bitwise_ops = &[
|
|
("#", BinaryOperator::PGBitwiseXor),
|
|
(">>", BinaryOperator::PGBitwiseShiftRight),
|
|
("<<", BinaryOperator::PGBitwiseShiftLeft),
|
|
];
|
|
|
|
for (str_op, op) in bitwise_ops {
|
|
let select = pg().verified_only_select(&format!("SELECT a {} b", &str_op));
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::BinaryOp {
|
|
left: Box::new(Expr::Identifier(Ident::new("a"))),
|
|
op: op.clone(),
|
|
right: Box::new(Expr::Identifier(Ident::new("b"))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_pg_unary_ops() {
|
|
let pg_unary_ops = &[
|
|
("~", UnaryOperator::PGBitwiseNot),
|
|
("|/", UnaryOperator::PGSquareRoot),
|
|
("||/", UnaryOperator::PGCubeRoot),
|
|
("!!", UnaryOperator::PGPrefixFactorial),
|
|
("@", UnaryOperator::PGAbs),
|
|
];
|
|
|
|
for (str_op, op) in pg_unary_ops {
|
|
let select = pg().verified_only_select(&format!("SELECT {} a", &str_op));
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::UnaryOp {
|
|
op: op.clone(),
|
|
expr: Box::new(Expr::Identifier(Ident::new("a"))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_pg_postfix_factorial() {
|
|
let postfix_factorial = &[("!", UnaryOperator::PGPostfixFactorial)];
|
|
|
|
for (str_op, op) in postfix_factorial {
|
|
let select = pg().verified_only_select(&format!("SELECT a{}", &str_op));
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::UnaryOp {
|
|
op: op.clone(),
|
|
expr: Box::new(Expr::Identifier(Ident::new("a"))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_pg_regex_match_ops() {
|
|
let pg_regex_match_ops = &[
|
|
("~", BinaryOperator::PGRegexMatch),
|
|
("~*", BinaryOperator::PGRegexIMatch),
|
|
("!~", BinaryOperator::PGRegexNotMatch),
|
|
("!~*", BinaryOperator::PGRegexNotIMatch),
|
|
];
|
|
|
|
for (str_op, op) in pg_regex_match_ops {
|
|
let select = pg().verified_only_select(&format!("SELECT 'abc' {} '^a'", &str_op));
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::BinaryOp {
|
|
left: Box::new(Expr::Value(Value::SingleQuotedString("abc".into()))),
|
|
op: op.clone(),
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString("^a".into()))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_array_index_expr() {
|
|
#[cfg(feature = "bigdecimal")]
|
|
let num: Vec<Expr> = (0..=10)
|
|
.into_iter()
|
|
.map(|s| Expr::Value(Value::Number(bigdecimal::BigDecimal::from(s), false)))
|
|
.collect();
|
|
#[cfg(not(feature = "bigdecimal"))]
|
|
let num: Vec<Expr> = (0..=10)
|
|
.into_iter()
|
|
.map(|s| Expr::Value(Value::Number(s.to_string(), false)))
|
|
.collect();
|
|
|
|
let sql = "SELECT foo[0] FROM foos";
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(
|
|
&Expr::ArrayIndex {
|
|
obj: Box::new(Expr::Identifier(Ident::new("foo"))),
|
|
indexes: vec![num[0].clone()],
|
|
},
|
|
expr_from_projection(only(&select.projection)),
|
|
);
|
|
|
|
let sql = "SELECT foo[0][0] FROM foos";
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(
|
|
&Expr::ArrayIndex {
|
|
obj: Box::new(Expr::Identifier(Ident::new("foo"))),
|
|
indexes: vec![num[0].clone(), num[0].clone()],
|
|
},
|
|
expr_from_projection(only(&select.projection)),
|
|
);
|
|
|
|
let sql = r#"SELECT bar[0]["baz"]["fooz"] FROM foos"#;
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(
|
|
&Expr::ArrayIndex {
|
|
obj: Box::new(Expr::Identifier(Ident::new("bar"))),
|
|
indexes: vec![
|
|
num[0].clone(),
|
|
Expr::Identifier(Ident {
|
|
value: "baz".to_string(),
|
|
quote_style: Some('"')
|
|
}),
|
|
Expr::Identifier(Ident {
|
|
value: "fooz".to_string(),
|
|
quote_style: Some('"')
|
|
})
|
|
],
|
|
},
|
|
expr_from_projection(only(&select.projection)),
|
|
);
|
|
|
|
let sql = "SELECT (CAST(ARRAY[ARRAY[2, 3]] AS INT[][]))[1][2]";
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(
|
|
&Expr::ArrayIndex {
|
|
obj: Box::new(Expr::Nested(Box::new(Expr::Cast {
|
|
expr: Box::new(Expr::Array(Array {
|
|
elem: vec![Expr::Array(Array {
|
|
elem: vec![num[2].clone(), num[3].clone(),],
|
|
named: true,
|
|
})],
|
|
named: true,
|
|
})),
|
|
data_type: DataType::Array(Box::new(DataType::Array(Box::new(DataType::Int(
|
|
None
|
|
)))))
|
|
}))),
|
|
indexes: vec![num[1].clone(), num[2].clone()],
|
|
},
|
|
expr_from_projection(only(&select.projection)),
|
|
);
|
|
|
|
let sql = "SELECT ARRAY[]";
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(
|
|
&Expr::Array(sqlparser::ast::Array {
|
|
elem: vec![],
|
|
named: true
|
|
}),
|
|
expr_from_projection(only(&select.projection)),
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_transaction_statement() {
|
|
let statement = pg().verified_stmt("SET TRANSACTION SNAPSHOT '000003A1-1'");
|
|
assert_eq!(
|
|
statement,
|
|
Statement::SetTransaction {
|
|
modes: vec![],
|
|
snapshot: Some(Value::SingleQuotedString(String::from("000003A1-1"))),
|
|
session: false
|
|
}
|
|
);
|
|
let statement = pg().verified_stmt("SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY, READ WRITE, ISOLATION LEVEL SERIALIZABLE");
|
|
assert_eq!(
|
|
statement,
|
|
Statement::SetTransaction {
|
|
modes: vec![
|
|
TransactionMode::AccessMode(TransactionAccessMode::ReadOnly),
|
|
TransactionMode::AccessMode(TransactionAccessMode::ReadWrite),
|
|
TransactionMode::IsolationLevel(TransactionIsolationLevel::Serializable),
|
|
],
|
|
snapshot: None,
|
|
session: true
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_savepoint() {
|
|
match pg().verified_stmt("SAVEPOINT test1") {
|
|
Statement::Savepoint { name } => {
|
|
assert_eq!(Ident::new("test1"), name);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_json() {
|
|
let sql = "SELECT params ->> 'name' FROM events";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::JsonAccess {
|
|
left: Box::new(Expr::Identifier(Ident::new("params"))),
|
|
operator: JsonOperator::LongArrow,
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString("name".to_string()))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
|
|
let sql = "SELECT params -> 'name' FROM events";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::JsonAccess {
|
|
left: Box::new(Expr::Identifier(Ident::new("params"))),
|
|
operator: JsonOperator::Arrow,
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString("name".to_string()))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
|
|
let sql = "SELECT info -> 'items' ->> 'product' FROM orders";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::JsonAccess {
|
|
left: Box::new(Expr::Identifier(Ident::new("info"))),
|
|
operator: JsonOperator::Arrow,
|
|
right: Box::new(Expr::JsonAccess {
|
|
left: Box::new(Expr::Value(Value::SingleQuotedString("items".to_string()))),
|
|
operator: JsonOperator::LongArrow,
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString(
|
|
"product".to_string()
|
|
)))
|
|
}),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
|
|
let sql = "SELECT info #> '{a,b,c}' FROM orders";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::JsonAccess {
|
|
left: Box::new(Expr::Identifier(Ident::new("info"))),
|
|
operator: JsonOperator::HashArrow,
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString(
|
|
"{a,b,c}".to_string()
|
|
))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
|
|
let sql = "SELECT info #>> '{a,b,c}' FROM orders";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::JsonAccess {
|
|
left: Box::new(Expr::Identifier(Ident::new("info"))),
|
|
operator: JsonOperator::HashLongArrow,
|
|
right: Box::new(Expr::Value(Value::SingleQuotedString(
|
|
"{a,b,c}".to_string()
|
|
))),
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_composite_value() {
|
|
let sql = "SELECT (on_hand.item).name FROM on_hand WHERE (on_hand.item).price > 9";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::CompositeAccess {
|
|
key: Ident::new("name"),
|
|
expr: Box::new(Expr::Nested(Box::new(Expr::CompoundIdentifier(vec![
|
|
Ident::new("on_hand"),
|
|
Ident::new("item")
|
|
]))))
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
|
|
#[cfg(feature = "bigdecimal")]
|
|
let num: Expr = Expr::Value(Value::Number(bigdecimal::BigDecimal::from(9), false));
|
|
#[cfg(not(feature = "bigdecimal"))]
|
|
let num: Expr = Expr::Value(Value::Number("9".to_string(), false));
|
|
assert_eq!(
|
|
select.selection,
|
|
Some(Expr::BinaryOp {
|
|
left: Box::new(Expr::CompositeAccess {
|
|
key: Ident::new("price"),
|
|
expr: Box::new(Expr::Nested(Box::new(Expr::CompoundIdentifier(vec![
|
|
Ident::new("on_hand"),
|
|
Ident::new("item")
|
|
]))))
|
|
}),
|
|
op: BinaryOperator::Gt,
|
|
right: Box::new(num)
|
|
})
|
|
);
|
|
|
|
let sql = "SELECT (information_schema._pg_expandarray(ARRAY['i', 'i'])).n";
|
|
let select = pg().verified_only_select(sql);
|
|
assert_eq!(
|
|
SelectItem::UnnamedExpr(Expr::CompositeAccess {
|
|
key: Ident::new("n"),
|
|
expr: Box::new(Expr::Nested(Box::new(Expr::Function(Function {
|
|
name: ObjectName(vec![
|
|
Ident::new("information_schema"),
|
|
Ident::new("_pg_expandarray")
|
|
]),
|
|
args: vec![FunctionArg::Unnamed(FunctionArgExpr::Expr(Expr::Array(
|
|
Array {
|
|
elem: vec![
|
|
Expr::Value(Value::SingleQuotedString("i".to_string())),
|
|
Expr::Value(Value::SingleQuotedString("i".to_string())),
|
|
],
|
|
named: true
|
|
}
|
|
)))],
|
|
over: None,
|
|
distinct: false,
|
|
}))))
|
|
}),
|
|
select.projection[0]
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_comments() {
|
|
match pg().verified_stmt("COMMENT ON COLUMN tab.name IS 'comment'") {
|
|
Statement::Comment {
|
|
object_type,
|
|
object_name,
|
|
comment: Some(comment),
|
|
} => {
|
|
assert_eq!("comment", comment);
|
|
assert_eq!("tab.name", object_name.to_string());
|
|
assert_eq!(CommentObject::Column, object_type);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
|
|
match pg().verified_stmt("COMMENT ON TABLE public.tab IS 'comment'") {
|
|
Statement::Comment {
|
|
object_type,
|
|
object_name,
|
|
comment: Some(comment),
|
|
} => {
|
|
assert_eq!("comment", comment);
|
|
assert_eq!("public.tab", object_name.to_string());
|
|
assert_eq!(CommentObject::Table, object_type);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
|
|
match pg().verified_stmt("COMMENT ON TABLE public.tab IS NULL") {
|
|
Statement::Comment {
|
|
object_type,
|
|
object_name,
|
|
comment: None,
|
|
} => {
|
|
assert_eq!("public.tab", object_name.to_string());
|
|
assert_eq!(CommentObject::Table, object_type);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_quoted_identifier() {
|
|
pg_and_generic().verified_stmt(r#"SELECT "quoted "" ident""#);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_quoted_identifier_2() {
|
|
pg_and_generic().verified_stmt(r#"SELECT """quoted ident""""#);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_local_and_global() {
|
|
pg_and_generic().verified_stmt("CREATE LOCAL TEMPORARY TABLE table (COL INT)");
|
|
}
|
|
|
|
#[test]
|
|
fn parse_on_commit() {
|
|
pg_and_generic()
|
|
.verified_stmt("CREATE TEMPORARY TABLE table (COL INT) ON COMMIT PRESERVE ROWS");
|
|
|
|
pg_and_generic().verified_stmt("CREATE TEMPORARY TABLE table (COL INT) ON COMMIT DELETE ROWS");
|
|
|
|
pg_and_generic().verified_stmt("CREATE TEMPORARY TABLE table (COL INT) ON COMMIT DROP");
|
|
}
|
|
|
|
fn pg() -> TestedDialects {
|
|
TestedDialects {
|
|
dialects: vec![Box::new(PostgreSqlDialect {})],
|
|
}
|
|
}
|
|
|
|
fn pg_and_generic() -> TestedDialects {
|
|
TestedDialects {
|
|
dialects: vec![Box::new(PostgreSqlDialect {}), Box::new(GenericDialect {})],
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_escaped_literal_string() {
|
|
let sql =
|
|
r#"SELECT E's1 \n s1', E's2 \\n s2', E's3 \\\n s3', E's4 \\\\n s4', E'\'', E'foo \\'"#;
|
|
let select = pg_and_generic().verified_only_select(sql);
|
|
assert_eq!(6, select.projection.len());
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("s1 \n s1".to_string())),
|
|
expr_from_projection(&select.projection[0])
|
|
);
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("s2 \\n s2".to_string())),
|
|
expr_from_projection(&select.projection[1])
|
|
);
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("s3 \\\n s3".to_string())),
|
|
expr_from_projection(&select.projection[2])
|
|
);
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("s4 \\\\n s4".to_string())),
|
|
expr_from_projection(&select.projection[3])
|
|
);
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("'".to_string())),
|
|
expr_from_projection(&select.projection[4])
|
|
);
|
|
assert_eq!(
|
|
&Expr::Value(Value::EscapedStringLiteral("foo \\".to_string())),
|
|
expr_from_projection(&select.projection[5])
|
|
);
|
|
|
|
let sql = r#"SELECT E'\'"#;
|
|
assert_eq!(
|
|
pg_and_generic()
|
|
.parse_sql_statements(sql)
|
|
.unwrap_err()
|
|
.to_string(),
|
|
"sql parser error: Unterminated encoded string literal at Line: 1, Column 8"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_declare() {
|
|
pg_and_generic()
|
|
.verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" CURSOR WITH HOLD FOR SELECT 1");
|
|
pg_and_generic()
|
|
.verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" CURSOR WITHOUT HOLD FOR SELECT 1");
|
|
pg_and_generic().verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" BINARY CURSOR FOR SELECT 1");
|
|
pg_and_generic()
|
|
.verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" ASENSITIVE CURSOR FOR SELECT 1");
|
|
pg_and_generic()
|
|
.verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" INSENSITIVE CURSOR FOR SELECT 1");
|
|
pg_and_generic().verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" SCROLL CURSOR FOR SELECT 1");
|
|
pg_and_generic()
|
|
.verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" NO SCROLL CURSOR FOR SELECT 1");
|
|
pg_and_generic().verified_stmt("DECLARE \"SQL_CUR0x7fa44801bc00\" BINARY INSENSITIVE SCROLL CURSOR WITH HOLD FOR SELECT * FROM table_name LIMIT 2222");
|
|
}
|
|
|
|
#[test]
|
|
fn parse_fetch() {
|
|
pg_and_generic().verified_stmt("FETCH 2048 IN \"SQL_CUR0x7fa44801bc00\"");
|
|
pg_and_generic().verified_stmt("FETCH 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH NEXT IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH PRIOR IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH FIRST IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH LAST IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH ABSOLUTE 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH RELATIVE 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic().verified_stmt("FETCH ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH FORWARD 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH FORWARD ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH BACKWARD 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
pg_and_generic()
|
|
.verified_stmt("FETCH BACKWARD ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
|
|
}
|