90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/bubbles/viewport"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
// 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")
|
|
}
|
|
}
|