110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Code generated by "internal/cmd/pdatagen/main.go". DO NOT EDIT.
|
|
// To regenerate this file run "make genpdata".
|
|
|
|
package internal
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
gootlptrace "go.opentelemetry.io/proto/slim/otlp/trace/v1"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
otlptrace "go.opentelemetry.io/collector/pdata/internal/data/protogen/trace/v1"
|
|
"go.opentelemetry.io/collector/pdata/internal/json"
|
|
)
|
|
|
|
func TestCopyOrigSpan(t *testing.T) {
|
|
src := NewOrigPtrSpan()
|
|
dest := NewOrigPtrSpan()
|
|
CopyOrigSpan(dest, src)
|
|
assert.Equal(t, NewOrigPtrSpan(), dest)
|
|
FillOrigTestSpan(src)
|
|
CopyOrigSpan(dest, src)
|
|
assert.Equal(t, src, dest)
|
|
}
|
|
|
|
func TestMarshalAndUnmarshalJSONOrigSpanUnknown(t *testing.T) {
|
|
iter := json.BorrowIterator([]byte(`{"unknown": "string"}`))
|
|
defer json.ReturnIterator(iter)
|
|
dest := NewOrigPtrSpan()
|
|
UnmarshalJSONOrigSpan(dest, iter)
|
|
require.NoError(t, iter.Error())
|
|
assert.Equal(t, NewOrigPtrSpan(), dest)
|
|
}
|
|
|
|
func TestMarshalAndUnmarshalJSONOrigSpan(t *testing.T) {
|
|
for name, src := range getEncodingTestValuesSpan() {
|
|
t.Run(name, func(t *testing.T) {
|
|
stream := json.BorrowStream(nil)
|
|
defer json.ReturnStream(stream)
|
|
MarshalJSONOrigSpan(src, stream)
|
|
require.NoError(t, stream.Error())
|
|
|
|
iter := json.BorrowIterator(stream.Buffer())
|
|
defer json.ReturnIterator(iter)
|
|
dest := NewOrigPtrSpan()
|
|
UnmarshalJSONOrigSpan(dest, iter)
|
|
require.NoError(t, iter.Error())
|
|
|
|
assert.Equal(t, src, dest)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMarshalAndUnmarshalProtoOrigSpanUnknown(t *testing.T) {
|
|
dest := NewOrigPtrSpan()
|
|
// message Test { required int64 field = 1313; } encoding { "field": "1234" }
|
|
require.NoError(t, UnmarshalProtoOrigSpan(dest, []byte{0x88, 0x52, 0xD2, 0x09}))
|
|
assert.Equal(t, NewOrigPtrSpan(), dest)
|
|
}
|
|
|
|
func TestMarshalAndUnmarshalProtoOrigSpan(t *testing.T) {
|
|
for name, src := range getEncodingTestValuesSpan() {
|
|
t.Run(name, func(t *testing.T) {
|
|
buf := make([]byte, SizeProtoOrigSpan(src))
|
|
gotSize := MarshalProtoOrigSpan(src, buf)
|
|
assert.Equal(t, len(buf), gotSize)
|
|
|
|
dest := NewOrigPtrSpan()
|
|
require.NoError(t, UnmarshalProtoOrigSpan(dest, buf))
|
|
assert.Equal(t, src, dest)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMarshalAndUnmarshalProtoViaProtobufSpan(t *testing.T) {
|
|
for name, src := range getEncodingTestValuesSpan() {
|
|
t.Run(name, func(t *testing.T) {
|
|
buf := make([]byte, SizeProtoOrigSpan(src))
|
|
gotSize := MarshalProtoOrigSpan(src, buf)
|
|
assert.Equal(t, len(buf), gotSize)
|
|
|
|
goDest := &gootlptrace.Span{}
|
|
require.NoError(t, proto.Unmarshal(buf, goDest))
|
|
|
|
goBuf, err := proto.Marshal(goDest)
|
|
require.NoError(t, err)
|
|
|
|
dest := NewOrigPtrSpan()
|
|
require.NoError(t, UnmarshalProtoOrigSpan(dest, goBuf))
|
|
assert.Equal(t, src, dest)
|
|
})
|
|
}
|
|
}
|
|
|
|
func getEncodingTestValuesSpan() map[string]*otlptrace.Span {
|
|
return map[string]*otlptrace.Span{
|
|
"empty": NewOrigPtrSpan(),
|
|
"fill_test": func() *otlptrace.Span {
|
|
src := NewOrigPtrSpan()
|
|
FillOrigTestSpan(src)
|
|
return src
|
|
}(),
|
|
}
|
|
}
|