136 lines
3.9 KiB
Go
136 lines
3.9 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/bubbles/viewport"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"sidekick/pkg/sidekick"
|
|
)
|
|
|
|
// Since sidekick-tui/main.go uses global variables and is a tea.Model,
|
|
// we test basic model behavior.
|
|
|
|
func TestInitialModel(t *testing.T) {
|
|
// For testing, sidekick.LoadConfig would fail or look in its default dir.
|
|
// This ensures that the model can be initialized even without a config.
|
|
m := initialModel()
|
|
|
|
if m.agent == nil {
|
|
t.Error("expected default agent to be initialized even without config")
|
|
}
|
|
|
|
if m.textarea.Value() != "" {
|
|
t.Errorf("expected empty textarea, got %s", m.textarea.Value())
|
|
}
|
|
|
|
if len(m.history) != 1 {
|
|
t.Errorf("expected 1 initial message in history, got %d", len(m.history))
|
|
}
|
|
if !strings.Contains(m.history[0].Content, "Sidekick initialized") {
|
|
t.Error("expected initial message content to contain 'Sidekick initialized'")
|
|
}
|
|
}
|
|
|
|
func TestHistoryStateInitialization(t *testing.T) {
|
|
m := initialModel()
|
|
if m.historyIndex != -1 {
|
|
t.Errorf("expected historyIndex to be -1, got %d", m.historyIndex)
|
|
}
|
|
}
|
|
|
|
func TestModelUpdate(t *testing.T) {
|
|
m := initialModel()
|
|
m.ready = true
|
|
m.width = 80
|
|
m.height = 24
|
|
|
|
// Test a key message (Enter)
|
|
m.textarea.SetValue("Hello")
|
|
newModel, cmd := m.Update(tea.KeyMsg{Type: tea.KeyEnter})
|
|
|
|
tm := newModel.(model)
|
|
if tm.isThinking != true {
|
|
t.Error("expected model to be in thinking state after Enter")
|
|
}
|
|
if tm.textarea.Value() != "" {
|
|
t.Error("expected textarea to be reset after Enter")
|
|
}
|
|
if cmd == nil {
|
|
t.Error("expected a command after Enter")
|
|
}
|
|
|
|
// Test an agent response message
|
|
respModel, _ := tm.Update(agentResponseMsg{response: "Hi"})
|
|
rm := respModel.(model)
|
|
if rm.isThinking != false {
|
|
t.Error("expected model to stop thinking after response")
|
|
}
|
|
if len(rm.history) < 2 {
|
|
t.Error("expected history list to grow after response")
|
|
}
|
|
}
|
|
|
|
func TestMessageWrapping(t *testing.T) {
|
|
m := initialModel()
|
|
m.ready = true
|
|
m.width = 10
|
|
m.height = 20
|
|
m.viewport = viewport.New(10, 5) // Very narrow
|
|
|
|
longMsg := "This is a very long message that should be wrapped."
|
|
respModel, _ := m.Update(agentResponseMsg{response: longMsg})
|
|
rm := respModel.(model)
|
|
|
|
content := rm.viewport.View()
|
|
if !strings.Contains(content, "\n") {
|
|
t.Errorf("expected long message to be wrapped, but no newline found in viewport")
|
|
}
|
|
}
|
|
|
|
func TestHistoryNavigation(t *testing.T) {
|
|
m := initialModel()
|
|
m.ready = true
|
|
m.history = append(m.history, sidekick.Message{Role: sidekick.MessageRoleUser, Content: "First"})
|
|
m.history = append(m.history, sidekick.Message{Role: sidekick.MessageRoleAssistant, Content: "AI Response"})
|
|
m.history = append(m.history, sidekick.Message{Role: sidekick.MessageRoleUser, Content: "Second"})
|
|
|
|
// Test UP from empty draft
|
|
m2, _ := m.Update(tea.KeyMsg{Type: tea.KeyUp})
|
|
tm2 := m2.(model)
|
|
if tm2.textarea.Value() != "Second" {
|
|
t.Errorf("expected 'Second', got '%s'", tm2.textarea.Value())
|
|
}
|
|
if tm2.historyIndex != 1 {
|
|
t.Errorf("expected historyIndex 1, got %d", tm2.historyIndex)
|
|
}
|
|
|
|
// Test UP again
|
|
m3, _ := tm2.Update(tea.KeyMsg{Type: tea.KeyUp})
|
|
tm3 := m3.(model)
|
|
if tm3.textarea.Value() != "First" {
|
|
t.Errorf("expected 'First', got '%s'", tm3.textarea.Value())
|
|
}
|
|
if tm3.historyIndex != 0 {
|
|
t.Errorf("expected historyIndex 0, got %d", tm3.historyIndex)
|
|
}
|
|
|
|
// Test DOWN
|
|
m4, _ := tm3.Update(tea.KeyMsg{Type: tea.KeyDown})
|
|
tm4 := m4.(model)
|
|
if tm4.textarea.Value() != "Second" {
|
|
t.Errorf("expected 'Second', got '%s'", tm4.textarea.Value())
|
|
}
|
|
|
|
// Test DOWN to draft
|
|
m5, _ := tm4.Update(tea.KeyMsg{Type: tea.KeyDown})
|
|
tm5 := m5.(model)
|
|
if tm5.textarea.Value() != "" {
|
|
t.Errorf("expected empty draft, got '%s'", tm5.textarea.Value())
|
|
}
|
|
if tm5.historyIndex != -1 {
|
|
t.Error("expected historyIndex to reset to -1")
|
|
}
|
|
}
|