kami
218a98034e
Render stage clarification questions as an AskUserQuestion-style form: header chip, prompt, selectable option chips, and a free-text custom slot. Arrow/hjkl nav, space to select, e for custom, enter to submit. Submit guards on all-answered and sends ClarificationResponse; the parked stage re-runs with the answers in context.
106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package app
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/correx/tui-go/internal/protocol"
|
|
"github.com/correx/tui-go/internal/ws"
|
|
)
|
|
|
|
func clarModel() Model {
|
|
m := NewModel(ws.New("", 0)) // unconnected; Send just buffers
|
|
m.width, m.height = 120, 40
|
|
m.theme = NewTheme(SoftBlue)
|
|
m.selectedID = "s1"
|
|
m.sessionEntered = true
|
|
m.applyServer(protocol.ServerMessage{
|
|
Type: protocol.TypeClarification,
|
|
SessionID: "s1",
|
|
RequestID: "req-1",
|
|
StageID: "analyst",
|
|
Questions: []protocol.ClarificationQuestionDto{
|
|
{ID: "stack", Prompt: "Which stack?", Options: []string{"React", "Vue"}, Header: "Stack"},
|
|
},
|
|
})
|
|
return m
|
|
}
|
|
|
|
func TestClarificationEntersStateAndRenders(t *testing.T) {
|
|
m := clarModel()
|
|
if m.displayState() != StateClarification {
|
|
t.Fatalf("want StateClarification, got %v", m.displayState())
|
|
}
|
|
out := m.clarificationModal()
|
|
for _, want := range []string{"clarifying questions", "Which stack?", "React", "Vue", "Stack", "custom"} {
|
|
if !strings.Contains(out, want) {
|
|
t.Fatalf("modal missing %q:\n%s", want, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestClarificationSelectAndSubmit(t *testing.T) {
|
|
m := clarModel()
|
|
// move the option cursor to "Vue" (index 1) and select it
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRight})
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeySpace})
|
|
if !m.clarChosenAt(0, 1) {
|
|
t.Fatalf("expected option 1 chosen, chosen=%v", m.clarChosen)
|
|
}
|
|
// submit
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
|
|
if s := m.session("s1"); s == nil || s.Clar != nil {
|
|
t.Fatalf("expected clarification cleared after submit")
|
|
}
|
|
if m.displayState() != StateInSession {
|
|
t.Fatalf("want StateInSession after submit, got %v", m.displayState())
|
|
}
|
|
}
|
|
|
|
func TestClarificationSubmitGuardWaitsForAllAnswers(t *testing.T) {
|
|
m := clarModel()
|
|
m.applyServer(protocol.ServerMessage{
|
|
Type: protocol.TypeClarification, SessionID: "s1", RequestID: "req-2", StageID: "analyst",
|
|
Questions: []protocol.ClarificationQuestionDto{
|
|
{ID: "a", Prompt: "Q A?", Options: []string{"x"}},
|
|
{ID: "b", Prompt: "Q B?", Options: []string{"y"}},
|
|
},
|
|
})
|
|
// answer only the first question, then attempt submit
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeySpace})
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
|
|
if s := m.session("s1"); s == nil || s.Clar == nil {
|
|
t.Fatalf("expected clarification to remain with partial answers")
|
|
}
|
|
if m.clarFocus != 1 {
|
|
t.Fatalf("expected focus nudged to first unanswered (1), got %d", m.clarFocus)
|
|
}
|
|
}
|
|
|
|
func TestClarificationCustomTextAnswer(t *testing.T) {
|
|
m := clarModel()
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("e")})
|
|
if !m.clarTyping {
|
|
t.Fatalf("expected typing mode after 'e'")
|
|
}
|
|
for _, r := range "Solid" {
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
|
|
}
|
|
m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
|
|
if m.clarTyping {
|
|
t.Fatalf("expected typing committed")
|
|
}
|
|
if m.clarTextAt(0) != "Solid" {
|
|
t.Fatalf("want custom text 'Solid', got %q", m.clarTextAt(0))
|
|
}
|
|
if !m.clarAnswered(0, m.session("s1").Clar.Questions[0]) {
|
|
t.Fatalf("expected question answered via custom text")
|
|
}
|
|
}
|
|
|
|
func applyClarKey(m Model, k tea.KeyMsg) Model {
|
|
updated, _ := m.handleClarificationKey(k)
|
|
return updated.(Model)
|
|
}
|