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