1// Copyright (c) 2016 Uber Technologies, Inc. 2// 3// Permission is hereby granted, free of charge, to any person obtaining a copy 4// of this software and associated documentation files (the "Software"), to deal 5// in the Software without restriction, including without limitation the rights 6// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7// copies of the Software, and to permit persons to whom the Software is 8// furnished to do so, subject to the following conditions: 9// 10// The above copyright notice and this permission notice shall be included in 11// all copies or substantial portions of the Software. 12// 13// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19// THE SOFTWARE. 20 21package zap 22 23import ( 24 "testing" 25 "time" 26 27 "go.uber.org/zap/zapcore" 28 29 "github.com/stretchr/testify/assert" 30) 31 32func BenchmarkBoolsArrayMarshaler(b *testing.B) { 33 // Keep this benchmark here to capture the overhead of the ArrayMarshaler 34 // wrapper. 35 bs := make([]bool, 50) 36 enc := zapcore.NewJSONEncoder(zapcore.EncoderConfig{}) 37 b.ResetTimer() 38 for i := 0; i < b.N; i++ { 39 Bools("array", bs).AddTo(enc.Clone()) 40 } 41} 42 43func BenchmarkBoolsReflect(b *testing.B) { 44 bs := make([]bool, 50) 45 enc := zapcore.NewJSONEncoder(zapcore.EncoderConfig{}) 46 b.ResetTimer() 47 for i := 0; i < b.N; i++ { 48 Reflect("array", bs).AddTo(enc.Clone()) 49 } 50} 51 52func TestArrayWrappers(t *testing.T) { 53 tests := []struct { 54 desc string 55 field Field 56 expected []interface{} 57 }{ 58 {"empty bools", Bools("", []bool{}), []interface{}{}}, 59 {"empty byte strings", ByteStrings("", [][]byte{}), []interface{}{}}, 60 {"empty complex128s", Complex128s("", []complex128{}), []interface{}{}}, 61 {"empty complex64s", Complex64s("", []complex64{}), []interface{}{}}, 62 {"empty durations", Durations("", []time.Duration{}), []interface{}{}}, 63 {"empty float64s", Float64s("", []float64{}), []interface{}{}}, 64 {"empty float32s", Float32s("", []float32{}), []interface{}{}}, 65 {"empty ints", Ints("", []int{}), []interface{}{}}, 66 {"empty int64s", Int64s("", []int64{}), []interface{}{}}, 67 {"empty int32s", Int32s("", []int32{}), []interface{}{}}, 68 {"empty int16s", Int16s("", []int16{}), []interface{}{}}, 69 {"empty int8s", Int8s("", []int8{}), []interface{}{}}, 70 {"empty strings", Strings("", []string{}), []interface{}{}}, 71 {"empty times", Times("", []time.Time{}), []interface{}{}}, 72 {"empty uints", Uints("", []uint{}), []interface{}{}}, 73 {"empty uint64s", Uint64s("", []uint64{}), []interface{}{}}, 74 {"empty uint32s", Uint32s("", []uint32{}), []interface{}{}}, 75 {"empty uint16s", Uint16s("", []uint16{}), []interface{}{}}, 76 {"empty uint8s", Uint8s("", []uint8{}), []interface{}{}}, 77 {"empty uintptrs", Uintptrs("", []uintptr{}), []interface{}{}}, 78 {"bools", Bools("", []bool{true, false}), []interface{}{true, false}}, 79 {"byte strings", ByteStrings("", [][]byte{{1, 2}, {3, 4}}), []interface{}{"\x01\x02", "\x03\x04"}}, 80 {"complex128s", Complex128s("", []complex128{1 + 2i, 3 + 4i}), []interface{}{1 + 2i, 3 + 4i}}, 81 {"complex64s", Complex64s("", []complex64{1 + 2i, 3 + 4i}), []interface{}{complex64(1 + 2i), complex64(3 + 4i)}}, 82 {"durations", Durations("", []time.Duration{1, 2}), []interface{}{time.Nanosecond, 2 * time.Nanosecond}}, 83 {"float64s", Float64s("", []float64{1.2, 3.4}), []interface{}{1.2, 3.4}}, 84 {"float32s", Float32s("", []float32{1.2, 3.4}), []interface{}{float32(1.2), float32(3.4)}}, 85 {"ints", Ints("", []int{1, 2}), []interface{}{1, 2}}, 86 {"int64s", Int64s("", []int64{1, 2}), []interface{}{int64(1), int64(2)}}, 87 {"int32s", Int32s("", []int32{1, 2}), []interface{}{int32(1), int32(2)}}, 88 {"int16s", Int16s("", []int16{1, 2}), []interface{}{int16(1), int16(2)}}, 89 {"int8s", Int8s("", []int8{1, 2}), []interface{}{int8(1), int8(2)}}, 90 {"strings", Strings("", []string{"foo", "bar"}), []interface{}{"foo", "bar"}}, 91 {"times", Times("", []time.Time{time.Unix(0, 0), time.Unix(0, 0)}), []interface{}{time.Unix(0, 0), time.Unix(0, 0)}}, 92 {"uints", Uints("", []uint{1, 2}), []interface{}{uint(1), uint(2)}}, 93 {"uint64s", Uint64s("", []uint64{1, 2}), []interface{}{uint64(1), uint64(2)}}, 94 {"uint32s", Uint32s("", []uint32{1, 2}), []interface{}{uint32(1), uint32(2)}}, 95 {"uint16s", Uint16s("", []uint16{1, 2}), []interface{}{uint16(1), uint16(2)}}, 96 {"uint8s", Uint8s("", []uint8{1, 2}), []interface{}{uint8(1), uint8(2)}}, 97 {"uintptrs", Uintptrs("", []uintptr{1, 2}), []interface{}{uintptr(1), uintptr(2)}}, 98 } 99 100 for _, tt := range tests { 101 enc := zapcore.NewMapObjectEncoder() 102 tt.field.Key = "k" 103 tt.field.AddTo(enc) 104 assert.Equal(t, tt.expected, enc.Fields["k"], "%s: unexpected map contents.", tt.desc) 105 assert.Equal(t, 1, len(enc.Fields), "%s: found extra keys in map: %v", tt.desc, enc.Fields) 106 } 107} 108