added tests for spinner package

spinner/spinner_test.go

@@ -0,0 +1,126 @@
+package spinner
+
+import (
+	"context"
+	"reflect"
+	"strings"
+	"testing"
+
+	"github.com/charmbracelet/bubbles/spinner"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
+)
+
+func TestNewSpinner(t *testing.T) {
+	s := New()
+	if s.title != "Loading..." {
+		t.Errorf("Expected default title 'Loading...', got '%s'", s.title)
+	}
+	if !reflect.DeepEqual(s.spinner.Spinner, spinner.Dot) {
+		t.Errorf("Expected default spinner type to be Dot, got %v", s.spinner.Spinner)
+	}
+}
+
+func TestSpinnerType(t *testing.T) {
+	s := New().Type(Dots)
+	if !reflect.DeepEqual(s.spinner.Spinner, spinner.Dot) {
+		t.Errorf("Expected spinner type to be Dot, got %v", s.spinner.Spinner)
+	}
+}
+
+func TestSpinnerDifferentTypes(t *testing.T) {
+	s := New().Type(Line)
+	if !reflect.DeepEqual(s.spinner.Spinner, spinner.Line) {
+		t.Errorf("Expected spinner type to be Line, got %v", s.spinner.Spinner)
+	}
+}
+
+func TestSpinnerView(t *testing.T) {
+	s := New().Title("Test")
+	view := s.View()
+
+	if !strings.Contains(view, "Test") {
+		t.Errorf("Expected view to contain title 'Test', got '%s'", view)
+	}
+}
+func TestSpinnerAction(t *testing.T) {
+	var actionCalled bool
+	testAction := func() { actionCalled = true }
+
+	s := New().Action(testAction)
+	s.Run()
+
+	if !actionCalled {
+		t.Errorf("Action was not called after Run")
+	}
+}
+
+func TestSpinnerContextCancellation(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	s := New().Context(ctx)
+	cancel() // Cancel before running
+
+	err := s.Run()
+	if err != nil {
+		t.Errorf("Run() returned an error after context cancellation: %v", err)
+	}
+}
+
+func TestSpinnerStyleMethods(t *testing.T) {
+	s := New()
+	style := lipgloss.NewStyle().Foreground(lipgloss.Color("red"))
+	titleStyle := lipgloss.NewStyle().Foreground(lipgloss.Color("blue"))
+
+	s.Style(style)
+	s.TitleStyle(titleStyle)
+
+	if !reflect.DeepEqual(s.spinner.Style, style) {
+		t.Errorf("Style was not set correctly")
+	}
+
+	if !reflect.DeepEqual(s.titleStyle, titleStyle) {
+		t.Errorf("TitleStyle was not set correctly")
+	}
+}
+
+func TestSpinnerInit(t *testing.T) {
+	s := New()
+	cmd := s.Init()
+
+	if cmd == nil {
+		t.Errorf("Init did not return a valid command")
+	}
+}
+
+func TestSpinnerUpdate(t *testing.T) {
+	s := New()
+	cmd := s.Init()
+	if cmd == nil {
+		t.Errorf("Init did not return a valid command")
+	}
+
+	model, cmd := s.Update(spinner.TickMsg{})
+	if reflect.TypeOf(model) != reflect.TypeOf(&Spinner{}) {
+		t.Errorf("Update did not return correct model type")
+	}
+
+	if cmd == nil {
+		t.Errorf("Update should return a non-nil command in this scenario")
+	}
+
+	// Simulate key press
+	_, cmd = s.Update(tea.KeyMsg{Type: tea.KeyCtrlC})
+	if cmd == nil {
+		t.Errorf("Update did not handle key press correctly")
+	}
+}
+
+func TestAccessibleSpinner(t *testing.T) {
+	s := New().Accessible(true)
+	err := s.Run()
+	if err != nil {
+		t.Errorf("Run() in accessible mode returned an error: %v", err)
+	}
+}