blob: f3d9dbb9d441fa76acf15fa2dc3c59137c49c848 [file] [log] [blame]
// Copyright 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package source
import (
"strings"
"testing"
)
func TestParseQuery(t *testing.T) {
tests := []struct {
query, s string
wantMatch bool
}{
{"", "anything", false},
{"any", "anything", true},
{"any$", "anything", false},
{"ing$", "anything", true},
{"ing$", "anythinG", true},
{"inG$", "anything", false},
{"^any", "anything", true},
{"^any", "Anything", true},
{"^Any", "anything", false},
{"at", "anything", true},
// TODO: this appears to be a bug in the fuzzy matching algorithm. 'At'
// should cause a case-sensitive match.
// {"At", "anything", false},
{"At", "Anything", true},
{"'yth", "Anything", true},
{"'yti", "Anything", false},
{"'any 'thing", "Anything", true},
{"anythn nythg", "Anything", true},
{"ntx", "Anything", false},
{"anythn", "anything", true},
{"ing", "anything", true},
{"anythn nythgx", "anything", false},
}
for _, test := range tests {
matcher := parseQuery(test.query)
if score := matcher(test.s); score > 0 != test.wantMatch {
t.Errorf("parseQuery(%q) match for %q: %.2g, want match: %t", test.query, test.s, score, test.wantMatch)
}
}
}
func TestBestMatch(t *testing.T) {
tests := []struct {
desc string
symbol string
matcher matcherFunc
wantMatch string
wantScore float64
}{
{
desc: "shortest match",
symbol: "foo/bar/baz.quux",
matcher: func(string) float64 { return 1.0 },
wantMatch: "quux",
wantScore: 1.0,
},
{
desc: "partial match",
symbol: "foo/bar/baz.quux",
matcher: func(s string) float64 {
if strings.HasPrefix(s, "bar") {
return 1.0
}
return 0.0
},
wantMatch: "bar/baz.quux",
wantScore: 1.0,
},
{
desc: "longest match",
symbol: "foo/bar/baz.quux",
matcher: func(s string) float64 {
parts := strings.Split(s, "/")
return float64(len(parts))
},
wantMatch: "foo/bar/baz.quux",
wantScore: 3.0,
},
}
for _, test := range tests {
test := test
t.Run(test.desc, func(t *testing.T) {
gotMatch, gotScore := bestMatch(test.symbol, test.matcher)
if gotMatch != test.wantMatch || gotScore != test.wantScore {
t.Errorf("bestMatch(%q, matcher) = (%q, %.2g), want (%q, %.2g)", test.symbol, gotMatch, gotScore, test.wantMatch, test.wantScore)
}
})
}
}