mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Bogdan DrutuandGitHub
dd4ff349f0
Also, this PR improves JSON unmarshal logic by avoiding stack calls caused by using the Read[Array|Object]CB and replace with simple non-recursive iterations. Signed-off-by: Bogdan Drutu <[email protected]>
95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package json
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNestedObject(t *testing.T) {
|
|
s := BorrowStream(nil)
|
|
defer ReturnStream(s)
|
|
|
|
s.WriteObjectStart()
|
|
s.WriteObjectField("field1")
|
|
s.WriteString("val1")
|
|
s.WriteObjectField("field2")
|
|
s.WriteObjectStart()
|
|
s.WriteObjectField("field3")
|
|
s.WriteObjectStart()
|
|
s.WriteObjectField("field4")
|
|
s.WriteString("val4")
|
|
s.WriteObjectField("field5")
|
|
s.WriteString("val5")
|
|
s.WriteObjectEnd()
|
|
s.WriteObjectField("field6")
|
|
s.WriteString("val6")
|
|
s.WriteObjectField("field7")
|
|
s.WriteObjectStart()
|
|
s.WriteObjectEnd()
|
|
s.WriteObjectEnd()
|
|
s.WriteObjectEnd()
|
|
|
|
expected := `{
|
|
"field1": "val1",
|
|
"field2": {
|
|
"field3": {
|
|
"field4": "val4",
|
|
"field5": "val5"
|
|
},
|
|
"field6": "val6",
|
|
"field7": {}
|
|
}
|
|
}`
|
|
assert.JSONEq(t, expected, string(s.Buffer()))
|
|
}
|
|
|
|
func TestMarshalFloat(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
inputFloat float64
|
|
expected string
|
|
}{
|
|
{
|
|
name: "positive infinity",
|
|
inputFloat: math.Inf(1),
|
|
expected: `"Infinity"`,
|
|
},
|
|
{
|
|
name: "negative infinity",
|
|
inputFloat: math.Inf(-1),
|
|
expected: `"-Infinity"`,
|
|
},
|
|
{
|
|
name: "not-a-number",
|
|
inputFloat: math.NaN(),
|
|
expected: `"NaN"`,
|
|
},
|
|
{
|
|
name: "regular float",
|
|
inputFloat: math.MaxFloat64,
|
|
expected: "1.7976931348623157e+308",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := BorrowStream(nil)
|
|
defer ReturnStream(s)
|
|
s.WriteFloat64(tt.inputFloat)
|
|
require.Equal(t, tt.expected, string(s.Buffer()))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWriteBytes(t *testing.T) {
|
|
s := BorrowStream(nil)
|
|
defer ReturnStream(s)
|
|
s.WriteBytes([]byte("test"))
|
|
require.Equal(t, `"dGVzdA=="`, string(s.Buffer()))
|
|
}
|