Adding upstream version 2.5.1.

Signed-off-by: Daniel Baumann <daniel@debian.org>

index/upsidedown/dump_test.go

@@ -0,0 +1,155 @@
+//  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 upsidedown
+
+import (
+	"testing"
+	"time"
+
+	"github.com/blevesearch/bleve/v2/index/upsidedown/store/boltdb"
+	index "github.com/blevesearch/bleve_index_api"
+
+	"github.com/blevesearch/bleve/v2/document"
+)
+
+func TestDump(t *testing.T) {
+	defer func() {
+		err := DestroyTest()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
+
+	analysisQueue := index.NewAnalysisQueue(1)
+	idx, err := NewUpsideDownCouch(boltdb.Name, boltTestConfig, analysisQueue)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = idx.Open()
+	if err != nil {
+		t.Errorf("error opening index: %v", err)
+	}
+	defer func() {
+		err := idx.Close()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
+
+	var expectedCount uint64
+	reader, err := idx.Reader()
+	if err != nil {
+		t.Fatal(err)
+	}
+	docCount, err := reader.DocCount()
+	if err != nil {
+		t.Error(err)
+	}
+	if docCount != expectedCount {
+		t.Errorf("Expected document count to be %d got %d", expectedCount, docCount)
+	}
+	err = reader.Close()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	doc := document.NewDocument("1")
+	doc.AddField(document.NewTextFieldWithIndexingOptions("name", []uint64{}, []byte("test"), index.IndexField|index.StoreField))
+	doc.AddField(document.NewNumericFieldWithIndexingOptions("age", []uint64{}, 35.99, index.IndexField|index.StoreField))
+	dateField, err := document.NewDateTimeFieldWithIndexingOptions("unixEpoch", []uint64{}, time.Unix(0, 0), time.RFC3339, index.IndexField|index.StoreField)
+	if err != nil {
+		t.Error(err)
+	}
+	doc.AddField(dateField)
+	err = idx.Update(doc)
+	if err != nil {
+		t.Errorf("Error updating index: %v", err)
+	}
+
+	doc = document.NewDocument("2")
+	doc.AddField(document.NewTextFieldWithIndexingOptions("name", []uint64{}, []byte("test2"), index.IndexField|index.StoreField))
+	doc.AddField(document.NewNumericFieldWithIndexingOptions("age", []uint64{}, 35.99, index.IndexField|index.StoreField))
+	dateField, err = document.NewDateTimeFieldWithIndexingOptions("unixEpoch", []uint64{}, time.Unix(0, 0), time.RFC3339, index.IndexField|index.StoreField)
+	if err != nil {
+		t.Error(err)
+	}
+	doc.AddField(dateField)
+	err = idx.Update(doc)
+	if err != nil {
+		t.Errorf("Error updating index: %v", err)
+	}
+
+	fieldsCount := 0
+	reader, err = idx.Reader()
+	if err != nil {
+		t.Fatal(err)
+	}
+	upsideDownReader, ok := reader.(*IndexReader)
+	if !ok {
+		t.Fatal("dump is only supported by index type upsidedown")
+	}
+	fieldsRows := upsideDownReader.DumpFields()
+	for range fieldsRows {
+		fieldsCount++
+	}
+	if fieldsCount != 3 {
+		t.Errorf("expected 3 fields, got %d", fieldsCount)
+	}
+
+	// 1 text term
+	// 16 numeric terms
+	// 16 date terms
+	// 3 stored fields
+	expectedDocRowCount := int(1 + (2 * (64 / document.DefaultPrecisionStep)) + 3)
+	docRowCount := 0
+	docRows := upsideDownReader.DumpDoc("1")
+	for range docRows {
+		docRowCount++
+	}
+	if docRowCount != expectedDocRowCount {
+		t.Errorf("expected %d rows for document, got %d", expectedDocRowCount, docRowCount)
+	}
+
+	docRowCount = 0
+	docRows = upsideDownReader.DumpDoc("2")
+	for range docRows {
+		docRowCount++
+	}
+	if docRowCount != expectedDocRowCount {
+		t.Errorf("expected %d rows for document, got %d", expectedDocRowCount, docRowCount)
+	}
+
+	// 1 version
+	// fieldsCount field rows
+	// 2 docs * expectedDocRowCount
+	// 2 back index rows
+	// 2 text term row count (2 different text terms)
+	// 16 numeric term row counts (shared for both docs, same numeric value)
+	// 16 date term row counts (shared for both docs, same date value)
+	expectedAllRowCount := int(1 + fieldsCount + (2 * expectedDocRowCount) + 2 + 2 + int((2 * (64 / document.DefaultPrecisionStep))))
+	allRowCount := 0
+	allRows := upsideDownReader.DumpAll()
+	for range allRows {
+		allRowCount++
+	}
+	if allRowCount != expectedAllRowCount {
+		t.Errorf("expected %d rows for all, got %d", expectedAllRowCount, allRowCount)
+	}
+
+	err = reader.Close()
+	if err != nil {
+		t.Fatal(err)
+	}
+}