From dbd3c780f27bd5bee23adf6e280b84d669230e0d Mon Sep 17 00:00:00 2001
From: mo khan <mo@mokhan.ca>
Date: Wed, 11 Jun 2025 15:48:45 -0600
Subject: test: fix commented out tests

---
 src/migrations.rs | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

(limited to 'src/migrations.rs')

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<()> {
-- 
cgit v1.2.3