Add filter tests

This commit is contained in:
c-bata 2017-07-18 19:04:11 +09:00
parent 58303e0e68
commit 850fa92045
2 changed files with 74 additions and 0 deletions

View File

@ -2,6 +2,8 @@ package prompt
import "strings"
type Filter func([]string, string, bool) []string
func FilterHasPrefix(completions []string, sub string, ignoreCase bool) []string {
if sub == "" {
return completions

72
filter_test.go Normal file
View File

@ -0,0 +1,72 @@
package prompt
import (
"testing"
"reflect"
)
func TestFilter(t *testing.T) {
var scenarioTable = [] struct {
scenario string
filter Filter
list []string
substr string
ignoreCase bool
expected []string
} {
{
scenario: "Contains don't ignore case",
filter: FilterContains,
list: []string{"abcde", "fghij", "ABCDE"},
substr: "cd",
ignoreCase: false,
expected: []string{"abcde"},
},
{
scenario: "Contains ignore case",
filter: FilterContains,
list: []string{"abcde", "fghij", "ABCDE"},
substr: "cd",
ignoreCase: true,
expected: []string{"abcde", "ABCDE"},
},
{
scenario: "HasPrefix don't ignore case",
filter: FilterHasPrefix,
list: []string{"abcde", "fghij", "ABCDE"},
substr: "abc",
ignoreCase: false,
expected: []string{"abcde"},
},
{
scenario: "HasPrefix ignore case",
filter: FilterHasPrefix,
list: []string{"abcde", "fabcj", "ABCDE"},
substr: "abc",
ignoreCase: true,
expected: []string{"abcde", "ABCDE"},
},
{
scenario: "HasSuffix don't ignore case",
filter: FilterHasSuffix,
list: []string{"abcde", "fcdej", "ABCDE"},
substr: "cde",
ignoreCase: false,
expected: []string{"abcde"},
},
{
scenario: "HasSuffix ignore case",
filter: FilterHasSuffix,
list: []string{"abcde", "fcdej", "ABCDE"},
substr: "cde",
ignoreCase: true,
expected: []string{"abcde", "ABCDE"},
},
}
for _, s := range scenarioTable {
if actual := s.filter(s.list, s.substr, s.ignoreCase); !reflect.DeepEqual(actual, s.expected) {
t.Errorf("%s: Should be %#v, but got %#v", s.scenario, s.expected, actual)
}
}
}