1 files changed, 68 insertions, 0 deletions
diff --git a/vendor/gopkg.in/olivere/elastic.v5/search_queries_percolator_test.go b/vendor/gopkg.in/olivere/elastic.v5/search_queries_percolator_test.go
new file mode 100644
index 000000000..8a22d4614
--- /dev/null
+++ b/vendor/gopkg.in/olivere/elastic.v5/search_queries_percolator_test.go
@@ -0,0 +1,68 @@
+// Copyright 2012-present Oliver Eilhard. All rights reserved.
+// Use of this source code is governed by a MIT-license.
+// See http://olivere.mit-license.org/license.txt for details.
+
+package elastic
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestPercolatorQuery(t *testing.T) {
+	q := NewPercolatorQuery().
+		Field("query").
+		DocumentType("doctype").
+		Document(map[string]interface{}{
+			"message": "Some message",
+		})
+	src, err := q.Source()
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err := json.Marshal(src)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"percolate":{"document":{"message":"Some message"},"document_type":"doctype","field":"query"}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestPercolatorQueryWithDetails(t *testing.T) {
+	q := NewPercolatorQuery().
+		Field("query").
+		DocumentType("doctype").
+		Document(map[string]interface{}{
+			"message": "Some message",
+		}).
+		IndexedDocumentIndex("index").
+		IndexedDocumentType("type").
+		IndexedDocumentId("1").
+		IndexedDocumentRouting("route").
+		IndexedDocumentPreference("one").
+		IndexedDocumentVersion(1)
+	src, err := q.Source()
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err := json.Marshal(src)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"percolate":{"document":{"message":"Some message"},"document_type":"doctype","field":"query","id":"1","index":"index","preference":"one","routing":"route","type":"type","version":1}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestPercolatorQueryWithMissingFields(t *testing.T) {
+	q := NewPercolatorQuery() // no Field, Document, or Query
+	_, err := q.Source()
+	if err == nil {
+		t.Fatal("expected error, got nil")
+	}
+}