elvish/pkg/cli/tk/combobox_test.go

package tk

import (
	"testing"
	"time"

	"src.elv.sh/pkg/cli/term"
	"src.elv.sh/pkg/ui"
)

var comboBoxRenderTests = []RenderTest{
	{
		Name: "rendering codearea and listbox",
		Given: NewComboBox(ComboBoxSpec{
			CodeArea: CodeAreaSpec{
				State: CodeAreaState{
					Buffer: CodeBuffer{Content: "filter", Dot: 6}}},
			ListBox: ListBoxSpec{
				State: ListBoxState{Items: TestItems{NItems: 2}}}}),
		Width: 10, Height: 24,
		Want: term.NewBufferBuilder(10).
			Write("filter").SetDotHere().
			Newline().Write("item 0    ", ui.Inverse).
			Newline().Write("item 1"),
	},
	{
		Name: "calling filter before rendering",
		Given: NewComboBox(ComboBoxSpec{
			CodeArea: CodeAreaSpec{
				State: CodeAreaState{
					Buffer: CodeBuffer{Content: "filter", Dot: 6}}},
			OnFilter: func(w ComboBox, filter string) {
				w.ListBox().Reset(TestItems{NItems: 2}, 0)
			}}),
		Width: 10, Height: 24,
		Want: term.NewBufferBuilder(10).
			Write("filter").SetDotHere().
			Newline().Write("item 0    ", ui.Inverse).
			Newline().Write("item 1"),
	},
}

func TestComboBox_Render(t *testing.T) {
	TestRender(t, comboBoxRenderTests)
}

func TestComboBox_Handle(t *testing.T) {
	var onFilterCalled bool
	var lastFilter string
	w := NewComboBox(ComboBoxSpec{
		OnFilter: func(w ComboBox, filter string) {
			onFilterCalled = true
			lastFilter = filter
		},
		ListBox: ListBoxSpec{
			State: ListBoxState{Items: TestItems{NItems: 2}}}})

	handled := w.Handle(term.K(ui.Down))
	if !handled {
		t.Errorf("listbox did not handle")
	}
	if w.ListBox().CopyState().Selected != 1 {
		t.Errorf("listbox state not changed")
	}

	handled = w.Handle(term.K('a'))
	if !handled {
		t.Errorf("codearea did not handle letter key")
	}
	if w.CodeArea().CopyState().Buffer.Content != "a" {
		t.Errorf("codearea state not changed")
	}
	if lastFilter != "a" {
		t.Errorf("OnFilter not called when codearea content changed")
	}

	onFilterCalled = false
	handled = w.Handle(term.PasteSetting(true))
	if !handled {
		t.Errorf("codearea did not handle PasteSetting")
	}
	if onFilterCalled {
		t.Errorf("OnFilter called when codearea content did not change")
	}
	w.Handle(term.PasteSetting(false))

	handled = w.Handle(term.K('D', ui.Ctrl))
	if handled {
		t.Errorf("key unhandled by codearea and listbox got handled")
	}
}

func TestRefilter(t *testing.T) {
	onFilter := make(chan string, 100)
	w := NewComboBox(ComboBoxSpec{
		OnFilter: func(w ComboBox, filter string) {
			onFilter <- filter
		}})
	<-onFilter // Ignore the initial OnFilter call.
	w.CodeArea().MutateState(func(s *CodeAreaState) { s.Buffer.Content = "new" })
	w.Refilter()
	select {
	case f := <-onFilter:
		if f != "new" {
			t.Errorf("OnFilter called with %q, want 'new'", f)
		}
	case <-time.After(time.Second):
		t.Errorf("OnFilter not called by Refilter")
	}
}