Fix SQLite test suite

This commit is contained in:
Chris Tsang 2021-09-18 16:16:33 +08:00
parent 68787d1412
commit 0a19db6977

View File

@ -1,6 +1,6 @@
pub use super::super::bakery_chain::*; pub use super::super::bakery_chain::*;
use pretty_assertions::assert_eq; use pretty_assertions::assert_eq;
use sea_orm::{error::*, sea_query, DbConn, EntityTrait, ExecResult, Schema}; use sea_orm::{error::*, sea_query, DbBackend, DbConn, EntityTrait, ExecResult, Schema};
use sea_query::{ use sea_query::{
Alias, ColumnDef, ForeignKey, ForeignKeyAction, Index, Table, TableCreateStatement, Alias, ColumnDef, ForeignKey, ForeignKeyAction, Index, Table, TableCreateStatement,
}; };
@ -14,13 +14,15 @@ where
E: EntityTrait, E: EntityTrait,
{ {
let builder = db.get_database_backend(); let builder = db.get_database_backend();
let stmt = builder.build( if builder != DbBackend::Sqlite {
Table::drop() let stmt = builder.build(
.table(Alias::new(create.get_table_name().unwrap().as_ref())) Table::drop()
.if_exists() .table(Alias::new(create.get_table_name().unwrap().as_ref()))
.cascade(), .if_exists()
); .cascade(),
db.execute(stmt).await?; );
db.execute(stmt).await?;
}
let stmt = builder.build(create); let stmt = builder.build(create);
assert_eq!( assert_eq!(