summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/olivere/elastic/bulk_index_request_test.go
diff options
context:
space:
mode:
authorChristopher Speller <crspeller@gmail.com>2018-01-29 14:17:40 -0800
committerGitHub <noreply@github.com>2018-01-29 14:17:40 -0800
commit961c04cae992eadb42d286d2f85f8a675bdc68c8 (patch)
tree3408f2d06f847e966c53485e2d54c692cdd037c1 /vendor/github.com/olivere/elastic/bulk_index_request_test.go
parent8d66523ba7d9a77129844be476732ebfd5272d64 (diff)
downloadchat-961c04cae992eadb42d286d2f85f8a675bdc68c8.tar.gz
chat-961c04cae992eadb42d286d2f85f8a675bdc68c8.tar.bz2
chat-961c04cae992eadb42d286d2f85f8a675bdc68c8.zip
Upgrading server dependancies (#8154)
Diffstat (limited to 'vendor/github.com/olivere/elastic/bulk_index_request_test.go')
-rw-r--r--vendor/github.com/olivere/elastic/bulk_index_request_test.go116
1 files changed, 116 insertions, 0 deletions
diff --git a/vendor/github.com/olivere/elastic/bulk_index_request_test.go b/vendor/github.com/olivere/elastic/bulk_index_request_test.go
new file mode 100644
index 000000000..79baf51fb
--- /dev/null
+++ b/vendor/github.com/olivere/elastic/bulk_index_request_test.go
@@ -0,0 +1,116 @@
+// 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"
+ "time"
+)
+
+func TestBulkIndexRequestSerialization(t *testing.T) {
+ tests := []struct {
+ Request BulkableRequest
+ Expected []string
+ }{
+ // #0
+ {
+ Request: NewBulkIndexRequest().Index("index1").Type("doc").Id("1").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"index":{"_index":"index1","_id":"1","_type":"doc"}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ // #1
+ {
+ Request: NewBulkIndexRequest().OpType("create").Index("index1").Type("doc").Id("1").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"create":{"_index":"index1","_id":"1","_type":"doc"}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ // #2
+ {
+ Request: NewBulkIndexRequest().OpType("index").Index("index1").Type("doc").Id("1").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"index":{"_index":"index1","_id":"1","_type":"doc"}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ // #3
+ {
+ Request: NewBulkIndexRequest().OpType("index").Index("index1").Type("doc").Id("1").RetryOnConflict(42).
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"index":{"_index":"index1","_id":"1","_type":"doc","retry_on_conflict":42}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ // #4
+ {
+ Request: NewBulkIndexRequest().OpType("index").Index("index1").Type("doc").Id("1").Pipeline("my_pipeline").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"index":{"_index":"index1","_id":"1","_type":"doc","pipeline":"my_pipeline"}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ // #5
+ {
+ Request: NewBulkIndexRequest().OpType("index").Index("index1").Type("doc").Id("1").
+ Routing("123").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)}),
+ Expected: []string{
+ `{"index":{"_index":"index1","_id":"1","_type":"doc","routing":"123"}}`,
+ `{"user":"olivere","message":"","retweets":0,"created":"2014-01-18T23:59:58Z"}`,
+ },
+ },
+ }
+
+ 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 bulkIndexRequestSerializationResult string
+
+func BenchmarkBulkIndexRequestSerialization(b *testing.B) {
+ b.Run("stdlib", func(b *testing.B) {
+ r := NewBulkIndexRequest().Index(testIndexName).Type("doc").Id("1").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)})
+ benchmarkBulkIndexRequestSerialization(b, r.UseEasyJSON(false))
+ })
+ b.Run("easyjson", func(b *testing.B) {
+ r := NewBulkIndexRequest().Index(testIndexName).Type("doc").Id("1").
+ Doc(tweet{User: "olivere", Created: time.Date(2014, 1, 18, 23, 59, 58, 0, time.UTC)})
+ benchmarkBulkIndexRequestSerialization(b, r.UseEasyJSON(true))
+ })
+}
+
+func benchmarkBulkIndexRequestSerialization(b *testing.B, r *BulkIndexRequest) {
+ var s string
+ for n := 0; n < b.N; n++ {
+ s = r.String()
+ r.source = nil // Don't let caching spoil the benchmark
+ }
+ bulkIndexRequestSerializationResult = s // ensure the compiler doesn't optimize
+ b.ReportAllocs()
+}