1 files changed, 120 insertions, 0 deletions
diff --git a/internal/store/sqlite_test.go b/internal/store/sqlite_test.go
index 5719e24..fc8a3b7 100644
--- a/internal/store/sqlite_test.go
+++ b/internal/store/sqlite_test.go
@@ -569,3 +569,123 @@ func TestGetCardsByDateRange(t *testing.T) {
 		t.Errorf("Expected card1, got %s", results[0].ID)
 	}
 }
+
+// setupTestStoreWithBugs creates a test store with bugs table
+func setupTestStoreWithBugs(t *testing.T) *Store {
+	t.Helper()
+
+	tempDir := t.TempDir()
+	dbPath := filepath.Join(tempDir, "test.db")
+
+	db, err := sql.Open("sqlite3", dbPath)
+	if err != nil {
+		t.Fatalf("Failed to open test database: %v", err)
+	}
+
+	db.SetMaxOpenConns(1)
+
+	store := &Store{db: db}
+
+	schema := `
+		CREATE TABLE IF NOT EXISTS bugs (
+			id INTEGER PRIMARY KEY AUTOINCREMENT,
+			description TEXT NOT NULL,
+			created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+			resolved_at DATETIME DEFAULT NULL
+		);
+	`
+	if _, err := db.Exec(schema); err != nil {
+		t.Fatalf("Failed to create schema: %v", err)
+	}
+
+	return store
+}
+
+func TestBugResolution(t *testing.T) {
+	store := setupTestStoreWithBugs(t)
+	defer store.Close()
+
+	// Save some bugs
+	if err := store.SaveBug("Bug 1"); err != nil {
+		t.Fatalf("Failed to save bug 1: %v", err)
+	}
+	if err := store.SaveBug("Bug 2"); err != nil {
+		t.Fatalf("Failed to save bug 2: %v", err)
+	}
+	if err := store.SaveBug("Bug 3"); err != nil {
+		t.Fatalf("Failed to save bug 3: %v", err)
+	}
+
+	// Verify all 3 bugs are unresolved
+	unresolved, err := store.GetUnresolvedBugs()
+	if err != nil {
+		t.Fatalf("GetUnresolvedBugs failed: %v", err)
+	}
+	if len(unresolved) != 3 {
+		t.Errorf("Expected 3 unresolved bugs, got %d", len(unresolved))
+	}
+
+	// Resolve bug 2
+	if err := store.ResolveBug(2); err != nil {
+		t.Fatalf("ResolveBug failed: %v", err)
+	}
+
+	// Verify only 2 unresolved bugs remain
+	unresolved, err = store.GetUnresolvedBugs()
+	if err != nil {
+		t.Fatalf("GetUnresolvedBugs after resolve failed: %v", err)
+	}
+	if len(unresolved) != 2 {
+		t.Errorf("Expected 2 unresolved bugs, got %d", len(unresolved))
+	}
+
+	// Verify bug 2 is not in the unresolved list
+	for _, bug := range unresolved {
+		if bug.ID == 2 {
+			t.Error("Bug 2 should have been resolved but is still in unresolved list")
+		}
+	}
+
+	// Verify all bugs still exist in GetBugs (including resolved)
+	allBugs, err := store.GetBugs()
+	if err != nil {
+		t.Fatalf("GetBugs failed: %v", err)
+	}
+	if len(allBugs) != 3 {
+		t.Errorf("Expected 3 total bugs, got %d", len(allBugs))
+	}
+
+	// Verify bug 2 has resolved_at set
+	for _, bug := range allBugs {
+		if bug.ID == 2 {
+			if bug.ResolvedAt == nil {
+				t.Error("Bug 2 should have resolved_at set")
+			}
+		}
+	}
+
+	// Unresolve bug 2
+	if err := store.UnresolveBug(2); err != nil {
+		t.Fatalf("UnresolveBug failed: %v", err)
+	}
+
+	// Verify all 3 bugs are unresolved again
+	unresolved, err = store.GetUnresolvedBugs()
+	if err != nil {
+		t.Fatalf("GetUnresolvedBugs after unresolve failed: %v", err)
+	}
+	if len(unresolved) != 3 {
+		t.Errorf("Expected 3 unresolved bugs after unresolve, got %d", len(unresolved))
+	}
+}
+
+func TestResolveBug_NonExistent(t *testing.T) {
+	store := setupTestStoreWithBugs(t)
+	defer store.Close()
+
+	// Resolving a non-existent bug should not error (no rows affected is fine)
+	err := store.ResolveBug(999)
+	if err != nil {
+		t.Errorf("ResolveBug on non-existent bug should not error, got: %v", err)
+	}
+}