bd sync: 2025-11-25 11:46:06

internal/importer/purge_test.go

@@ -0,0 +1,233 @@
+package importer
+
+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"
+)
+
+// TestPurgeDeletedIssues tests that issues in the deletions manifest are purged during import
+func TestPurgeDeletedIssues(t *testing.T) {
+	ctx := context.Background()
+	tmpDir := t.TempDir()
+
+	// Create database
+	dbPath := filepath.Join(tmpDir, "beads.db")
+	store, err := sqlite.New(ctx, dbPath)
+	if err != nil {
+		t.Fatalf("failed to create database: %v", err)
+	}
+	defer store.Close()
+
+	// Initialize prefix
+	if err := store.SetConfig(ctx, "issue_prefix", "test"); err != nil {
+		t.Fatalf("failed to set prefix: %v", err)
+	}
+
+	// Create some issues in the database
+	issue1 := &types.Issue{
+		ID:        "test-abc",
+		Title:     "Issue 1",
+		Status:    types.StatusOpen,
+		Priority:  1,
+		IssueType: types.TypeTask,
+	}
+	issue2 := &types.Issue{
+		ID:        "test-def",
+		Title:     "Issue 2",
+		Status:    types.StatusOpen,
+		Priority:  1,
+		IssueType: types.TypeTask,
+	}
+	issue3 := &types.Issue{
+		ID:        "test-ghi",
+		Title:     "Issue 3 (local work)",
+		Status:    types.StatusOpen,
+		Priority:  1,
+		IssueType: types.TypeTask,
+	}
+
+	for _, iss := range []*types.Issue{issue1, issue2, issue3} {
+		if err := store.CreateIssue(ctx, iss, "test"); err != nil {
+			t.Fatalf("failed to create issue %s: %v", iss.ID, err)
+		}
+	}
+
+	// Create a deletions manifest with issue2 deleted
+	deletionsPath := deletions.DefaultPath(tmpDir)
+	delRecord := deletions.DeletionRecord{
+		ID:        "test-def",
+		Timestamp: time.Now().UTC(),
+		Actor:     "test-user",
+		Reason:    "test deletion",
+	}
+	if err := deletions.AppendDeletion(deletionsPath, delRecord); err != nil {
+		t.Fatalf("failed to create deletions manifest: %v", err)
+	}
+
+	// Simulate import with only issue1 in the JSONL (issue2 was deleted, issue3 is local work)
+	jsonlIssues := []*types.Issue{issue1}
+
+	result := &Result{
+		IDMapping:        make(map[string]string),
+		MismatchPrefixes: make(map[string]int),
+	}
+
+	// Call purgeDeletedIssues
+	if err := purgeDeletedIssues(ctx, store, dbPath, jsonlIssues, result); err != nil {
+		t.Fatalf("purgeDeletedIssues failed: %v", err)
+	}
+
+	// Verify issue2 was purged
+	if result.Purged != 1 {
+		t.Errorf("expected 1 purged issue, got %d", result.Purged)
+	}
+	if len(result.PurgedIDs) != 1 || result.PurgedIDs[0] != "test-def" {
+		t.Errorf("expected PurgedIDs to contain 'test-def', got %v", result.PurgedIDs)
+	}
+
+	// Verify issue2 is gone from database
+	iss2, err := store.GetIssue(ctx, "test-def")
+	if err != nil {
+		t.Fatalf("GetIssue failed: %v", err)
+	}
+	if iss2 != nil {
+		t.Errorf("expected issue2 to be deleted, but it still exists")
+	}
+
+	// Verify issue1 still exists (in JSONL)
+	iss1, err := store.GetIssue(ctx, "test-abc")
+	if err != nil {
+		t.Fatalf("GetIssue failed: %v", err)
+	}
+	if iss1 == nil {
+		t.Errorf("expected issue1 to still exist")
+	}
+
+	// Verify issue3 still exists (local work, not in deletions manifest)
+	iss3, err := store.GetIssue(ctx, "test-ghi")
+	if err != nil {
+		t.Fatalf("GetIssue failed: %v", err)
+	}
+	if iss3 == nil {
+		t.Errorf("expected issue3 (local work) to still exist")
+	}
+}
+
+// TestPurgeDeletedIssues_NoDeletionsManifest tests that import works without a deletions manifest
+func TestPurgeDeletedIssues_NoDeletionsManifest(t *testing.T) {
+	ctx := context.Background()
+	tmpDir := t.TempDir()
+
+	// Create database
+	dbPath := filepath.Join(tmpDir, "beads.db")
+	store, err := sqlite.New(ctx, dbPath)
+	if err != nil {
+		t.Fatalf("failed to create database: %v", err)
+	}
+	defer store.Close()
+
+	// Initialize prefix
+	if err := store.SetConfig(ctx, "issue_prefix", "test"); err != nil {
+		t.Fatalf("failed to set prefix: %v", err)
+	}
+
+	// Create an issue in the database
+	issue := &types.Issue{
+		ID:        "test-abc",
+		Title:     "Issue 1",
+		Status:    types.StatusOpen,
+		Priority:  1,
+		IssueType: types.TypeTask,
+	}
+	if err := store.CreateIssue(ctx, issue, "test"); err != nil {
+		t.Fatalf("failed to create issue: %v", err)
+	}
+
+	// No deletions manifest exists
+	jsonlIssues := []*types.Issue{issue}
+
+	result := &Result{
+		IDMapping:        make(map[string]string),
+		MismatchPrefixes: make(map[string]int),
+	}
+
+	// Call purgeDeletedIssues - should succeed with no errors
+	if err := purgeDeletedIssues(ctx, store, dbPath, jsonlIssues, result); err != nil {
+		t.Fatalf("purgeDeletedIssues failed: %v", err)
+	}
+
+	// Verify nothing was purged
+	if result.Purged != 0 {
+		t.Errorf("expected 0 purged issues, got %d", result.Purged)
+	}
+
+	// Verify issue still exists
+	iss, err := store.GetIssue(ctx, "test-abc")
+	if err != nil {
+		t.Fatalf("GetIssue failed: %v", err)
+	}
+	if iss == nil {
+		t.Errorf("expected issue to still exist")
+	}
+}
+
+// TestPurgeDeletedIssues_EmptyDeletionsManifest tests that import works with empty deletions manifest
+func TestPurgeDeletedIssues_EmptyDeletionsManifest(t *testing.T) {
+	ctx := context.Background()
+	tmpDir := t.TempDir()
+
+	// Create database
+	dbPath := filepath.Join(tmpDir, "beads.db")
+	store, err := sqlite.New(ctx, dbPath)
+	if err != nil {
+		t.Fatalf("failed to create database: %v", err)
+	}
+	defer store.Close()
+
+	// Initialize prefix
+	if err := store.SetConfig(ctx, "issue_prefix", "test"); err != nil {
+		t.Fatalf("failed to set prefix: %v", err)
+	}
+
+	// Create an issue in the database
+	issue := &types.Issue{
+		ID:        "test-abc",
+		Title:     "Issue 1",
+		Status:    types.StatusOpen,
+		Priority:  1,
+		IssueType: types.TypeTask,
+	}
+	if err := store.CreateIssue(ctx, issue, "test"); err != nil {
+		t.Fatalf("failed to create issue: %v", err)
+	}
+
+	// Create empty deletions manifest
+	deletionsPath := deletions.DefaultPath(tmpDir)
+	if err := os.WriteFile(deletionsPath, []byte{}, 0644); err != nil {
+		t.Fatalf("failed to create empty deletions manifest: %v", err)
+	}
+
+	jsonlIssues := []*types.Issue{issue}
+
+	result := &Result{
+		IDMapping:        make(map[string]string),
+		MismatchPrefixes: make(map[string]int),
+	}
+
+	// Call purgeDeletedIssues - should succeed with no errors
+	if err := purgeDeletedIssues(ctx, store, dbPath, jsonlIssues, result); err != nil {
+		t.Fatalf("purgeDeletedIssues failed: %v", err)
+	}
+
+	// Verify nothing was purged
+	if result.Purged != 0 {
+		t.Errorf("expected 0 purged issues, got %d", result.Purged)
+	}
+}