complete_segment_test.go 3.19 KB
Newer Older
chzyer's avatar
chzyer committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
package readline

import (
	"fmt"
	"testing"

	"github.com/chzyer/test"
)

func rs(s [][]rune) []string {
	ret := make([]string, len(s))
	for idx, ss := range s {
		ret[idx] = string(ss)
	}
	return ret
}

func sr(s ...string) [][]rune {
	ret := make([][]rune, len(s))
	for idx, ss := range s {
		ret[idx] = []rune(ss)
	}
	return ret
}

func TestRetSegment(t *testing.T) {
	defer test.New(t)
	// a
	// |- a1
	// |--- a11
	// |--- a12
	// |- a2
	// |--- a21
	// b
chzyer's avatar
chzyer committed
35 36
	// add
	// adddomain
chzyer's avatar
chzyer committed
37 38 39 40 41 42 43
	ret := []struct {
		Segments [][]rune
		Cands    [][]rune
		idx      int
		Ret      [][]rune
		pos      int
	}{
chzyer's avatar
chzyer committed
44 45
		{sr(""), sr("a", "b", "add", "adddomain"), 0, sr("a", "b", "add", "adddomain"), 0},
		{sr("a"), sr("a", "add", "adddomain"), 1, sr("", "dd", "dddomain"), 1},
chzyer's avatar
chzyer committed
46 47 48
		{sr("a", ""), sr("a1", "a2"), 0, sr("a1", "a2"), 0},
		{sr("a", "a"), sr("a1", "a2"), 1, sr("1", "2"), 1},
		{sr("a", "a1"), sr("a1"), 2, sr(""), 2},
chzyer's avatar
chzyer committed
49
		{sr("add"), sr("add", "adddomain"), 2, sr("", "domain"), 2},
chzyer's avatar
chzyer committed
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
	}
	for idx, r := range ret {
		ret, pos := RetSegment(r.Segments, r.Cands, r.idx)
		test.Equal(ret, r.Ret, fmt.Errorf("%v", idx))
		test.Equal(pos, r.pos, fmt.Errorf("%v", idx))
	}
}

func TestSplitSegment(t *testing.T) {
	defer test.New(t)
	// a
	// |- a1
	// |--- a11
	// |--- a12
	// |- a2
	// |--- a21
	// b
	ret := []struct {
		Line     string
		Pos      int
		Segments [][]rune
		Idx      int
	}{
		{"", 0, sr(""), 0},
		{"a", 1, sr("a"), 1},
		{"a ", 2, sr("a", ""), 0},
		{"a a", 3, sr("a", "a"), 1},
		{"a a1", 4, sr("a", "a1"), 2},
		{"a a1 ", 5, sr("a", "a1", ""), 0},
	}

	for i, r := range ret {
		ret, idx := SplitSegment([]rune(r.Line), r.Pos)
		test.Equal(rs(ret), rs(r.Segments), fmt.Errorf("%v", i))
		test.Equal(idx, r.Idx, fmt.Errorf("%v", i))
	}
}

type Tree struct {
	Name     string
	Children []Tree
}

func TestSegmentCompleter(t *testing.T) {
	defer test.New(t)

	tree := Tree{"", []Tree{
		{"a", []Tree{
			{"a1", []Tree{
				{"a11", nil},
				{"a12", nil},
			}},
			{"a2", []Tree{
				{"a21", nil},
			}},
		}},
		{"b", nil},
chzyer's avatar
chzyer committed
107 108 109 110
		{"route", []Tree{
			{"add", nil},
			{"adddomain", nil},
		}},
chzyer's avatar
chzyer committed
111 112 113 114
	}}
	s := SegmentFunc(func(ret [][]rune, n int) [][]rune {
		tree := tree
	main:
chzyer's avatar
chzyer committed
115
		for level := 0; level < len(ret)-1; {
chzyer's avatar
chzyer committed
116 117 118 119 120 121 122 123 124 125
			name := string(ret[level])
			for _, t := range tree.Children {
				if t.Name == name {
					tree = t
					level++
					continue main
				}
			}
		}

chzyer's avatar
chzyer committed
126 127 128 129 130
		ret = make([][]rune, len(tree.Children))
		for idx, r := range tree.Children {
			ret[idx] = []rune(r.Name)
		}
		return ret
chzyer's avatar
chzyer committed
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
	})

	// a
	// |- a1
	// |--- a11
	// |--- a12
	// |- a2
	// |--- a21
	// b
	ret := []struct {
		Line  string
		Pos   int
		Ret   [][]rune
		Share int
	}{
chzyer's avatar
chzyer committed
146
		{"", 0, sr("a", "b", "route"), 0},
chzyer's avatar
chzyer committed
147 148 149 150 151 152 153 154
		{"a", 1, sr(""), 1},
		{"a ", 2, sr("a1", "a2"), 0},
		{"a a", 3, sr("1", "2"), 1},
		{"a a1", 4, sr(""), 2},
		{"a a1 ", 5, sr("a11", "a12"), 0},
		{"a a1 a", 6, sr("11", "12"), 1},
		{"a a1 a1", 7, sr("1", "2"), 2},
		{"a a1 a11", 8, sr(""), 3},
chzyer's avatar
chzyer committed
155
		{"route add", 9, sr("", "domain"), 3},
chzyer's avatar
chzyer committed
156
	}
chzyer's avatar
chzyer committed
157 158 159 160 161
	for _, r := range ret {
		for idx, rr := range r.Ret {
			r.Ret[idx] = append(rr, ' ')
		}
	}
chzyer's avatar
chzyer committed
162 163
	for i, r := range ret {
		newLine, length := s.Do([]rune(r.Line), r.Pos)
chzyer's avatar
chzyer committed
164
		test.Equal(rs(newLine), rs(r.Ret), fmt.Errorf("%v", i))
chzyer's avatar
chzyer committed
165 166 167
		test.Equal(length, r.Share, fmt.Errorf("%v", i))
	}
}