92 lines
2.5 KiB
Rust
92 lines
2.5 KiB
Rust
mod migrator;
|
|
use migrator::Migrator;
|
|
|
|
use sea_orm::{Database, DbErr};
|
|
use sea_orm_migration::prelude::*;
|
|
|
|
#[async_std::test]
|
|
async fn main() -> Result<(), DbErr> {
|
|
let url = std::env::var("DATABASE_URL").expect("Environment variable 'DATABASE_URL' not set");
|
|
let db_name = "sea_orm_migration";
|
|
let url = format!("{}/{}", url, db_name);
|
|
let db = &Database::connect(&url).await?;
|
|
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?);
|
|
|
|
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(())
|
|
}
|