mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Matthieu MORELandGitHub
1fbdd09ea9
### Description * uses testify instead of testing.Fatal or testing.Error in processor Signed-off-by: Matthieu MOREL <[email protected]>
90 lines
2.6 KiB
Go
90 lines
2.6 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package pprofileotlp // import "go.opentelemetry.io/collector/pdata/pprofile/pprofileotlp"
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var unexpectedBytes = "expected the same bytes from unmarshaling and marshaling."
|
|
|
|
func FuzzRequestUnmarshalJSON(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, data []byte) {
|
|
er := NewExportRequest()
|
|
err := er.UnmarshalJSON(data)
|
|
if err != nil {
|
|
return
|
|
}
|
|
b1, err := er.MarshalJSON()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
er = NewExportRequest()
|
|
require.NoError(t, er.UnmarshalJSON(b1), "failed to unmarshal valid bytes")
|
|
b2, err := er.MarshalJSON()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
require.True(t, bytes.Equal(b1, b2), "%s. \nexpected %d but got %d\n", unexpectedBytes, b1, b2)
|
|
})
|
|
}
|
|
|
|
func FuzzResponseUnmarshalJSON(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, data []byte) {
|
|
er := NewExportResponse()
|
|
err := er.UnmarshalJSON(data)
|
|
if err != nil {
|
|
return
|
|
}
|
|
b1, err := er.MarshalJSON()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
er = NewExportResponse()
|
|
require.NoError(t, er.UnmarshalJSON(b1), "failed to unmarshal valid bytes")
|
|
b2, err := er.MarshalJSON()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
require.True(t, bytes.Equal(b1, b2), "%s. \nexpected %d but got %d\n", unexpectedBytes, b1, b2)
|
|
})
|
|
}
|
|
|
|
func FuzzRequestUnmarshalProto(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, data []byte) {
|
|
er := NewExportRequest()
|
|
err := er.UnmarshalProto(data)
|
|
if err != nil {
|
|
return
|
|
}
|
|
b1, err := er.MarshalProto()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
er = NewExportRequest()
|
|
require.NoError(t, er.UnmarshalProto(b1), "failed to unmarshal valid bytes")
|
|
b2, err := er.MarshalProto()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
require.True(t, bytes.Equal(b1, b2), "%s. \nexpected %d but got %d\n", unexpectedBytes, b1, b2)
|
|
})
|
|
}
|
|
|
|
func FuzzResponseUnmarshalProto(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, data []byte) {
|
|
er := NewExportResponse()
|
|
err := er.UnmarshalProto(data)
|
|
if err != nil {
|
|
return
|
|
}
|
|
b1, err := er.MarshalProto()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
er = NewExportResponse()
|
|
require.NoError(t, er.UnmarshalProto(b1), "failed to unmarshal valid bytes")
|
|
b2, err := er.MarshalProto()
|
|
require.NoError(t, err, "failed to marshal valid struct")
|
|
|
|
require.True(t, bytes.Equal(b1, b2), "%s. \nexpected %d but got %d\n", unexpectedBytes, b1, b2)
|
|
})
|
|
}
|