caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor basic test

Browse Source
This commit is contained in:
Chris Tsang 2021-09-18 16:37:07 +08:00
parent 0a19db6977
commit 3001fca466
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/basic.rs
View File

@ -2,22 +2,21 @@ pub mod common;
pub use sea_orm::{entity::*, error::*, sea_query, tests_cfg::*, Database, DbConn}; pub use sea_orm::{entity::*, error::*, sea_query, tests_cfg::*, Database, DbConn};
// DATABASE_URL="sqlite::memory:" cargo test --features sqlx-sqlite,runtime-async-std-native-tls --test basic // cargo test --features sqlx-sqlite,runtime-async-std-native-tls --test basic
#[sea_orm_macros::test] #[sea_orm_macros::test]
#[cfg(feature = "sqlx-sqlite")] #[cfg(feature = "sqlx-sqlite")]
async fn main() { async fn main() -> Result<(), DbErr> {
use std::env; let base_url = std::env::var("DATABASE_URL").unwrap_or_else(|_| "sqlite::memory:".to_owned());
let base_url = env::var("DATABASE_URL").unwrap_or_else(|_| "sqlite::memory:".to_owned());
let db: DbConn = Database::connect(&base_url).await.unwrap(); let db: DbConn = Database::connect(&base_url).await?;
setup_schema(&db).await?;
crud_cake(&db).await?;
setup_schema(&db).await; Ok(())
crud_cake(&db).await.unwrap();
} }
#[cfg(feature = "sqlx-sqlite")] #[cfg(feature = "sqlx-sqlite")]
async fn setup_schema(db: &DbConn) { async fn setup_schema(db: &DbConn) -> Result<(), DbErr> {
use sea_query::*; use sea_query::*;
let stmt = sea_query::Table::create() let stmt = sea_query::Table::create()
@ -33,8 +32,10 @@ async fn setup_schema(db: &DbConn) {
.to_owned(); .to_owned();
let builder = db.get_database_backend(); let builder = db.get_database_backend();
let result = db.execute(builder.build(&stmt)).await; let result = db.execute(builder.build(&stmt)).await?;
println!("Create table cake: {:?}", result); println!("Create table cake: {:?}", result);
Ok(())
} }
#[cfg(feature = "sqlx-sqlite")] #[cfg(feature = "sqlx-sqlite")]