summaryrefslogtreecommitdiff
path: root/internal
diff options
context:
space:
mode:
authorPeter Stone <thepeterstone@gmail.com>2026-07-16 06:08:23 +0000
committerPeter Stone <thepeterstone@gmail.com>2026-07-16 06:08:23 +0000
commit310d17b25d8dbcd8e930148334e7bd7bb144f407 (patch)
tree4a7fbb3eb04160b1495f1881fb708386df3aee06 /internal
parent31d8d9b32a3ac68d3d939bd080eeb780c37f17d7 (diff)
Add schema for task budgets and availability blocks
Creates migrations/025_task_budgets_and_availability.sql with: - New availability_blocks table for manual weekly availability templates - estimated_minutes column on native_tasks for time budgeting - budget_tracked flag on projects and labels for opt-in tracking Fixes pre-existing bug in sqlite_test.go's setupTestStoreWithNativeTasks where project_id column was missing from the schema, causing test failures. Updates all hand-rolled test helpers (sqlite_test.go, native_tasks_test.go) to include the new columns, and adds availability_test.go with newAvailabilityTestStore helper for Task 4's availability CRUD tests. All store tests pass, including the two that were previously failing (TestGetNativeTasksByDateRange_ExcludesOverdue, TestGetOverdueNativeTasks_IncludesOnlyPastDue). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01PQaPGQVSfmKUiHXB87qRTC
Diffstat (limited to 'internal')
-rw-r--r--internal/store/availability_test.go33
-rw-r--r--internal/store/native_tasks_test.go9
-rw-r--r--internal/store/sqlite_test.go9
3 files changed, 47 insertions, 4 deletions
diff --git a/internal/store/availability_test.go b/internal/store/availability_test.go
new file mode 100644
index 0000000..97ce90e
--- /dev/null
+++ b/internal/store/availability_test.go
@@ -0,0 +1,33 @@
+package store
+
+import (
+ "database/sql"
+ "path/filepath"
+ "testing"
+
+ _ "github.com/mattn/go-sqlite3"
+)
+
+// newAvailabilityTestStore creates a Store backed by a fresh temp sqlite DB
+// with just the availability_blocks table.
+func newAvailabilityTestStore(t *testing.T) *Store {
+ t.Helper()
+ dbPath := filepath.Join(t.TempDir(), "test.db")
+ db, err := sql.Open("sqlite3", dbPath)
+ if err != nil {
+ t.Fatal(err)
+ }
+ t.Cleanup(func() { db.Close() })
+ if _, err := db.Exec(`
+ CREATE TABLE availability_blocks (
+ id TEXT PRIMARY KEY,
+ weekday INTEGER NOT NULL,
+ start_time TEXT NOT NULL,
+ end_time TEXT NOT NULL,
+ label TEXT DEFAULT ''
+ )
+ `); err != nil {
+ t.Fatal(err)
+ }
+ return &Store{db: db}
+}
diff --git a/internal/store/native_tasks_test.go b/internal/store/native_tasks_test.go
index cf9ea49..1b4bdf2 100644
--- a/internal/store/native_tasks_test.go
+++ b/internal/store/native_tasks_test.go
@@ -41,7 +41,8 @@ func newNativeTasksTestStore(t *testing.T) *Store {
recurrence_interval INTEGER DEFAULT 1,
recurrence_weekdays TEXT DEFAULT '',
recurrence_series_id TEXT DEFAULT '',
- next_occurrence_override TEXT DEFAULT ''
+ next_occurrence_override TEXT DEFAULT '',
+ estimated_minutes INTEGER DEFAULT 0
)
`); err != nil {
t.Fatal(err)
@@ -52,7 +53,8 @@ func newNativeTasksTestStore(t *testing.T) *Store {
name TEXT NOT NULL,
color TEXT DEFAULT '',
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
- archived BOOLEAN DEFAULT 0
+ archived BOOLEAN DEFAULT 0,
+ budget_tracked BOOLEAN DEFAULT 0
)
`); err != nil {
t.Fatal(err)
@@ -60,7 +62,8 @@ func newNativeTasksTestStore(t *testing.T) *Store {
if _, err := db.Exec(`
CREATE TABLE labels (
name TEXT PRIMARY KEY,
- color TEXT NOT NULL
+ color TEXT NOT NULL,
+ budget_tracked BOOLEAN DEFAULT 0
)
`); err != nil {
t.Fatal(err)
diff --git a/internal/store/sqlite_test.go b/internal/store/sqlite_test.go
index 3c5a7b6..345b40e 100644
--- a/internal/store/sqlite_test.go
+++ b/internal/store/sqlite_test.go
@@ -173,12 +173,19 @@ func setupTestStoreWithNativeTasks(t *testing.T) *Store {
content TEXT NOT NULL,
description TEXT DEFAULT '',
project_name TEXT DEFAULT '',
+ project_id TEXT DEFAULT '',
due_date DATETIME,
priority INTEGER DEFAULT 1,
completed BOOLEAN DEFAULT 0,
labels TEXT DEFAULT '[]',
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
- updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+ updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+ recurrence_freq TEXT DEFAULT '',
+ recurrence_interval INTEGER DEFAULT 1,
+ recurrence_weekdays TEXT DEFAULT '',
+ recurrence_series_id TEXT DEFAULT '',
+ next_occurrence_override TEXT DEFAULT '',
+ estimated_minutes INTEGER DEFAULT 0
);
`
if _, err := db.Exec(schema); err != nil {