diff options
Diffstat (limited to 'internal/store')
| -rw-r--r-- | internal/store/sqlite.go | 84 | ||||
| -rw-r--r-- | internal/store/sqlite_test.go | 120 |
2 files changed, 203 insertions, 1 deletions
diff --git a/internal/store/sqlite.go b/internal/store/sqlite.go index 5b67234..5b95607 100644 --- a/internal/store/sqlite.go +++ b/internal/store/sqlite.go @@ -569,6 +569,7 @@ type Bug struct { ID int64 Description string CreatedAt time.Time + ResolvedAt *time.Time } // SaveBug saves a new bug report @@ -579,7 +580,30 @@ func (s *Store) SaveBug(description string) error { // GetBugs retrieves all bugs, newest first func (s *Store) GetBugs() ([]Bug, error) { - rows, err := s.db.Query(`SELECT id, description, created_at FROM bugs ORDER BY created_at DESC`) + rows, err := s.db.Query(`SELECT id, description, created_at, resolved_at FROM bugs ORDER BY created_at DESC`) + if err != nil { + return nil, err + } + defer rows.Close() + + var bugs []Bug + for rows.Next() { + var b Bug + var resolvedAt sql.NullTime + if err := rows.Scan(&b.ID, &b.Description, &b.CreatedAt, &resolvedAt); err != nil { + return nil, err + } + if resolvedAt.Valid { + b.ResolvedAt = &resolvedAt.Time + } + bugs = append(bugs, b) + } + return bugs, rows.Err() +} + +// GetUnresolvedBugs retrieves bugs that haven't been resolved yet +func (s *Store) GetUnresolvedBugs() ([]Bug, error) { + rows, err := s.db.Query(`SELECT id, description, created_at FROM bugs WHERE resolved_at IS NULL ORDER BY created_at DESC`) if err != nil { return nil, err } @@ -596,6 +620,64 @@ func (s *Store) GetBugs() ([]Bug, error) { return bugs, rows.Err() } +// ResolveBug marks a bug as resolved +func (s *Store) ResolveBug(id int64) error { + _, err := s.db.Exec(`UPDATE bugs SET resolved_at = CURRENT_TIMESTAMP WHERE id = ?`, id) + return err +} + +// UnresolveBug marks a bug as unresolved (reopens it) +func (s *Store) UnresolveBug(id int64) error { + _, err := s.db.Exec(`UPDATE bugs SET resolved_at = NULL WHERE id = ?`, id) + return err +} + +// UserShoppingItem represents a user-added shopping item +type UserShoppingItem struct { + ID int64 + Name string + Store string + Checked bool + CreatedAt time.Time +} + +// SaveUserShoppingItem saves a new user shopping item +func (s *Store) SaveUserShoppingItem(name, store string) error { + _, err := s.db.Exec(`INSERT INTO user_shopping_items (name, store) VALUES (?, ?)`, name, store) + return err +} + +// GetUserShoppingItems retrieves all user shopping items +func (s *Store) GetUserShoppingItems() ([]UserShoppingItem, error) { + rows, err := s.db.Query(`SELECT id, name, store, checked, created_at FROM user_shopping_items ORDER BY store, created_at DESC`) + if err != nil { + return nil, err + } + defer rows.Close() + + var items []UserShoppingItem + for rows.Next() { + var item UserShoppingItem + if err := rows.Scan(&item.ID, &item.Name, &item.Store, &item.Checked, &item.CreatedAt); err != nil { + return nil, err + } + items = append(items, item) + } + return items, rows.Err() +} + +// ToggleUserShoppingItem toggles the checked state of a user shopping item +func (s *Store) ToggleUserShoppingItem(id int64, checked bool) error { + _, err := s.db.Exec(`UPDATE user_shopping_items SET checked = ? WHERE id = ?`, checked, id) + return err +} + +// DeleteUserShoppingItem removes a user shopping item +func (s *Store) DeleteUserShoppingItem(id int64) error { + _, err := s.db.Exec(`DELETE FROM user_shopping_items WHERE id = ?`, id) + return err +} + // GetTasksByDateRange retrieves tasks due within a specific date range func (s *Store) GetTasksByDateRange(start, end time.Time) ([]models.Task, error) { rows, err := s.db.Query(` 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) + } +} |