From 39791b4552b605fe60d03a5e98c5533bd741b729 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bensima.com>
Date: Thu, 27 Nov 2025 15:21:18 -0500
Subject: Fix SQL query and add simple migration test

---
 Omni/Task/MigrationTest.hs | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 Omni/Task/MigrationTest.hs

(limited to 'Omni/Task/MigrationTest.hs')

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