added tests
This commit is contained in:
parent
d7341524aa
commit
8150146dc7
73
search/fragment_formatter_html_test.go
Normal file
73
search/fragment_formatter_html_test.go
Normal file
|
@ -0,0 +1,73 @@
|
|||
package search
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestHTMLFragmentFormatterDefault(t *testing.T) {
|
||||
tests := []struct {
|
||||
fragment *Fragment
|
||||
tlm TermLocationMap
|
||||
output string
|
||||
}{
|
||||
{
|
||||
fragment: &Fragment{
|
||||
orig: []byte("the quick brown fox"),
|
||||
start: 0,
|
||||
end: 19,
|
||||
},
|
||||
tlm: TermLocationMap{
|
||||
"quick": Locations{
|
||||
&Location{
|
||||
Pos: 2,
|
||||
Start: 4,
|
||||
End: 9,
|
||||
},
|
||||
},
|
||||
},
|
||||
output: "the <b>quick</b> brown fox",
|
||||
},
|
||||
}
|
||||
|
||||
emHtmlFormatter := NewHTMLFragmentFormatter()
|
||||
for _, test := range tests {
|
||||
result := emHtmlFormatter.Format(test.fragment, test.tlm)
|
||||
if result != test.output {
|
||||
t.Errorf("expected `%s`, got `%s`", test.output, result)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTMLFragmentFormatterCustom(t *testing.T) {
|
||||
tests := []struct {
|
||||
fragment *Fragment
|
||||
tlm TermLocationMap
|
||||
output string
|
||||
}{
|
||||
{
|
||||
fragment: &Fragment{
|
||||
orig: []byte("the quick brown fox"),
|
||||
start: 0,
|
||||
end: 19,
|
||||
},
|
||||
tlm: TermLocationMap{
|
||||
"quick": Locations{
|
||||
&Location{
|
||||
Pos: 2,
|
||||
Start: 4,
|
||||
End: 9,
|
||||
},
|
||||
},
|
||||
},
|
||||
output: "the <em>quick</em> brown fox",
|
||||
},
|
||||
}
|
||||
|
||||
emHtmlFormatter := NewHTMLFragmentFormatterCustom("<em>", "</em>")
|
||||
for _, test := range tests {
|
||||
result := emHtmlFormatter.Format(test.fragment, test.tlm)
|
||||
if result != test.output {
|
||||
t.Errorf("expected `%s`, got `%s`", test.output, result)
|
||||
}
|
||||
}
|
||||
}
|
|
@ -35,6 +35,27 @@ func TestQuerySyntaxParserValid(t *testing.T) {
|
|||
Explain: true,
|
||||
},
|
||||
},
|
||||
{
|
||||
input: `"test phrase 1"`,
|
||||
mapping: document.Mapping{},
|
||||
result: &TermBooleanQuery{
|
||||
Should: &TermDisjunctionQuery{
|
||||
Terms: []Query{
|
||||
&MatchPhraseQuery{
|
||||
MatchPhrase: "test phrase 1",
|
||||
Field: "_all",
|
||||
BoostVal: 1.0,
|
||||
Explain: true,
|
||||
},
|
||||
},
|
||||
BoostVal: 1.0,
|
||||
Explain: true,
|
||||
Min: 1.0,
|
||||
},
|
||||
BoostVal: 1.0,
|
||||
Explain: true,
|
||||
},
|
||||
},
|
||||
{
|
||||
input: "field:test",
|
||||
mapping: document.Mapping{},
|
||||
|
|
Loading…
Reference in New Issue
Block a user