add test for term searcher
This commit is contained in:
parent
63fdd841ac
commit
2d238d9872
|
@ -0,0 +1,91 @@
|
|||
// Copyright (c) 2013 Couchbase, Inc.
|
||||
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
|
||||
// except in compliance with the License. You may obtain a copy of the License at
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
// Unless required by applicable law or agreed to in writing, software distributed under the
|
||||
// License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
|
||||
// either express or implied. See the License for the specific language governing permissions
|
||||
// and limitations under the License.
|
||||
package search
|
||||
|
||||
import (
|
||||
// "math"
|
||||
// "reflect"
|
||||
"testing"
|
||||
|
||||
// "github.com/couchbaselabs/bleve/index"
|
||||
"github.com/couchbaselabs/bleve/document"
|
||||
"github.com/couchbaselabs/bleve/index/mock"
|
||||
)
|
||||
|
||||
func TestTermSearcher(t *testing.T) {
|
||||
|
||||
query := TermQuery{
|
||||
Term: "beer",
|
||||
Field: "desc",
|
||||
BoostVal: 1.0,
|
||||
Explain: true,
|
||||
}
|
||||
|
||||
i := mock.NewMockIndex()
|
||||
i.Update(&document.Document{
|
||||
ID: "a",
|
||||
Fields: []*document.Field{
|
||||
document.NewTextField("desc", []byte("beer")),
|
||||
},
|
||||
})
|
||||
i.Update(&document.Document{
|
||||
ID: "b",
|
||||
Fields: []*document.Field{
|
||||
document.NewTextField("desc", []byte("beer")),
|
||||
},
|
||||
})
|
||||
i.Update(&document.Document{
|
||||
ID: "c",
|
||||
Fields: []*document.Field{
|
||||
document.NewTextField("desc", []byte("beer")),
|
||||
},
|
||||
})
|
||||
|
||||
searcher, err := NewTermSearcher(i, &query)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if searcher.Count() != 3 {
|
||||
t.Errorf("expected count of 3, got %d", searcher.Count())
|
||||
}
|
||||
|
||||
docMatch, err := searcher.Next()
|
||||
if err != nil {
|
||||
t.Errorf("expected result, got %v", err)
|
||||
}
|
||||
if docMatch.ID != "a" {
|
||||
t.Errorf("expected result ID to be 'a', got '%s", docMatch.ID)
|
||||
}
|
||||
docMatch, err = searcher.Advance("c")
|
||||
if err != nil {
|
||||
t.Errorf("expected result, got %v", err)
|
||||
}
|
||||
if docMatch.ID != "c" {
|
||||
t.Errorf("expected result ID to be 'c' got '%s'", docMatch.ID)
|
||||
}
|
||||
|
||||
// try advancing past end
|
||||
docMatch, err = searcher.Advance("f")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if docMatch != nil {
|
||||
t.Errorf("expected nil, got %v", docMatch)
|
||||
}
|
||||
|
||||
// try pushing next past end
|
||||
docMatch, err = searcher.Next()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if docMatch != nil {
|
||||
t.Errorf("expected nil, got %v", docMatch)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue