First Commit

db/task_metadata_test.go

@@ -0,0 +1,104 @@
+package db
+
+import (
+	"fmt"
+	"testing"
+	"time"
+)
+
+func TestTaskPinOrderingAndRename(t *testing.T) {
+	d, err := Open(testDSN(t))
+	if err != nil {
+		t.Skipf("postgres unavailable (%v) — skipping", err)
+	}
+	defer d.Close()
+
+	suffix := time.Now().UnixNano()
+	first, err := d.CreateTask(fmt.Sprintf("task-pin-first-%d", suffix), "goal", nil, 0, 0)
+	if err != nil {
+		t.Fatal(err)
+	}
+	second, err := d.CreateTask(fmt.Sprintf("task-pin-second-%d", suffix), "goal", nil, 0, 0)
+	if err != nil {
+		_ = d.DeleteTask(first.ID)
+		t.Fatal(err)
+	}
+	defer func() {
+		_ = d.DeleteTask(first.ID)
+		_ = d.DeleteTask(second.ID)
+	}()
+
+	pinned := true
+	first, err = d.UpdateTask(first.ID, TaskPatch{Pinned: &pinned})
+	if err != nil || first == nil || !first.Pinned || first.PinnedAt == nil {
+		t.Fatalf("pin first = %+v, %v", first, err)
+	}
+	firstPinnedAt := *first.PinnedAt
+	time.Sleep(5 * time.Millisecond)
+	second, err = d.UpdateTask(second.ID, TaskPatch{Pinned: &pinned})
+	if err != nil || second == nil || !second.Pinned || second.PinnedAt == nil {
+		t.Fatalf("pin second = %+v, %v", second, err)
+	}
+
+	name := "renamed pinned task"
+	first, err = d.UpdateTask(first.ID, TaskPatch{Name: &name, Pinned: &pinned})
+	if err != nil || first == nil {
+		t.Fatal(err)
+	}
+	if first.Name != name || first.PinnedAt == nil || !first.PinnedAt.Equal(firstPinnedAt) {
+		t.Fatalf("repeat pin should preserve pin time: %+v (want %v)", first, firstPinnedAt)
+	}
+
+	tasks, err := d.ListTasks()
+	if err != nil {
+		t.Fatal(err)
+	}
+	positions := map[int64]int{}
+	for index, task := range tasks {
+		positions[task.ID] = index
+	}
+	if positions[second.ID] >= positions[first.ID] {
+		t.Fatalf("newer pin must sort first: second=%d first=%d", positions[second.ID], positions[first.ID])
+	}
+
+	pinned = false
+	first, err = d.UpdateTask(first.ID, TaskPatch{Pinned: &pinned})
+	if err != nil || first == nil || first.Pinned || first.PinnedAt != nil {
+		t.Fatalf("unpin first = %+v, %v", first, err)
+	}
+}
+
+func TestDeleteConversationsReturnsExistingIDs(t *testing.T) {
+	d, err := Open(testDSN(t))
+	if err != nil {
+		t.Skipf("postgres unavailable (%v) — skipping", err)
+	}
+	defer d.Close()
+
+	first, err := d.CreateConversation("mainagent", "batch-delete-first", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	second, err := d.CreateConversation("mainagent", "batch-delete-second", nil)
+	if err != nil {
+		_ = d.DeleteConversation(first.ID)
+		t.Fatal(err)
+	}
+	defer func() {
+		_ = d.DeleteConversation(first.ID)
+		_ = d.DeleteConversation(second.ID)
+	}()
+
+	missing := int64(1<<62 - 1)
+	deleted, err := d.DeleteConversations([]int64{first.ID, missing, second.ID})
+	if err != nil {
+		t.Fatal(err)
+	}
+	seen := make(map[int64]bool, len(deleted))
+	for _, id := range deleted {
+		seen[id] = true
+	}
+	if !seen[first.ID] || !seen[second.ID] || seen[missing] || len(deleted) != 2 {
+		t.Fatalf("deleted=%v, want existing ids only", deleted)
+	}
+}