You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.2 KiB
79 lines
2.2 KiB
// 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" |
|
) |
|
|
|
func TestBulkDeleteRequestSerialization(t *testing.T) { |
|
tests := []struct { |
|
Request BulkableRequest |
|
Expected []string |
|
}{ |
|
// #0 |
|
{ |
|
Request: NewBulkDeleteRequest().Index("index1").Type("tweet").Id("1"), |
|
Expected: []string{ |
|
`{"delete":{"_id":"1","_index":"index1","_type":"tweet"}}`, |
|
}, |
|
}, |
|
// #1 |
|
{ |
|
Request: NewBulkDeleteRequest().Index("index1").Type("tweet").Id("1").Parent("2"), |
|
Expected: []string{ |
|
`{"delete":{"_id":"1","_index":"index1","_parent":"2","_type":"tweet"}}`, |
|
}, |
|
}, |
|
// #2 |
|
{ |
|
Request: NewBulkDeleteRequest().Index("index1").Type("tweet").Id("1").Routing("3"), |
|
Expected: []string{ |
|
`{"delete":{"_id":"1","_index":"index1","_routing":"3","_type":"tweet"}}`, |
|
}, |
|
}, |
|
} |
|
|
|
for i, test := range tests { |
|
lines, err := test.Request.Source() |
|
if err != nil { |
|
t.Fatalf("case #%d: expected no error, got: %v", i, err) |
|
} |
|
if lines == nil { |
|
t.Fatalf("case #%d: expected lines, got nil", i) |
|
} |
|
if len(lines) != len(test.Expected) { |
|
t.Fatalf("case #%d: expected %d lines, got %d", i, len(test.Expected), len(lines)) |
|
} |
|
for j, line := range lines { |
|
if line != test.Expected[j] { |
|
t.Errorf("case #%d: expected line #%d to be %s, got: %s", i, j, test.Expected[j], line) |
|
} |
|
} |
|
} |
|
} |
|
|
|
var bulkDeleteRequestSerializationResult string |
|
|
|
func BenchmarkBulkDeleteRequestSerialization(b *testing.B) { |
|
b.Run("stdlib", func(b *testing.B) { |
|
r := NewBulkDeleteRequest().Index(testIndexName).Type("doc").Id("1") |
|
benchmarkBulkDeleteRequestSerialization(b, r.UseEasyJSON(false)) |
|
}) |
|
b.Run("easyjson", func(b *testing.B) { |
|
r := NewBulkDeleteRequest().Index(testIndexName).Type("doc").Id("1") |
|
benchmarkBulkDeleteRequestSerialization(b, r.UseEasyJSON(true)) |
|
}) |
|
} |
|
|
|
func benchmarkBulkDeleteRequestSerialization(b *testing.B, r *BulkDeleteRequest) { |
|
var s string |
|
for n := 0; n < b.N; n++ { |
|
s = r.String() |
|
r.source = nil // Don't let caching spoil the benchmark |
|
} |
|
bulkDeleteRequestSerializationResult = s // ensure the compiler doesn't optimize |
|
b.ReportAllocs() |
|
}
|
|
|