✅ Add unit tests for command management and storage functionalities

Author: geekloper

internal/core/core_test.go

@@ -0,0 +1,276 @@
+package core
+
+import (
+	"os"
+	"strings"
+	"testing"
+)
+
+func setupTestManager(t *testing.T) *Manager {
+	tmpDir := t.TempDir()
+	os.Setenv("ZUT_CONFIG_DIR", tmpDir)
+	t.Cleanup(func() {
+		os.Unsetenv("ZUT_CONFIG_DIR")
+	})
+
+	manager, err := NewManager()
+	if err != nil {
+		t.Fatalf("NewManager() failed: %v", err)
+	}
+
+	return manager
+}
+
+func TestAddCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	err := manager.AddCommand("test-alias", "echo test", "Test tag")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Verify command was added
+	entry, err := manager.GetCommand("test-alias")
+	if err != nil {
+		t.Fatalf("GetCommand() failed: %v", err)
+	}
+
+	if entry.Alias != "test-alias" {
+		t.Errorf("Expected alias 'test-alias', got '%s'", entry.Alias)
+	}
+	if entry.Command != "echo test" {
+		t.Errorf("Expected command 'echo test', got '%s'", entry.Command)
+	}
+	if entry.Tag != "Test tag" {
+		t.Errorf("Expected tag 'Test tag', got '%s'", entry.Tag)
+	}
+}
+
+func TestAddDuplicateAlias(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add first command
+	err := manager.AddCommand("test-alias", "echo test", "Test tag")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Try to add duplicate
+	err = manager.AddCommand("test-alias", "echo duplicate", "Duplicate")
+	if err == nil {
+		t.Fatal("Expected error for duplicate alias, got nil")
+	}
+
+	if !strings.Contains(err.Error(), "already exists") {
+		t.Errorf("Expected 'already exists' error, got: %v", err)
+	}
+}
+
+func TestDeleteCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add command
+	err := manager.AddCommand("test-alias", "echo test", "Test tag")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Delete it
+	err = manager.DeleteCommand("test-alias")
+	if err != nil {
+		t.Fatalf("DeleteCommand() failed: %v", err)
+	}
+
+	// Verify it's gone
+	_, err = manager.GetCommand("test-alias")
+	if err == nil {
+		t.Fatal("Expected error for non-existent command, got nil")
+	}
+}
+
+func TestDeleteNonExistentCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	err := manager.DeleteCommand("non-existent")
+	if err == nil {
+		t.Fatal("Expected error for non-existent command, got nil")
+	}
+
+	if !strings.Contains(err.Error(), "not found") {
+		t.Errorf("Expected 'not found' error, got: %v", err)
+	}
+}
+
+func TestEditCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add command
+	err := manager.AddCommand("test-alias", "echo test", "Test tag")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Edit command
+	err = manager.EditCommand("test-alias", "echo updated", "Updated tag")
+	if err != nil {
+		t.Fatalf("EditCommand() failed: %v", err)
+	}
+
+	// Verify changes
+	entry, err := manager.GetCommand("test-alias")
+	if err != nil {
+		t.Fatalf("GetCommand() failed: %v", err)
+	}
+
+	if entry.Command != "echo updated" {
+		t.Errorf("Expected command 'echo updated', got '%s'", entry.Command)
+	}
+	if entry.Tag != "Updated tag" {
+		t.Errorf("Expected tag 'Updated tag', got '%s'", entry.Tag)
+	}
+}
+
+func TestEditCommandPartial(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add command
+	err := manager.AddCommand("test-alias", "echo test", "Test tag")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Edit only command
+	err = manager.EditCommand("test-alias", "echo updated", "")
+	if err != nil {
+		t.Fatalf("EditCommand() failed: %v", err)
+	}
+
+	// Verify only command changed
+	entry, err := manager.GetCommand("test-alias")
+	if err != nil {
+		t.Fatalf("GetCommand() failed: %v", err)
+	}
+
+	if entry.Command != "echo updated" {
+		t.Errorf("Expected command 'echo updated', got '%s'", entry.Command)
+	}
+	if entry.Tag != "Test tag" {
+		t.Errorf("Expected tag 'Test tag' (unchanged), got '%s'", entry.Tag)
+	}
+}
+
+func TestEditNonExistentCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	err := manager.EditCommand("non-existent", "echo test", "Test")
+	if err == nil {
+		t.Fatal("Expected error for non-existent command, got nil")
+	}
+
+	if !strings.Contains(err.Error(), "not found") {
+		t.Errorf("Expected 'not found' error, got: %v", err)
+	}
+}
+
+func TestListCommands(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add multiple commands
+	commands := []struct {
+		alias   string
+		command string
+		tag     string
+	}{
+		{"cmd1", "echo 1", "Tag 1"},
+		{"cmd2", "echo 2", "Tag 2"},
+		{"cmd3", "echo 3", "Tag 3"},
+	}
+
+	for _, cmd := range commands {
+		err := manager.AddCommand(cmd.alias, cmd.command, cmd.tag)
+		if err != nil {
+			t.Fatalf("AddCommand() failed: %v", err)
+		}
+	}
+
+	// List all
+	entries := manager.ListCommands()
+	if len(entries) != len(commands) {
+		t.Errorf("Expected %d entries, got %d", len(commands), len(entries))
+	}
+}
+
+func TestSearchCommands(t *testing.T) {
+	manager := setupTestManager(t)
+
+	// Add test commands
+	testData := []struct {
+		alias   string
+		command string
+		tag     string
+	}{
+		{"git-push", "git push origin main", "Push to main"},
+		{"git-pull", "git pull origin main", "Pull from main"},
+		{"docker-ps", "docker ps -a", "List containers"},
+		{"docker-clean", "docker system prune", "Clean Docker"},
+	}
+
+	for _, data := range testData {
+		err := manager.AddCommand(data.alias, data.command, data.tag)
+		if err != nil {
+			t.Fatalf("AddCommand() failed: %v", err)
+		}
+	}
+
+	// Test searches
+	tests := []struct {
+		search   string
+		expected int
+	}{
+		{"git", 2},
+		{"docker", 2},
+		{"push", 1},
+		{"main", 2},
+		{"container", 1},
+		{"nonexistent", 0},
+		{"GIT", 2}, // Case insensitive
+	}
+
+	for _, tt := range tests {
+		results := manager.SearchCommands(tt.search)
+		if len(results) != tt.expected {
+			t.Errorf("Search '%s': expected %d results, got %d", tt.search, tt.expected, len(results))
+		}
+	}
+}
+
+func TestSearchCaseInsensitive(t *testing.T) {
+	manager := setupTestManager(t)
+
+	err := manager.AddCommand("TEST", "ECHO TEST", "TEST TAG")
+	if err != nil {
+		t.Fatalf("AddCommand() failed: %v", err)
+	}
+
+	// Search with different cases
+	searches := []string{"test", "TEST", "TeSt", "echo", "ECHO", "tag"}
+	for _, search := range searches {
+		results := manager.SearchCommands(search)
+		if len(results) != 1 {
+			t.Errorf("Search '%s': expected 1 result, got %d", search, len(results))
+		}
+	}
+}
+
+func TestGetNonExistentCommand(t *testing.T) {
+	manager := setupTestManager(t)
+
+	_, err := manager.GetCommand("non-existent")
+	if err == nil {
+		t.Fatal("Expected error for non-existent command, got nil")
+	}
+
+	if !strings.Contains(err.Error(), "not found") {
+		t.Errorf("Expected 'not found' error, got: %v", err)
+	}
+}