1 files changed, 42 insertions, 0 deletions
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