Files
beads/cmd/bd/delete_recording_test.go
2025-11-25 11:46:06 -08:00

224 lines
6.0 KiB
Go

package main
import (
"context"
"os"
"path/filepath"
"testing"
"time"
"github.com/steveyegge/beads/internal/deletions"
"github.com/steveyegge/beads/internal/storage/sqlite"
"github.com/steveyegge/beads/internal/types"
)
// TestRecordDeletion tests that recordDeletion creates deletion manifest entries
func TestRecordDeletion(t *testing.T) {
tmpDir := t.TempDir()
// Set up dbPath so getDeletionsPath() works
oldDbPath := dbPath
dbPath = filepath.Join(tmpDir, "beads.db")
defer func() { dbPath = oldDbPath }()
// Create the .beads directory
if err := os.MkdirAll(tmpDir, 0750); err != nil {
t.Fatalf("failed to create directory: %v", err)
}
// Test recordDeletion
err := recordDeletion("test-abc", "test-user", "test reason")
if err != nil {
t.Fatalf("recordDeletion failed: %v", err)
}
// Verify the deletion was recorded
deletionsPath := getDeletionsPath()
result, err := deletions.LoadDeletions(deletionsPath)
if err != nil {
t.Fatalf("LoadDeletions failed: %v", err)
}
if len(result.Records) != 1 {
t.Fatalf("expected 1 deletion record, got %d", len(result.Records))
}
del, found := result.Records["test-abc"]
if !found {
t.Fatalf("deletion record for 'test-abc' not found")
}
if del.Actor != "test-user" {
t.Errorf("expected actor 'test-user', got '%s'", del.Actor)
}
if del.Reason != "test reason" {
t.Errorf("expected reason 'test reason', got '%s'", del.Reason)
}
// Timestamp should be recent (within last minute)
if time.Since(del.Timestamp) > time.Minute {
t.Errorf("timestamp seems too old: %v", del.Timestamp)
}
}
// TestRecordDeletions tests that recordDeletions creates multiple deletion manifest entries
func TestRecordDeletions(t *testing.T) {
tmpDir := t.TempDir()
// Set up dbPath so getDeletionsPath() works
oldDbPath := dbPath
dbPath = filepath.Join(tmpDir, "beads.db")
defer func() { dbPath = oldDbPath }()
// Create the .beads directory
if err := os.MkdirAll(tmpDir, 0750); err != nil {
t.Fatalf("failed to create directory: %v", err)
}
// Test recordDeletions with multiple IDs
ids := []string{"test-abc", "test-def", "test-ghi"}
err := recordDeletions(ids, "batch-user", "batch cleanup")
if err != nil {
t.Fatalf("recordDeletions failed: %v", err)
}
// Verify the deletions were recorded
deletionsPath := getDeletionsPath()
result, err := deletions.LoadDeletions(deletionsPath)
if err != nil {
t.Fatalf("LoadDeletions failed: %v", err)
}
if len(result.Records) != 3 {
t.Fatalf("expected 3 deletion records, got %d", len(result.Records))
}
for _, id := range ids {
del, found := result.Records[id]
if !found {
t.Errorf("deletion record for '%s' not found", id)
continue
}
if del.Actor != "batch-user" {
t.Errorf("expected actor 'batch-user' for %s, got '%s'", id, del.Actor)
}
if del.Reason != "batch cleanup" {
t.Errorf("expected reason 'batch cleanup' for %s, got '%s'", id, del.Reason)
}
}
}
// TestGetActorWithGit tests actor sourcing logic
func TestGetActorWithGit(t *testing.T) {
// Save original actor value
oldActor := actor
defer func() { actor = oldActor }()
// Test case 1: actor is set from flag/env
actor = "flag-user"
result := getActorWithGit()
if result != "flag-user" {
t.Errorf("expected 'flag-user' when actor is set, got '%s'", result)
}
// Test case 2: actor is "unknown" - should try git config
actor = "unknown"
result = getActorWithGit()
// Can't test exact result since it depends on git config, but it shouldn't be empty
if result == "" {
t.Errorf("expected non-empty result when actor is 'unknown'")
}
// Test case 3: actor is empty - should try git config
actor = ""
result = getActorWithGit()
if result == "" {
t.Errorf("expected non-empty result when actor is empty")
}
}
// TestDeleteRecordingOrderOfOperations verifies deletion is recorded before DB delete
func TestDeleteRecordingOrderOfOperations(t *testing.T) {
ctx := context.Background()
tmpDir := t.TempDir()
// Set up dbPath
oldDbPath := dbPath
dbPath = filepath.Join(tmpDir, "beads.db")
defer func() { dbPath = oldDbPath }()
// Create database
testStore, err := sqlite.New(ctx, dbPath)
if err != nil {
t.Fatalf("failed to create database: %v", err)
}
defer testStore.Close()
// Initialize prefix
if err := testStore.SetConfig(ctx, "issue_prefix", "test"); err != nil {
t.Fatalf("failed to set prefix: %v", err)
}
// Create an issue
issue := &types.Issue{
ID: "test-delete-order",
Title: "Test Order of Operations",
Status: types.StatusOpen,
Priority: 1,
IssueType: types.TypeTask,
}
if err := testStore.CreateIssue(ctx, issue, "test"); err != nil {
t.Fatalf("failed to create issue: %v", err)
}
// Record deletion (simulating what delete command does)
if err := recordDeletion(issue.ID, "test-user", "order test"); err != nil {
t.Fatalf("recordDeletion failed: %v", err)
}
// Verify record was created BEFORE any DB changes
deletionsPath := getDeletionsPath()
result, err := deletions.LoadDeletions(deletionsPath)
if err != nil {
t.Fatalf("LoadDeletions failed: %v", err)
}
if _, found := result.Records[issue.ID]; !found {
t.Error("deletion record should exist before DB deletion")
}
// Now verify the issue still exists in DB (we only recorded, didn't delete)
existing, err := testStore.GetIssue(ctx, issue.ID)
if err != nil {
t.Fatalf("GetIssue failed: %v", err)
}
if existing == nil {
t.Error("issue should still exist in DB (we only recorded the deletion)")
}
// Now delete from DB
if err := testStore.DeleteIssue(ctx, issue.ID); err != nil {
t.Fatalf("DeleteIssue failed: %v", err)
}
// Verify both: deletion record exists AND issue is gone from DB
result, err = deletions.LoadDeletions(deletionsPath)
if err != nil {
t.Fatalf("LoadDeletions failed: %v", err)
}
if _, found := result.Records[issue.ID]; !found {
t.Error("deletion record should still exist after DB deletion")
}
existing, err = testStore.GetIssue(ctx, issue.ID)
if err != nil {
t.Fatalf("GetIssue failed: %v", err)
}
if existing != nil {
t.Error("issue should be gone from DB after deletion")
}
}