Files
beads/cmd/bd/comments_test.go
Steve Yegge fc2a9e183e Implement bd comment alias for bd comments add (bd-d3f0)
- Add commentCmd as top-level alias that delegates to commentsAddCmd
- Add flags for --file and --author to the alias
- Add comprehensive tests in comments_test.go
- Maintain backward compatibility with bd comments add
- Close bd-d3f0
2025-11-03 22:23:36 -08:00

204 lines
5.0 KiB
Go

package main
import (
"context"
"fmt"
"os"
"path/filepath"
"testing"
"github.com/steveyegge/beads/internal/types"
)
const testUserAlice = "alice"
func TestCommentsCommand(t *testing.T) {
tmpDir, err := os.MkdirTemp("", "bd-test-comments-*")
if err != nil {
t.Fatalf("Failed to create temp dir: %v", err)
}
defer os.RemoveAll(tmpDir)
testDB := filepath.Join(tmpDir, "test.db")
s := newTestStore(t, testDB)
defer s.Close()
ctx := context.Background()
// Create test issue
issue := &types.Issue{
Title: "Test Issue",
Description: "Test description",
Priority: 1,
IssueType: types.TypeBug,
Status: types.StatusOpen,
}
if err := s.CreateIssue(ctx, issue, "test-user"); err != nil {
t.Fatalf("Failed to create issue: %v", err)
}
t.Run("add comment", func(t *testing.T) {
comment, err := s.AddIssueComment(ctx, issue.ID, testUserAlice, "This is a test comment")
if err != nil {
t.Fatalf("Failed to add comment: %v", err)
}
if comment.IssueID != issue.ID {
t.Errorf("Expected issue ID %s, got %s", issue.ID, comment.IssueID)
}
if comment.Author != testUserAlice {
t.Errorf("Expected author alice, got %s", comment.Author)
}
if comment.Text != "This is a test comment" {
t.Errorf("Expected text 'This is a test comment', got %s", comment.Text)
}
})
t.Run("list comments", func(t *testing.T) {
comments, err := s.GetIssueComments(ctx, issue.ID)
if err != nil {
t.Fatalf("Failed to get comments: %v", err)
}
if len(comments) != 1 {
t.Errorf("Expected 1 comment, got %d", len(comments))
}
if comments[0].Text != "This is a test comment" {
t.Errorf("Expected comment text, got %s", comments[0].Text)
}
})
t.Run("multiple comments", func(t *testing.T) {
_, err := s.AddIssueComment(ctx, issue.ID, "bob", "Second comment")
if err != nil {
t.Fatalf("Failed to add second comment: %v", err)
}
comments, err := s.GetIssueComments(ctx, issue.ID)
if err != nil {
t.Fatalf("Failed to get comments: %v", err)
}
if len(comments) != 2 {
t.Errorf("Expected 2 comments, got %d", len(comments))
}
})
t.Run("comments on non-existent issue", func(t *testing.T) {
comments, err := s.GetIssueComments(ctx, "bd-nonexistent")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if len(comments) != 0 {
t.Errorf("Expected 0 comments for non-existent issue, got %d", len(comments))
}
})
}
func TestCommentAlias(t *testing.T) {
tmpDir, err := os.MkdirTemp("", "bd-test-comment-alias-*")
if err != nil {
t.Fatalf("Failed to create temp dir: %v", err)
}
defer os.RemoveAll(tmpDir)
testDB := filepath.Join(tmpDir, "test.db")
s := newTestStore(t, testDB)
defer s.Close()
ctx := context.Background()
// Create test issue
issue := &types.Issue{
Title: "Test Issue",
Description: "Test description",
Priority: 1,
IssueType: types.TypeBug,
Status: types.StatusOpen,
}
if err := s.CreateIssue(ctx, issue, "test-user"); err != nil {
t.Fatalf("Failed to create issue: %v", err)
}
t.Run("comment alias shares Run function with comments add", func(t *testing.T) {
// This verifies that commentCmd reuses commentsAddCmd.Run
if commentCmd.Run == nil {
t.Error("commentCmd.Run is nil")
}
if commentsAddCmd.Run == nil {
t.Error("commentsAddCmd.Run is nil")
}
// Verify they share the same Run function (same memory address)
// This is a compile-time guarantee from how we defined it
// Just verify the command structure is set up correctly
if commentCmd.Use != "comment [issue-id] [text]" {
t.Errorf("Expected Use to be 'comment [issue-id] [text]', got %s", commentCmd.Use)
}
if commentCmd.Short != "Add a comment to an issue (alias for 'comments add')" {
t.Errorf("Unexpected Short description: %s", commentCmd.Short)
}
})
t.Run("comment added via storage API works", func(t *testing.T) {
// Test direct storage API (which is what the command uses under the hood)
comment, err := s.AddIssueComment(ctx, issue.ID, testUserAlice, "Test comment")
if err != nil {
t.Fatalf("Failed to add comment: %v", err)
}
if comment.Text != "Test comment" {
t.Errorf("Expected 'Test comment', got %s", comment.Text)
}
// Verify via GetIssueComments
comments, err := s.GetIssueComments(ctx, issue.ID)
if err != nil {
t.Fatalf("Failed to get comments: %v", err)
}
if len(comments) != 1 {
t.Fatalf("Expected 1 comment, got %d", len(comments))
}
})
}
func TestIsUnknownOperationError(t *testing.T) {
tests := []struct {
name string
err error
expected bool
}{
{
name: "nil error",
err: nil,
expected: false,
},
{
name: "unknown operation error",
err: fmt.Errorf("unknown operation: test"),
expected: true,
},
{
name: "other error",
err: fmt.Errorf("some other error"),
expected: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := isUnknownOperationError(tt.err)
if result != tt.expected {
t.Errorf("Expected %v, got %v for error: %v", tt.expected, result, tt.err)
}
})
}
}