mod common;
use common::TestDatabase;
#[tokio::test]
async fn run_reversible_migrations() {
let all_migrations: Vec<i64> = vec![
20230101000000,
20230201000000,
20230301000000,
20230401000000,
20230501000000,
];
{
let db = TestDatabase::new("migrate_run_reversible_latest", "migrations_reversible");
db.run_migration(false, None, false).success();
assert_eq!(db.applied_migrations().await, all_migrations);
}
{
let db = TestDatabase::new(
"migrate_run_reversible_latest_explicit",
"migrations_reversible",
);
db.run_migration(false, Some(20230501000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations);
db.run_migration(false, Some(20230501000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations);
db.run_migration(false, Some(20230301000000), false)
.failure();
assert_eq!(db.applied_migrations().await, all_migrations);
}
{
let db = TestDatabase::new(
"migrate_run_reversible_incremental",
"migrations_reversible",
);
db.run_migration(false, Some(20230101000000), false)
.success();
assert_eq!(db.applied_migrations().await, vec![20230101000000]);
db.run_migration(false, None, true).success();
assert_eq!(db.applied_migrations().await, vec![20230101000000]);
db.run_migration(false, Some(20230301000000), true)
.success();
assert_eq!(db.applied_migrations().await, vec![20230101000000]);
db.run_migration(false, Some(20230901000000999), false)
.failure();
assert_eq!(db.applied_migrations().await, vec![20230101000000]);
db.run_migration(false, Some(20230201000000), false)
.success();
assert_eq!(
db.applied_migrations().await,
vec![20230101000000, 20230201000000]
);
db.run_migration(false, Some(20230401000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations[..4]);
}
}
#[tokio::test]
async fn revert_migrations() {
let all_migrations: Vec<i64> = vec![
20230101000000,
20230201000000,
20230301000000,
20230401000000,
20230501000000,
];
{
let db = TestDatabase::new("migrate_revert_incremental", "migrations_reversible");
db.run_migration(false, None, false).success();
db.run_migration(true, None, true).success();
assert_eq!(db.applied_migrations().await, all_migrations);
db.run_migration(true, None, false).success();
assert_eq!(db.applied_migrations().await, all_migrations[..4]);
db.run_migration(true, None, false).success();
assert_eq!(db.applied_migrations().await, all_migrations[..3]);
}
{
let db = TestDatabase::new("migrate_revert_incremental", "migrations_reversible");
db.run_migration(false, None, false).success();
db.run_migration(true, Some(20230301000000), true).success();
assert_eq!(db.applied_migrations().await, all_migrations);
db.run_migration(true, Some(20230301000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations[..3]);
db.run_migration(true, Some(20230301000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations[..3]);
db.run_migration(true, Some(20230401000000), false)
.failure();
assert_eq!(db.applied_migrations().await, all_migrations[..3]);
db.run_migration(true, Some(9999), false).failure();
assert_eq!(db.applied_migrations().await, all_migrations[..3]);
db.run_migration(false, Some(20230401000000), false)
.success();
assert_eq!(db.applied_migrations().await, all_migrations[..4]);
db.run_migration(true, Some(0), false).success();
assert_eq!(db.applied_migrations().await, vec![] as Vec<i64>);
}
}