test(migrate): add latest_version check in reversible migration tests

Author: breuhan

sqlx-core/src/migrate/migrator.rs

@@ -110,6 +110,24 @@ impl Migrator {
         self.iter().any(|m| m.version == version)
     }
 
+    /// Get the latest version of the applied migrations.
+    pub async fn latest_version<'a, A>(&self, migrator: A) -> Result<Option<i64>, MigrateError>
+    where
+        A: Acquire<'a>,
+        <A::Connection as Deref>::Target: Migrate,
+    {
+        let mut conn = migrator.acquire().await?;
+
+        conn.ensure_migrations_table().await?;
+        let applied_migrations = conn.list_applied_migrations().await?;
+        let latest_version = applied_migrations
+            .iter()
+            .max_by(|x, y| x.version.cmp(&y.version))
+            .map(|migration| migration.version);
+
+        Ok(latest_version)
+    }
+
     /// Run any pending migrations against the database; and, validate previously applied migrations
     /// against the current migration source to detect accidental changes in previously-applied migrations.
     ///

tests/mysql/migrate.rs

@@ -48,6 +48,9 @@ async fn reversible(mut conn: PoolConnection<MySql>) -> anyhow::Result<()> {
         .get(0);
     assert_eq!(res, 101);
 
+    let latest_version = migrator.latest_version(&mut conn).await?.unwrap();
+    assert_eq!(latest_version, 20220721125033);
+
     // roll back nothing (last version)
     migrator.undo(&mut conn, 20220721125033).await?;
 

tests/postgres/migrate.rs

@@ -50,6 +50,9 @@ async fn reversible(mut conn: PoolConnection<Postgres>) -> anyhow::Result<()> {
         .get(0);
     assert_eq!(res, 101);
 
+    let latest_version = migrator.latest_version(&mut conn).await?.unwrap();
+    assert_eq!(latest_version, 20220721125033);
+
     // roll back nothing (last version)
     migrator.undo(&mut conn, 20220721125033).await?;
 

tests/sqlite/migrate.rs

@@ -48,6 +48,9 @@ async fn reversible(mut conn: PoolConnection<Sqlite>) -> anyhow::Result<()> {
         .get(0);
     assert_eq!(res, 101);
 
+    let latest_version = migrator.latest_version(&mut conn).await?.unwrap();
+    assert_eq!(latest_version, 20220721125033);
+
     // roll back nothing (last version)
     migrator.undo(&mut conn, 20220721125033).await?;