use arel::prelude::*;
#[arel::arel]
struct User {
id: i64,
desc: String,
done: Option<bool>,
}
#[cfg(test)]
#[cfg(feature = "sqlite")]
mod sqlite_sqlx {
use super::*;
async fn init_db() -> anyhow::Result<()> {
let db_state = arel::visitors::get_or_init_db_state(|| sqlx::any::AnyPoolOptions::new().max_connections(5).connect("sqlite::memory:")).await?;
sqlx::query("CREATE TABLE IF NOT EXISTS users
(
id INTEGER PRIMARY KEY NOT NULL,
desc TEXT NOT NULL,
done BOOLEAN NOT NULL DEFAULT 0
);"
).execute(db_state.pool()).await?;
for i in 0..10 {
sqlx::query(&User::create(json!({
"desc": format!("test-{}", i)
})).to_sql()?).execute(db_state.pool()).await?;
}
Ok(())
}
async fn main_test() -> anyhow::Result<()> {
init_db().await?;
assert_eq!(User::table_column_names(), vec!["id", "desc", "done"]);
assert_eq!(User::id_table_column_name(), "id");
assert_eq!(User::desc_table_column_name(), "desc");
assert_eq!(User::done_table_column_name(), "done");
let user = User::query().fetch_one().await?;
assert_eq!(user.id(), Some(&1));
let users = User::query().fetch_all().await?;
assert_eq!(users.len(), 10);
User::update_all(json!({"desc": "update_1"})).execute().await?;
let user = User::query().fetch_one().await?;
assert_eq!(user.desc(), Some(&"update_1".to_string()));
Ok(())
}
#[test]
fn test_sqlite_sqlx() {
assert!(tokio_test::block_on(main_test()).is_ok());
}
}