#sea-orm #database-migrations #testing #sqlite

assert-migrator-reversible

For testing if Sea Orm Migrators are reversible

11 stable releases (5 major)

6.2.0 Sep 10, 2024
6.1.0 May 13, 2024
6.0.0 Mar 23, 2024
5.0.0 Aug 13, 2023
1.1.0 Oct 30, 2022

#459 in Database interfaces

22 downloads per month

MIT license

29KB
460 lines

Assert Migrator Reversible
for Sea Orm

crate docs

A crate for testing Sea Orm Migrators. To check if when you call up and then down on them. They work in both directions.

It runs your migrations up and down one at a time. Taking a look at the differences it does to a database. Checking if the reverse returns a database into it's previous state.

Supports Postgres and SQLite migrations.

Example

The most common use case is simply to test if your Migrator is reversible. In a test. Then error if it is not.

To do this add the following test to your migrations project ...

assert-migrator-reversible = "1"
#[cfg(test)]
mod test_migrator {
    use crate::path::to::my::Migrator;
    use ::assert_migrator_reversible::assert_migrator_reversible;

    #[test]
    fn it_should_have_reversible_migrations() {
        assert_migrator_reversible(Migrator, None);
    }
}

This test will use SQLite by default (that's what passing in None does).

Example with tokio::test

If you are already using Tokio to test your project. Then the following may be better.

assert-migrator-reversible = { version = "1", default-features = false }
#[cfg(test)]
mod test_migrator {
    use crate::path::to::my::Migrator;
    use ::assert_migrator_reversible::assert_migrator_reversible_async;

    #[tokio::test]
    async fn it_should_have_reversible_migrations() {
        assert_migrator_reversible_async(Migrator, None).await
    }
}

(Again, this example will use SQLite as it passes in None).

Example using PostgresSQL

Testing against Postgres requires passing in the url to the database. It will not be picked up by default from the environment variables. You must specify it.

#[cfg(test)]
mod test_migrator {
    use crate::path::to::my::Migrator;
    use ::assert_migrator_reversible::assert_migrator_reversible;
    use ::assert_migrator_reversible::DbConnection;

    const POSTGRES_DB_URL : &'static str = &"postgres://user:password@localhost:5432/my_database";

    #[test]
    fn it_should_have_reversible_migrations() {
        let db_conn = Some(DbConnection::Url(POSTGRES_DB_URL));
        assert_migrator_reversible(Migrator, db_conn);
    }
}

Example using own DatabaseConnection

You can also build your own Sea Orm DatabaseConnection object and pass this in for use ...

#[cfg(test)]
mod test_migrator {
    use crate::path::to::my::Migrator;

    use ::assert_migrator_reversible::assert_migrator_reversible;
    use ::assert_migrator_reversible::DbConnection;

    use ::sea_orm_migration::sea_orm::Database;
    use ::sea_orm_migration::sea_orm::DatabaseConnection;

    const POSTGRES_DB_URL : &'static str = &"postgres://user:password@localhost:5432/my_database";

    #[tokio::test]
    async fn it_should_have_reversible_migrations() {
        let db_connection = Database::connect(db_url)
            .await
            .expect("expect DatabaseConnection to be created");

        let db_conn = Some(DbConnection::DatabaseConnection(db_connection));
        assert_migrator_reversible_async(Migrator, db_conn).await
    }
}

Caveats

  • This only checks DB Table structure changes. It does not look for other changes. Such as data, enums, indexes, sql functions, etc.
  • The default option is to use an in-memory SQLite database; this is quite limited as lots of features aren't supported.
  • Testing against Postgres requires spinning up your own Postgres server. This crate will not do that for you.
  • It does not support MySQL (maybe in the future).

API

The library provides two non-async functions. These handle the async bits for you by bundling Tokio.

  • assert_migrator_reversible - The main way to test if your Migrator is reversible. Pass in a Migrator. It'll run it up and down. If it isn't reversible, it will panic.
  • find_index_of_non_reversible_migration - This is very similar to assert_migrator_reversible. It will find a migration that isn't reversible. When found, it will return the index. It will not panic.

Async versions of those functions are available. This is useful if you wish to not have this import Tokio (as it's quite big). Instead handle this yourself.

Features

  • tokio Default - This adds Tokio support. Which enables the functions assert_migrator_reversible and find_index_of_non_reversible_migration. This makes testing easier and simpler. You might want to disable this if you are already using Tokio in your tests, and wish to make this dependency smaller.
  • runtime-actix-native-tls - Sets Sea-Orm Migrations to use this runtime.
  • runtime-actix-rustls - Sets Sea-Orm Migrations to use this runtime.
  • runtime-async-std-native-tls - Sets Sea-Orm Migrations to use this runtime.
  • runtime-async-std-rustls - Sets Sea-Orm Migrations to use this runtime.
  • runtime-tokio-native-tls Default - Sets Sea-Orm Migrations to use this runtime.
  • runtime-tokio-rustls - Sets Sea-Orm Migrations to use this runtime.

Local development

To run the tests for this. You will need Docker installed, and to start the Postgres Docker image first.

The exact steps to run the tests are ...

./scripts/start-postgres.sh
cargo test
./scripts/stop-postgres.sh

Dependencies

~60MB
~1M SLoC