2014-07-30 18:30:38 +02:00
|
|
|
// Copyright (c) 2014 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 bleve
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
|
|
|
"testing"
|
2014-08-15 22:50:34 +02:00
|
|
|
"time"
|
2014-07-30 18:30:38 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type Address struct {
|
|
|
|
Street string `json:"street"`
|
|
|
|
City string `json:"city"`
|
|
|
|
State string `json:"state"`
|
|
|
|
Zip string `json:"zip"`
|
|
|
|
}
|
|
|
|
|
|
|
|
type Person struct {
|
|
|
|
Identifier string `json:"id"`
|
|
|
|
Name string `json:"name"`
|
2014-08-15 22:50:34 +02:00
|
|
|
Age float64 `json:"age"`
|
|
|
|
Title string `json:"title"`
|
|
|
|
Birthday time.Time `json:"birthday"`
|
2014-07-30 18:30:38 +02:00
|
|
|
Address *Address `json:"address"`
|
|
|
|
Hideouts []*Address `json:"hideouts"`
|
|
|
|
Tags []string `json:"tags"`
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *Person) Type() string {
|
|
|
|
return "person"
|
|
|
|
}
|
|
|
|
|
2014-08-15 22:50:34 +02:00
|
|
|
var nameMapping = NewDocumentMapping().
|
|
|
|
AddFieldMapping(NewFieldMapping("", "text", "en", true, true, true, true))
|
|
|
|
|
|
|
|
var tagsMapping = NewDocumentMapping().
|
|
|
|
AddFieldMapping(NewFieldMapping("", "text", "standard", true, true, true, false))
|
|
|
|
var personMapping = NewDocumentMapping().
|
|
|
|
AddSubDocumentMapping("name", nameMapping).
|
|
|
|
AddSubDocumentMapping("id", NewDocumentDisabledMapping()).
|
|
|
|
AddSubDocumentMapping("tags", tagsMapping)
|
|
|
|
|
|
|
|
var mapping = NewIndexMapping().
|
|
|
|
AddDocumentMapping("person", personMapping)
|
|
|
|
|
|
|
|
var people = []*Person{
|
|
|
|
&Person{
|
|
|
|
Identifier: "a",
|
|
|
|
Name: "marty",
|
|
|
|
Age: 19,
|
|
|
|
Birthday: time.Unix(1000000000, 0),
|
|
|
|
Title: "mista",
|
|
|
|
},
|
|
|
|
&Person{
|
|
|
|
Identifier: "b",
|
|
|
|
Name: "steve has a long name",
|
|
|
|
Age: 27,
|
|
|
|
Birthday: time.Unix(1000000000, 0),
|
|
|
|
Title: "missess",
|
|
|
|
},
|
|
|
|
&Person{
|
|
|
|
Identifier: "c",
|
|
|
|
Name: "bob walks home",
|
|
|
|
Age: 64,
|
|
|
|
Birthday: time.Unix(1400000000, 0),
|
|
|
|
Title: "masta",
|
|
|
|
},
|
|
|
|
&Person{
|
|
|
|
Identifier: "d",
|
|
|
|
Name: "bobbleheaded wings top the phone",
|
|
|
|
Age: 72,
|
|
|
|
Birthday: time.Unix(1400000000, 0),
|
|
|
|
Title: "mizz",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
// FIXME needs more assertions
|
|
|
|
func TestIndex(t *testing.T) {
|
|
|
|
defer os.RemoveAll("testidx")
|
|
|
|
|
|
|
|
index, err := Open("testidx", mapping)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2014-08-15 22:50:34 +02:00
|
|
|
// index all the people
|
|
|
|
for _, person := range people {
|
|
|
|
err = index.Index(person.Identifier, person)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
termQuery := NewTermQuery("marti").SetField("name")
|
|
|
|
searchRequest := NewSearchRequest(termQuery)
|
|
|
|
searchResult, err := index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(1) {
|
|
|
|
t.Errorf("expected 1 total hit for term query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "a" {
|
|
|
|
t.Errorf("expected top hit id 'a', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
termQuery = NewTermQuery("noone").SetField("name")
|
|
|
|
searchRequest = NewSearchRequest(termQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(0) {
|
|
|
|
t.Errorf("expected 0 total hits")
|
|
|
|
}
|
|
|
|
|
|
|
|
matchPhraseQuery := NewMatchPhraseQuery("long name")
|
|
|
|
searchRequest = NewSearchRequest(matchPhraseQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(1) {
|
|
|
|
t.Errorf("expected 1 total hit for phrase query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "b" {
|
|
|
|
t.Errorf("expected top hit id 'b', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
termQuery = NewTermQuery("walking").SetField("name")
|
|
|
|
searchRequest = NewSearchRequest(termQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(0) {
|
|
|
|
t.Errorf("expected 0 total hits")
|
|
|
|
}
|
|
|
|
|
|
|
|
matchQuery := NewMatchQuery("walking").SetField("name")
|
|
|
|
searchRequest = NewSearchRequest(matchQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(1) {
|
|
|
|
t.Errorf("expected 1 total hit for match query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "c" {
|
|
|
|
t.Errorf("expected top hit id 'c', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
prefixQuery := NewPrefixQuery("bobble").SetField("name")
|
|
|
|
searchRequest = NewSearchRequest(prefixQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
2014-07-30 18:30:38 +02:00
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
2014-08-15 22:50:34 +02:00
|
|
|
if searchResult.Total != uint64(1) {
|
|
|
|
t.Errorf("expected 1 total hit for prefix query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "d" {
|
|
|
|
t.Errorf("expected top hit id 'd', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
syntaxQuery := NewSyntaxQuery("+name:phone")
|
|
|
|
searchRequest = NewSearchRequest(syntaxQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(1) {
|
|
|
|
t.Errorf("expected 1 total hit for syntax query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "d" {
|
|
|
|
t.Errorf("expected top hit id 'd', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
maxAge := 30.0
|
|
|
|
numericRangeQuery := NewNumericRangeQuery(nil, &maxAge).SetField("age")
|
|
|
|
searchRequest = NewSearchRequest(numericRangeQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(2) {
|
|
|
|
t.Errorf("expected 2 total hits for numeric range query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "b" {
|
|
|
|
t.Errorf("expected top hit id 'b', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
if searchResult.Hits[1].ID != "a" {
|
|
|
|
t.Errorf("expected next hit id 'a', got '%s'", searchResult.Hits[1].ID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
startDate = "2010-01-01"
|
|
|
|
dateRangeQuery := NewDateRangeQuery(&startDate, nil).SetField("birthday")
|
|
|
|
searchRequest = NewSearchRequest(dateRangeQuery)
|
|
|
|
searchResult, err = index.Search(searchRequest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if searchResult.Total != uint64(2) {
|
|
|
|
t.Errorf("expected 2 total hits for numeric range query, got %d", searchResult.Total)
|
|
|
|
} else {
|
|
|
|
if searchResult.Hits[0].ID != "d" {
|
|
|
|
t.Errorf("expected top hit id 'd', got '%s'", searchResult.Hits[0].ID)
|
|
|
|
}
|
|
|
|
if searchResult.Hits[1].ID != "c" {
|
|
|
|
t.Errorf("expected next hit id 'c', got '%s'", searchResult.Hits[1].ID)
|
|
|
|
}
|
|
|
|
}
|
2014-07-30 18:30:38 +02:00
|
|
|
}
|