diff options
author | Christopher Speller <crspeller@gmail.com> | 2018-01-29 14:17:40 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-01-29 14:17:40 -0800 |
commit | 961c04cae992eadb42d286d2f85f8a675bdc68c8 (patch) | |
tree | 3408f2d06f847e966c53485e2d54c692cdd037c1 /vendor/github.com/olivere/elastic/bulk_update_request_test.go | |
parent | 8d66523ba7d9a77129844be476732ebfd5272d64 (diff) | |
download | chat-961c04cae992eadb42d286d2f85f8a675bdc68c8.tar.gz chat-961c04cae992eadb42d286d2f85f8a675bdc68c8.tar.bz2 chat-961c04cae992eadb42d286d2f85f8a675bdc68c8.zip |
Upgrading server dependancies (#8154)
Diffstat (limited to 'vendor/github.com/olivere/elastic/bulk_update_request_test.go')
-rw-r--r-- | vendor/github.com/olivere/elastic/bulk_update_request_test.go | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/vendor/github.com/olivere/elastic/bulk_update_request_test.go b/vendor/github.com/olivere/elastic/bulk_update_request_test.go new file mode 100644 index 000000000..53e73bd40 --- /dev/null +++ b/vendor/github.com/olivere/elastic/bulk_update_request_test.go @@ -0,0 +1,149 @@ +// 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 TestBulkUpdateRequestSerialization(t *testing.T) { + tests := []struct { + Request BulkableRequest + Expected []string + }{ + // #0 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("1").Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"1"}}`, + `{"doc":{"counter":42}}`, + }, + }, + // #1 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("1"). + Routing("123"). + RetryOnConflict(3). + DocAsUpsert(true). + Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"1","retry_on_conflict":3,"routing":"123"}}`, + `{"doc":{"counter":42},"doc_as_upsert":true}`, + }, + }, + // #2 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("1"). + RetryOnConflict(3). + Script(NewScript(`ctx._source.retweets += param1`).Lang("javascript").Param("param1", 42)). + Upsert(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"1","retry_on_conflict":3}}`, + `{"script":{"lang":"javascript","params":{"param1":42},"source":"ctx._source.retweets += param1"},"upsert":{"counter":42}}`, + }, + }, + // #3 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("1").DetectNoop(true).Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"1"}}`, + `{"detect_noop":true,"doc":{"counter":42}}`, + }, + }, + // #4 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("1"). + RetryOnConflict(3). + ScriptedUpsert(true). + Script(NewScript(`ctx._source.retweets += param1`).Lang("javascript").Param("param1", 42)). + Upsert(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"1","retry_on_conflict":3}}`, + `{"script":{"lang":"javascript","params":{"param1":42},"source":"ctx._source.retweets += param1"},"scripted_upsert":true,"upsert":{"counter":42}}`, + }, + }, + // #5 + { + Request: NewBulkUpdateRequest().Index("index1").Type("doc").Id("4").ReturnSource(true).Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }), + Expected: []string{ + `{"update":{"_index":"index1","_type":"doc","_id":"4"}}`, + `{"doc":{"counter":42},"_source":true}`, + }, + }, + } + + for i, test := range tests { + lines, err := test.Request.Source() + if err != nil { + t.Fatalf("#%d: expected no error, got: %v", i, err) + } + if lines == nil { + t.Fatalf("#%d: expected lines, got nil", i) + } + if len(lines) != len(test.Expected) { + t.Fatalf("#%d: expected %d lines, got %d", i, len(test.Expected), len(lines)) + } + for j, line := range lines { + if line != test.Expected[j] { + t.Errorf("#%d: expected line #%d to be\n%s\nbut got:\n%s", i, j, test.Expected[j], line) + } + } + } +} + +var bulkUpdateRequestSerializationResult string + +func BenchmarkBulkUpdateRequestSerialization(b *testing.B) { + b.Run("stdlib", func(b *testing.B) { + r := NewBulkUpdateRequest().Index("index1").Type("doc").Id("1").Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }) + benchmarkBulkUpdateRequestSerialization(b, r.UseEasyJSON(false)) + }) + b.Run("easyjson", func(b *testing.B) { + r := NewBulkUpdateRequest().Index("index1").Type("doc").Id("1").Doc(struct { + Counter int64 `json:"counter"` + }{ + Counter: 42, + }).UseEasyJSON(false) + benchmarkBulkUpdateRequestSerialization(b, r.UseEasyJSON(true)) + }) +} + +func benchmarkBulkUpdateRequestSerialization(b *testing.B, r *BulkUpdateRequest) { + var s string + for n := 0; n < b.N; n++ { + s = r.String() + r.source = nil // Don't let caching spoil the benchmark + } + bulkUpdateRequestSerializationResult = s // ensure the compiler doesn't optimize + b.ReportAllocs() +} |