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.
|
2014-08-25 18:06:10 +02:00
|
|
|
|
2014-07-30 18:30:38 +02:00
|
|
|
package bleve
|
|
|
|
|
|
|
|
import (
|
2014-08-20 22:58:20 +02:00
|
|
|
"io/ioutil"
|
2014-07-30 18:30:38 +02:00
|
|
|
"os"
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2014-11-25 21:56:43 +01:00
|
|
|
func TestCrud(t *testing.T) {
|
|
|
|
defer os.RemoveAll("testidx")
|
|
|
|
|
|
|
|
index, err := New("testidx", NewIndexMapping())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
doca := map[string]interface{}{
|
|
|
|
"name": "marty",
|
|
|
|
"desc": "gophercon india",
|
|
|
|
}
|
|
|
|
err = index.Index("a", doca)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
docy := map[string]interface{}{
|
|
|
|
"name": "jasper",
|
|
|
|
"desc": "clojure",
|
|
|
|
}
|
|
|
|
err = index.Index("y", docy)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
err = index.Delete("y")
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
docx := map[string]interface{}{
|
|
|
|
"name": "rose",
|
|
|
|
"desc": "googler",
|
|
|
|
}
|
|
|
|
err = index.Index("x", docx)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
err = index.SetInternal([]byte("status"), []byte("pending"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
docb := map[string]interface{}{
|
|
|
|
"name": "steve",
|
|
|
|
"desc": "cbft master",
|
|
|
|
}
|
|
|
|
batch := NewBatch()
|
|
|
|
batch.Index("b", docb)
|
|
|
|
batch.Delete("x")
|
|
|
|
batch.SetInternal([]byte("batchi"), []byte("batchv"))
|
|
|
|
batch.DeleteInternal([]byte("status"))
|
|
|
|
err = index.Batch(batch)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
val, err := index.GetInternal([]byte("batchi"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if string(val) != "batchv" {
|
|
|
|
t.Errorf("expected 'batchv', got '%s'", val)
|
|
|
|
}
|
|
|
|
val, err = index.GetInternal([]byte("status"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if val != nil {
|
|
|
|
t.Errorf("expected nil, got '%s'", val)
|
|
|
|
}
|
|
|
|
|
|
|
|
err = index.SetInternal([]byte("seqno"), []byte("7"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
err = index.SetInternal([]byte("status"), []byte("ready"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
err = index.DeleteInternal([]byte("status"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
val, err = index.GetInternal([]byte("status"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if val != nil {
|
|
|
|
t.Errorf("expected nil, got '%s'", val)
|
|
|
|
}
|
|
|
|
|
|
|
|
val, err = index.GetInternal([]byte("seqno"))
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
if string(val) != "7" {
|
|
|
|
t.Errorf("expected '7', got '%s'", val)
|
|
|
|
}
|
2014-11-25 22:23:48 +01:00
|
|
|
|
|
|
|
// close the index, open it again, and try some more things
|
|
|
|
index.Close()
|
|
|
|
|
|
|
|
index, err = Open("testidx")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
2014-12-01 18:51:01 +01:00
|
|
|
defer index.Close()
|
2014-11-25 22:23:48 +01:00
|
|
|
|
|
|
|
count, err := index.DocCount()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if count != 2 {
|
|
|
|
t.Errorf("expected doc count 2, got %d", count)
|
|
|
|
}
|
|
|
|
|
|
|
|
doc, err := index.Document("a")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
if doc == nil {
|
|
|
|
t.Errorf("expected doc not nil, got nil")
|
|
|
|
}
|
|
|
|
foundNameField := false
|
|
|
|
for _, field := range doc.Fields {
|
|
|
|
if field.Name() == "name" && string(field.Value()) == "marty" {
|
|
|
|
foundNameField = true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !foundNameField {
|
|
|
|
t.Errorf("expected to find field named 'name' with value 'marty'")
|
|
|
|
}
|
2014-11-26 21:36:58 +01:00
|
|
|
|
|
|
|
fields, err := index.Fields()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
expectedFields := map[string]bool{
|
|
|
|
"_all": false,
|
|
|
|
"name": false,
|
|
|
|
"desc": false,
|
|
|
|
}
|
|
|
|
if len(fields) != len(expectedFields) {
|
|
|
|
t.Fatalf("expected %d fields got %d", len(expectedFields), len(fields))
|
|
|
|
}
|
|
|
|
for _, f := range fields {
|
|
|
|
expectedFields[f] = true
|
|
|
|
}
|
|
|
|
for ef, efp := range expectedFields {
|
|
|
|
if !efp {
|
|
|
|
t.Errorf("field %s is missing", ef)
|
|
|
|
}
|
|
|
|
}
|
2014-11-25 21:56:43 +01:00
|
|
|
}
|
|
|
|
|
2014-08-20 22:58:20 +02:00
|
|
|
func TestIndexCreateNewOverExisting(t *testing.T) {
|
|
|
|
defer os.RemoveAll("testidx")
|
|
|
|
|
|
|
|
index, err := New("testidx", NewIndexMapping())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
index.Close()
|
|
|
|
index, err = New("testidx", NewIndexMapping())
|
2014-09-02 20:14:05 +02:00
|
|
|
if err != ErrorIndexPathExists {
|
2014-08-20 22:58:20 +02:00
|
|
|
t.Fatalf("expected error index path exists, got %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestIndexOpenNonExisting(t *testing.T) {
|
|
|
|
_, err := Open("doesnotexist")
|
2014-09-02 20:14:05 +02:00
|
|
|
if err != ErrorIndexPathDoesNotExist {
|
2014-08-20 22:58:20 +02:00
|
|
|
t.Fatalf("expected error index path does not exist, got %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestIndexOpenMetaMissingOrCorrupt(t *testing.T) {
|
|
|
|
defer os.RemoveAll("testidx")
|
|
|
|
|
|
|
|
index, err := New("testidx", NewIndexMapping())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
index.Close()
|
|
|
|
|
2014-11-21 22:47:20 +01:00
|
|
|
// now intentionally change the storage type
|
|
|
|
ioutil.WriteFile("testidx/index_meta.json", []byte(`{"storage":"mystery"}`), 0666)
|
|
|
|
|
|
|
|
index, err = Open("testidx")
|
|
|
|
if err != ErrorUnknownStorageType {
|
|
|
|
t.Fatalf("expected error unkown storage type, got %v", err)
|
|
|
|
}
|
|
|
|
|
2014-08-20 22:58:20 +02:00
|
|
|
// now intentionally corrupt the metadata
|
|
|
|
ioutil.WriteFile("testidx/index_meta.json", []byte("corrupted"), 0666)
|
|
|
|
|
|
|
|
index, err = Open("testidx")
|
2014-09-02 20:14:05 +02:00
|
|
|
if err != ErrorIndexMetaCorrupt {
|
2014-08-20 22:58:20 +02:00
|
|
|
t.Fatalf("expected error index metadata corrupted, got %v", err)
|
|
|
|
}
|
|
|
|
|
2014-12-18 18:43:12 +01:00
|
|
|
// now intentionally remove the metadata
|
2014-08-20 22:58:20 +02:00
|
|
|
os.Remove("testidx/index_meta.json")
|
|
|
|
|
|
|
|
index, err = Open("testidx")
|
2014-09-02 20:14:05 +02:00
|
|
|
if err != ErrorIndexMetaMissing {
|
2014-08-20 22:58:20 +02:00
|
|
|
t.Fatalf("expected error index metadata missing, got %v", err)
|
|
|
|
}
|
|
|
|
}
|
2014-11-21 22:47:20 +01:00
|
|
|
|
|
|
|
func TestInMemIndex(t *testing.T) {
|
|
|
|
|
|
|
|
index, err := New("", NewIndexMapping())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
index.Close()
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClosedIndex(t *testing.T) {
|
|
|
|
index, err := New("", NewIndexMapping())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
index.Close()
|
|
|
|
|
|
|
|
err = index.Index("test", "test")
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
err = index.Delete("test")
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
b := NewBatch()
|
|
|
|
err = index.Batch(b)
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = index.Document("test")
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = index.DocCount()
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = index.Search(NewSearchRequest(NewTermQuery("test")))
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = index.Fields()
|
|
|
|
if err != ErrorIndexClosed {
|
|
|
|
t.Errorf("expected error index closed, got %v", err)
|
|
|
|
}
|
|
|
|
}
|