mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-07-08 01:15:00 +00:00
09ca14fe8e
In MySQL it's AUTO_INCREMENT (see https://dev.mysql.com/doc/refman/8.0/en/create-table.html) and in SQLite it's AUTOINCREMENT. We use `ColumnOption::DialectSpecific(Vec<Token>)` to avoid adding a new variant for each vendor-specific column option.
95 lines
3.1 KiB
Rust
95 lines
3.1 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 SQLite. The parser based on the
|
|
//! generic dialect is also tested (on the inputs it can handle).
|
|
|
|
use sqlparser::ast::*;
|
|
use sqlparser::dialect::GenericDialect;
|
|
use sqlparser::test_utils::*;
|
|
use sqlparser::tokenizer::Token;
|
|
|
|
#[test]
|
|
fn parse_create_table_without_rowid() {
|
|
let sql = "CREATE TABLE t (a INT) WITHOUT ROWID";
|
|
match sqlite_and_generic().verified_stmt(sql) {
|
|
Statement::CreateTable {
|
|
name,
|
|
without_rowid: true,
|
|
..
|
|
} => {
|
|
assert_eq!("t", name.to_string());
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_virtual_table() {
|
|
let sql = "CREATE VIRTUAL TABLE IF NOT EXISTS t USING module_name (arg1, arg2)";
|
|
match sqlite_and_generic().verified_stmt(sql) {
|
|
Statement::CreateVirtualTable {
|
|
name,
|
|
if_not_exists: true,
|
|
module_name,
|
|
module_args,
|
|
} => {
|
|
let args = vec![Ident::new("arg1"), Ident::new("arg2")];
|
|
assert_eq!("t", name.to_string());
|
|
assert_eq!("module_name", module_name.to_string());
|
|
assert_eq!(args, module_args);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
|
|
let sql = "CREATE VIRTUAL TABLE t USING module_name";
|
|
sqlite_and_generic().verified_stmt(sql);
|
|
}
|
|
|
|
#[test]
|
|
fn parse_create_table_auto_increment() {
|
|
let sql = "CREATE TABLE foo (bar INT PRIMARY KEY AUTOINCREMENT)";
|
|
match sqlite_and_generic().verified_stmt(sql) {
|
|
Statement::CreateTable { name, columns, .. } => {
|
|
assert_eq!(name.to_string(), "foo");
|
|
assert_eq!(
|
|
vec![ColumnDef {
|
|
name: "bar".into(),
|
|
data_type: DataType::Int,
|
|
collation: None,
|
|
options: vec![
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::Unique { is_primary: true }
|
|
},
|
|
ColumnOptionDef {
|
|
name: None,
|
|
option: ColumnOption::DialectSpecific(vec![Token::make_keyword(
|
|
"AUTOINCREMENT"
|
|
)])
|
|
}
|
|
],
|
|
}],
|
|
columns
|
|
);
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
fn sqlite_and_generic() -> TestedDialects {
|
|
TestedDialects {
|
|
// we don't have a separate SQLite dialect, so test only the generic dialect for now
|
|
dialects: vec![Box::new(GenericDialect {})],
|
|
}
|
|
}
|