2 files changed, 44 insertions, 1 deletions
diff --git a/Omni/Task/Core.hs b/Omni/Task/Core.hs
index 4eac9b5..63bc533 100644
--- a/Omni/Task/Core.hs
+++ b/Omni/Task/Core.hs
@@ -437,7 +437,8 @@ migrateTable conn tableName expectedCols = do
 -- | Get list of column names for a table using PRAGMA table_info
 getTableColumns :: SQL.Connection -> Text -> IO [Text]
 getTableColumns conn tableName = do
-  rows <- SQL.query conn "PRAGMA table_info(?)" (SQL.Only tableName) :: IO [(Int, Text, Text, Int, Maybe Text, Int)]
+  let query = SQL.Query <| "PRAGMA table_info(" <> tableName <> ")"
+  rows <- SQL.query_ conn query :: IO [(Int, Text, Text, Int, Maybe Text, Int)]
   pure [colName | (_, colName, _, _, _, _) <- rows]
 
 -- | Add a column to a table
diff --git a/Omni/Task/MigrationTest.hs b/Omni/Task/MigrationTest.hs
new file mode 100644
index 0000000..f16f782
--- /dev/null
+++ b/Omni/Task/MigrationTest.hs
@@ -0,0 +1,42 @@
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE NoImplicitPrelude #-}
+
+module Omni.Task.MigrationTest where
+
+import Alpha
+import qualified Data.Set as Set
+import Omni.Task.Core
+import qualified Omni.Test as Test
+import System.Directory (doesFileExist, removeFile)
+import System.Environment (setEnv)
+
+test :: Test.Tree
+test = Test.group "Omni.Task.Migration" [migrationStartupTest]
+
+migrationStartupTest :: Test.Tree
+migrationStartupTest =
+  Test.unit "database initializes with schema migrations" <| do
+    setEnv "TASK_TEST_MODE" "1"
+
+    let testFile = "_/tmp/tasks-test.db"
+    exists <- doesFileExist testFile
+    when exists <| removeFile testFile
+
+    initTaskDb
+
+    withDb <| \conn -> do
+      tasksCols <- getTableColumns conn "tasks"
+      activityCols <- getTableColumns conn "task_activity"
+      retryCols <- getTableColumns conn "retry_context"
+
+      Set.fromList ["id", "title", "status"]
+        `Set.isSubsetOf` Set.fromList tasksCols
+        Test.@?= True
+      Set.fromList ["id", "task_id", "stage"]
+        `Set.isSubsetOf` Set.fromList activityCols
+        Test.@?= True
+      Set.fromList ["task_id", "attempt", "reason"]
+        `Set.isSubsetOf` Set.fromList retryCols
+        Test.@?= True
+
+    removeFile testFile