127 lines
3.6 KiB
Rust
127 lines
3.6 KiB
Rust
mod migrator;
|
|
use migrator::Migrator;
|
|
|
|
use sea_orm::{ConnectionTrait, Database, DbBackend, DbErr, Statement};
|
|
use sea_orm_migration::prelude::*;
|
|
|
|
#[async_std::test]
|
|
async fn main() -> Result<(), DbErr> {
|
|
tracing_subscriber::fmt()
|
|
.with_max_level(tracing::Level::DEBUG)
|
|
.with_test_writer()
|
|
.init();
|
|
|
|
let url = std::env::var("DATABASE_URL").expect("Environment variable 'DATABASE_URL' not set");
|
|
let db_name = "sea_orm_migration";
|
|
let db = Database::connect(&url).await?;
|
|
let db = &match db.get_database_backend() {
|
|
DbBackend::MySql => {
|
|
db.execute(Statement::from_string(
|
|
db.get_database_backend(),
|
|
format!("CREATE DATABASE IF NOT EXISTS `{}`;", db_name),
|
|
))
|
|
.await?;
|
|
|
|
let url = format!("{}/{}", url, db_name);
|
|
Database::connect(&url).await?
|
|
}
|
|
DbBackend::Postgres => {
|
|
db.execute(Statement::from_string(
|
|
db.get_database_backend(),
|
|
format!("DROP DATABASE IF EXISTS \"{}\";", db_name),
|
|
))
|
|
.await?;
|
|
db.execute(Statement::from_string(
|
|
db.get_database_backend(),
|
|
format!("CREATE DATABASE \"{}\";", db_name),
|
|
))
|
|
.await?;
|
|
|
|
let url = format!("{}/{}", url, db_name);
|
|
Database::connect(&url).await?
|
|
}
|
|
DbBackend::Sqlite => db,
|
|
};
|
|
let manager = SchemaManager::new(db);
|
|
|
|
println!("\nMigrator::status");
|
|
Migrator::status(db).await?;
|
|
|
|
println!("\nMigrator::install");
|
|
Migrator::install(db).await?;
|
|
|
|
assert!(manager.has_table("seaql_migrations").await?);
|
|
|
|
println!("\nMigrator::reset");
|
|
Migrator::reset(db).await?;
|
|
|
|
assert!(!manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::up");
|
|
Migrator::up(db, Some(0)).await?;
|
|
|
|
assert!(!manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::up");
|
|
Migrator::up(db, Some(1)).await?;
|
|
|
|
assert!(manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::down");
|
|
Migrator::down(db, Some(0)).await?;
|
|
|
|
assert!(manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::down");
|
|
Migrator::down(db, Some(1)).await?;
|
|
|
|
assert!(!manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::up");
|
|
Migrator::up(db, None).await?;
|
|
|
|
println!("\nMigrator::status");
|
|
Migrator::status(db).await?;
|
|
|
|
assert!(manager.has_table("cake").await?);
|
|
assert!(manager.has_table("fruit").await?);
|
|
|
|
assert!(manager.has_column("cake", "name").await?);
|
|
assert!(manager.has_column("fruit", "cake_id").await?);
|
|
|
|
println!("\nMigrator::down");
|
|
Migrator::down(db, None).await?;
|
|
|
|
assert!(manager.has_table("seaql_migrations").await?);
|
|
assert!(!manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::fresh");
|
|
Migrator::fresh(db).await?;
|
|
|
|
assert!(manager.has_table("cake").await?);
|
|
assert!(manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::refresh");
|
|
Migrator::refresh(db).await?;
|
|
|
|
assert!(manager.has_table("cake").await?);
|
|
assert!(manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::reset");
|
|
Migrator::reset(db).await?;
|
|
|
|
assert!(!manager.has_table("cake").await?);
|
|
assert!(!manager.has_table("fruit").await?);
|
|
|
|
println!("\nMigrator::status");
|
|
Migrator::status(db).await?;
|
|
|
|
Ok(())
|
|
}
|