1 files changed, 72 insertions, 0 deletions
diff --git a/vendor/github.com/olivere/elastic/request_test.go b/vendor/github.com/olivere/elastic/request_test.go
new file mode 100644
index 000000000..04fbecbab
--- /dev/null
+++ b/vendor/github.com/olivere/elastic/request_test.go
@@ -0,0 +1,72 @@
+// 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 "testing"
+
+var testReq *Request // used as a temporary variable to avoid compiler optimizations in tests/benchmarks
+
+func TestRequestSetContentType(t *testing.T) {
+	req, err := NewRequest("GET", "/")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if want, have := "application/json", req.Header.Get("Content-Type"); want != have {
+		t.Fatalf("want %q, have %q", want, have)
+	}
+	req.Header.Set("Content-Type", "application/x-ndjson")
+	if want, have := "application/x-ndjson", req.Header.Get("Content-Type"); want != have {
+		t.Fatalf("want %q, have %q", want, have)
+	}
+}
+
+func BenchmarkRequestSetBodyString(b *testing.B) {
+	req, err := NewRequest("GET", "/")
+	if err != nil {
+		b.Fatal(err)
+	}
+	for i := 0; i < b.N; i++ {
+		body := `{"query":{"match_all":{}}}`
+		err = req.SetBody(body)
+		if err != nil {
+			b.Fatal(err)
+		}
+	}
+	testReq = req
+}
+
+func BenchmarkRequestSetBodyBytes(b *testing.B) {
+	req, err := NewRequest("GET", "/")
+	if err != nil {
+		b.Fatal(err)
+	}
+	for i := 0; i < b.N; i++ {
+		body := []byte(`{"query":{"match_all":{}}}`)
+		err = req.SetBody(body)
+		if err != nil {
+			b.Fatal(err)
+		}
+	}
+	testReq = req
+}
+
+func BenchmarkRequestSetBodyMap(b *testing.B) {
+	req, err := NewRequest("GET", "/")
+	if err != nil {
+		b.Fatal(err)
+	}
+	for i := 0; i < b.N; i++ {
+		body := map[string]interface{}{
+			"query": map[string]interface{}{
+				"match_all": map[string]interface{}{},
+			},
+		}
+		err = req.SetBody(body)
+		if err != nil {
+			b.Fatal(err)
+		}
+	}
+	testReq = req
+}