test: fix commented out tests

1 files changed, 21 insertions, 5 deletions

diff --git a/src/migrations.rs b/src/migrations.rs
index 5076a9e..61c5b19 100644
--- a/src/migrations.rs
+++ b/src/migrations.rs
@@ -62,12 +62,28 @@ impl<'a> MigrationRunner<'a> {
     }
 
     fn get_current_version(&self) -> Result<i32> {
-        let version = self.conn.query_row(
-            "SELECT COALESCE(MAX(version), 0) FROM schema_migrations",
+        // Check if schema_migrations table exists first
+        let table_exists = self.conn.query_row(
+            "SELECT name FROM sqlite_master WHERE type='table' AND name='schema_migrations'",
             [],
-            |row| row.get::<_, i32>(0),
-        )?;
-        Ok(version)
+            |_| Ok(()),
+        );
+
+        match table_exists {
+            Ok(_) => {
+                // Table exists, get the current version
+                let version = self.conn.query_row(
+                    "SELECT COALESCE(MAX(version), 0) FROM schema_migrations",
+                    [],
+                    |row| row.get::<_, i32>(0),
+                )?;
+                Ok(version)
+            }
+            Err(_) => {
+                // Table doesn't exist, we're at version 0
+                Ok(0)
+            }
+        }
     }
 
     fn run_migration(&self, migration: &Migration) -> Result<()> {