1 files changed, 58 insertions, 0 deletions
diff --git a/vendor/gopkg.in/olivere/elastic.v5/suggester_completion_test.go b/vendor/gopkg.in/olivere/elastic.v5/suggester_completion_test.go
index 6bffddfe7..adbf58657 100644
--- a/vendor/gopkg.in/olivere/elastic.v5/suggester_completion_test.go
+++ b/vendor/gopkg.in/olivere/elastic.v5/suggester_completion_test.go
@@ -28,6 +28,64 @@ func TestCompletionSuggesterSource(t *testing.T) {
 	}
 }
 
+func TestCompletionSuggesterPrefixSource(t *testing.T) {
+	s := NewCompletionSuggester("song-suggest").
+		Prefix("nir").
+		Field("suggest")
+	src, err := s.Source(true)
+	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 := `{"song-suggest":{"prefix":"nir","completion":{"field":"suggest"}}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestCompletionSuggesterPrefixWithFuzzySource(t *testing.T) {
+	s := NewCompletionSuggester("song-suggest").
+		Prefix("nor").
+		Field("suggest").
+		FuzzyOptions(NewFuzzyCompletionSuggesterOptions().EditDistance(2))
+	src, err := s.Source(true)
+	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 := `{"song-suggest":{"prefix":"nor","completion":{"field":"suggest","fuzzy":{"fuzziness":2}}}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestCompletionSuggesterRegexSource(t *testing.T) {
+	s := NewCompletionSuggester("song-suggest").
+		Regex("n[ever|i]r").
+		Field("suggest")
+	src, err := s.Source(true)
+	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 := `{"song-suggest":{"regex":"n[ever|i]r","completion":{"field":"suggest"}}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
 func TestCompletionSuggesterSourceWithMultipleContexts(t *testing.T) {
 	s := NewCompletionSuggester("song-suggest").
 		Text("n").