fields_test.go

// Copyright 2020 OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package sumologicexporter

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"go.opentelemetry.io/collector/pdata/pcommon"
)

func TestFields(t *testing.T) {
	testcases := []struct {
		name     string
		fields   map[string]string
		expected string
	}{
		{
			name: "string",
			fields: map[string]string{
				"key1": "value1",
				"key3": "value3",
				"key2": "value2",
			},
			expected: "key1=value1, key2=value2, key3=value3",
		},
		{
			name: "sanitization",
			fields: map[string]string{
				"key1":   "value,1",
				"key3":   "value\n3",
				"key=,2": "valu,e=2",
			},
			expected: "key1=value_1, key3=value_3, key:_2=valu_e:2",
		},
		{
			name: "empty element",
			fields: map[string]string{
				"key1": "value1",
				"key3": "value3",
				"key2": "",
			},
			expected: "key1=value1, key3=value3",
		},
	}

	for _, tc := range testcases {
		t.Run(tc.name, func(t *testing.T) {
			flds := fieldsFromMap(tc.fields)

			assert.Equal(t, tc.expected, flds.string())
		})
	}
}

func BenchmarkFields(b *testing.B) {
	attrMap := pcommon.NewMap()
	flds := map[string]interface{}{
		"key1": "value1",
		"key3": "value3",
		"key2": "",
		"map": map[string]string{
			"key1": "value1",
			"key3": "value3",
			"key2": "",
		},
	}
	for k, v := range flds {
		switch v := v.(type) {
		case string:
			attrMap.PutStr(k, v)
		case map[string]string:
			m := pcommon.NewValueMap()
			mm := m.Map().AsRaw()
			for kk, vv := range v {
				mm[kk] = vv
			}
			m.CopyTo(attrMap.PutEmpty(k))
		}
	}
	sut := newFields(attrMap)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_ = sut.string()
	}
}