1 files changed, 311 insertions, 0 deletions
diff --git a/internal/promptstore/store_test.go b/internal/promptstore/store_test.go
new file mode 100644
index 0000000..8dd506a
--- /dev/null
+++ b/internal/promptstore/store_test.go
@@ -0,0 +1,311 @@
+// Summary: Tests for prompt store operations
+package promptstore
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestJSONLStore_Get(t *testing.T) {
+	tests := []struct {
+		name       string
+		promptName string
+		want       *Prompt
+		wantErr    bool
+	}{
+		{
+			name:       "get built-in prompt",
+			promptName: "code_review",
+			want: &Prompt{
+				Name:        "code_review",
+				Title:       "Request Code Review",
+				Description: "Analyzes code quality, style, and suggests improvements",
+			},
+			wantErr: false,
+		},
+		{
+			name:       "prompt not found",
+			promptName: "nonexistent",
+			want:       nil,
+			wantErr:    true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			tmpDir := t.TempDir()
+			store, err := NewJSONLStore(tmpDir)
+			if err != nil {
+				t.Fatalf("NewJSONLStore() error = %v", err)
+			}
+
+			got, err := store.Get(tt.promptName)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Get() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !tt.wantErr {
+				if got.Name != tt.want.Name {
+					t.Errorf("Get() name = %v, want %v", got.Name, tt.want.Name)
+				}
+				if got.Title != tt.want.Title {
+					t.Errorf("Get() title = %v, want %v", got.Title, tt.want.Title)
+				}
+			}
+		})
+	}
+}
+
+func TestJSONLStore_List(t *testing.T) {
+	tmpDir := t.TempDir()
+	store, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	// List all prompts
+	prompts, cursor, err := store.List("", 100)
+	if err != nil {
+		t.Fatalf("List() error = %v", err)
+	}
+
+	// Should have built-in prompts
+	if len(prompts) < 5 {
+		t.Errorf("List() got %d prompts, want at least 5", len(prompts))
+	}
+
+	// No cursor for full list
+	if cursor != "" {
+		t.Errorf("List() cursor = %v, want empty", cursor)
+	}
+
+	// Test pagination
+	prompts1, cursor1, err := store.List("", 3)
+	if err != nil {
+		t.Fatalf("List() error = %v", err)
+	}
+	if len(prompts1) != 3 {
+		t.Errorf("List() got %d prompts, want 3", len(prompts1))
+	}
+	if cursor1 == "" {
+		t.Error("List() expected cursor, got empty")
+	}
+
+	// Get next page
+	prompts2, cursor2, err := store.List(cursor1, 3)
+	if err != nil {
+		t.Fatalf("List() error = %v", err)
+	}
+	if len(prompts2) == 0 {
+		t.Error("List() second page empty")
+	}
+	if cursor2 == "" && len(prompts) > 6 {
+		t.Error("List() expected cursor2, got empty")
+	}
+}
+
+func TestJSONLStore_Create(t *testing.T) {
+	tmpDir := t.TempDir()
+	store, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	now := time.Now()
+	prompt := &Prompt{
+		Name:        "test_prompt",
+		Title:       "Test Prompt",
+		Description: "A test prompt",
+		Arguments: []PromptArgument{
+			{Name: "input", Description: "Test input", Required: true},
+		},
+		Messages: []PromptMessage{
+			{
+				Role: "user",
+				Content: MessageContent{
+					Type: "text",
+					Text: "Test: {{input}}",
+				},
+			},
+		},
+		Tags:    []string{"test"},
+		Created: now,
+		Updated: now,
+	}
+
+	// Create prompt
+	if err := store.Create(prompt); err != nil {
+		t.Fatalf("Create() error = %v", err)
+	}
+
+	// Verify it exists
+	got, err := store.Get("test_prompt")
+	if err != nil {
+		t.Fatalf("Get() after Create() error = %v", err)
+	}
+	if got.Name != prompt.Name {
+		t.Errorf("Get() name = %v, want %v", got.Name, prompt.Name)
+	}
+
+	// Try to create duplicate
+	err = store.Create(prompt)
+	if err == nil {
+		t.Error("Create() duplicate should fail")
+	}
+}
+
+func TestJSONLStore_Update(t *testing.T) {
+	tmpDir := t.TempDir()
+	store, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	now := time.Now()
+	prompt := &Prompt{
+		Name:        "test_update",
+		Title:       "Original Title",
+		Description: "Original description",
+		Messages:    []PromptMessage{},
+		Created:     now,
+		Updated:     now,
+	}
+
+	// Create initial prompt
+	if err := store.Create(prompt); err != nil {
+		t.Fatalf("Create() error = %v", err)
+	}
+
+	// Update prompt
+	prompt.Title = "Updated Title"
+	prompt.Description = "Updated description"
+	if err := store.Update(prompt); err != nil {
+		t.Fatalf("Update() error = %v", err)
+	}
+
+	// Verify update
+	got, err := store.Get("test_update")
+	if err != nil {
+		t.Fatalf("Get() after Update() error = %v", err)
+	}
+	if got.Title != "Updated Title" {
+		t.Errorf("Get() title = %v, want Updated Title", got.Title)
+	}
+}
+
+func TestJSONLStore_Delete(t *testing.T) {
+	tmpDir := t.TempDir()
+	store, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	now := time.Now()
+	prompt := &Prompt{
+		Name:        "test_delete",
+		Title:       "Delete Me",
+		Description: "Will be deleted",
+		Messages:    []PromptMessage{},
+		Created:     now,
+		Updated:     now,
+	}
+
+	// Create prompt
+	if err := store.Create(prompt); err != nil {
+		t.Fatalf("Create() error = %v", err)
+	}
+
+	// Delete prompt
+	if err := store.Delete("test_delete"); err != nil {
+		t.Fatalf("Delete() error = %v", err)
+	}
+
+	// Verify deletion
+	_, err = store.Get("test_delete")
+	if err == nil {
+		t.Error("Get() after Delete() should fail")
+	}
+}
+
+func TestJSONLStore_SearchByTags(t *testing.T) {
+	tmpDir := t.TempDir()
+	store, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	// Search for development tag
+	results, err := store.SearchByTags([]string{"development"})
+	if err != nil {
+		t.Fatalf("SearchByTags() error = %v", err)
+	}
+
+	if len(results) < 3 {
+		t.Errorf("SearchByTags() got %d results, want at least 3", len(results))
+	}
+
+	// Search for multiple tags
+	results, err = store.SearchByTags([]string{"development", "review"})
+	if err != nil {
+		t.Fatalf("SearchByTags() error = %v", err)
+	}
+
+	// Should find code_review prompt
+	found := false
+	for _, p := range results {
+		if p.Name == "code_review" {
+			found = true
+			break
+		}
+	}
+	if !found {
+		t.Error("SearchByTags() should find code_review prompt")
+	}
+}
+
+func TestBuiltinPrompts(t *testing.T) {
+	prompts := GetBuiltinPrompts()
+
+	if len(prompts) < 5 {
+		t.Errorf("GetBuiltinPrompts() got %d prompts, want at least 5", len(prompts))
+	}
+
+	// Verify each prompt has required fields
+	for _, p := range prompts {
+		if p.Name == "" {
+			t.Error("Prompt missing name")
+		}
+		if p.Title == "" {
+			t.Errorf("Prompt %s missing title", p.Name)
+		}
+		if len(p.Messages) == 0 {
+			t.Errorf("Prompt %s has no messages", p.Name)
+		}
+	}
+}
+
+func TestJSONLStore_DefaultFileCreation(t *testing.T) {
+	tmpDir := t.TempDir()
+	_, err := NewJSONLStore(tmpDir)
+	if err != nil {
+		t.Fatalf("NewJSONLStore() error = %v", err)
+	}
+
+	// Verify default.jsonl was created
+	defaultPath := filepath.Join(tmpDir, "default.jsonl")
+	if _, err := os.Stat(defaultPath); os.IsNotExist(err) {
+		t.Error("default.jsonl was not created")
+	}
+
+	// Verify it contains prompts
+	data, err := os.ReadFile(defaultPath)
+	if err != nil {
+		t.Fatalf("ReadFile() error = %v", err)
+	}
+
+	if len(data) == 0 {
+		t.Error("default.jsonl is empty")
+	}
+}