1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
|
// 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, false)
if err != nil {
b.Fatal(err)
}
}
testReq = req
}
func BenchmarkRequestSetBodyStringGzip(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, true)
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, false)
if err != nil {
b.Fatal(err)
}
}
testReq = req
}
func BenchmarkRequestSetBodyBytesGzip(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, true)
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, false)
if err != nil {
b.Fatal(err)
}
}
testReq = req
}
func BenchmarkRequestSetBodyMapGzip(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, true)
if err != nil {
b.Fatal(err)
}
}
testReq = req
}
|