1 files changed, 87 insertions, 0 deletions
diff --git a/tui/termfo/termfo_test.go b/tui/termfo/termfo_test.go
new file mode 100644
index 0000000..60564ef
--- /dev/null
+++ b/tui/termfo/termfo_test.go
@@ -0,0 +1,87 @@
+package termfo
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"strings"
+	"testing"
+)
+
+func BenchmarkNew(b *testing.B) {
+	t := os.Getenv("TERM")
+	b.ReportAllocs()
+	for n := 0; n < b.N; n++ {
+		New(t)
+	}
+}
+
+func TestNew(t *testing.T) {
+	tests := []struct {
+		term    string
+		intSize int
+	}{
+		{"xterm", 2},
+		{"xterm-256color", 4},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.term, func(t *testing.T) {
+			ti, err := New(tt.term)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			if ti.Name != tt.term {
+				t.Errorf("wrong name: %q", ti.Name)
+			}
+			if ti.IntSize != tt.intSize {
+				t.Errorf("intsize: %d", ti.IntSize)
+			}
+
+			// Just a basic sanity check.
+			if len(ti.Bools) < 10 || len(ti.Numbers) < 5 || len(ti.Strings) < 200 || len(ti.Extended) < 50 {
+				t.Errorf("bools: %d  nums: %d  strs: %d  ext: %d", len(ti.Bools), len(ti.Numbers), len(ti.Strings), len(ti.Extended))
+			}
+		})
+	}
+}
+
+func TestKeys(t *testing.T) {
+	run := func(ti *Terminfo, seq string) []string {
+		var collect []string
+		ch := ti.FindKeys(strings.NewReader(seq))
+		for e := <-ch; ; e = <-ch {
+			if e.Err != nil {
+				if e.Err == io.EOF {
+					break
+				}
+				t.Fatal(e.Err)
+			}
+			collect = append(collect, e.Key.String())
+		}
+		return collect
+	}
+
+	{
+		ti, err := New("xterm")
+		if err != nil {
+			t.Fatal(err)
+		}
+		f := run(ti, "\x1bOP\x1b[1;2P\x1bOA")
+		if fmt.Sprintf("%s", f) != "[<F1> <S-F1> <Up>]" {
+			t.Error(f)
+		}
+	}
+
+	{
+		ti, err := New("vi200")
+		if err != nil {
+			t.Fatal(err)
+		}
+		f := run(ti, "\x1b?q\x1bA")
+		if fmt.Sprintf("%s", f) != "[<F1> <Up>]" {
+			t.Error(f)
+		}
+	}
+}