mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
Zach LeslieandGitHub
7b773771b8
* Update prometheus dependency and function signature * Update vendor * Update serverless vendor * Downgrade protobuf patch release for round-trip gogocodec issue * Hold back protobuf * Skip failing test * Replace test equality
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
// This is copied over from Jaeger and modified to work for Tempo
|
|
|
|
package gogocodec
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
|
|
"github.com/grafana/tempo/pkg/tempopb"
|
|
)
|
|
|
|
func TestCodecMarshallAndUnmarshall_tempo_type(t *testing.T) {
|
|
// marshal a tempo object using the custom codec
|
|
c := NewCodec()
|
|
req1 := &tempopb.TraceByIDRequest{
|
|
TraceID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
}
|
|
data, err := c.Marshal(req1)
|
|
require.NoError(t, err)
|
|
|
|
// unmarshal and check if its the same
|
|
req2 := &tempopb.TraceByIDRequest{}
|
|
err = c.Unmarshal(data, req2)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, req1, req2)
|
|
}
|
|
|
|
func TestCodecMarshallAndUnmarshall_foreign_type(t *testing.T) {
|
|
// marshal a foreign object (anything other than Tempo/Cortex/Jaeger) using the custom codec
|
|
c := NewCodec()
|
|
goprotoMessage1 := &emptypb.Empty{}
|
|
data, err := c.Marshal(goprotoMessage1)
|
|
require.NoError(t, err)
|
|
|
|
// unmarshal and check if its the same
|
|
goprotoMessage2 := &emptypb.Empty{}
|
|
err = c.Unmarshal(data, goprotoMessage2)
|
|
require.NoError(t, err)
|
|
assert.True(t, proto.Equal(goprotoMessage1, goprotoMessage2))
|
|
}
|
|
|
|
func TestWireCompatibility(t *testing.T) {
|
|
// marshal a tempo object using the custom codec
|
|
c := NewCodec()
|
|
req1 := &tempopb.TraceByIDRequest{
|
|
TraceID: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
|
|
}
|
|
data, err := c.Marshal(req1)
|
|
require.NoError(t, err)
|
|
|
|
// unmarshal this into the generic empty type using golang proto
|
|
var goprotoMessage emptypb.Empty
|
|
err = proto.Unmarshal(data, &goprotoMessage)
|
|
require.NoError(t, err)
|
|
|
|
// marshal emptypb using golang proto
|
|
data2, err := proto.Marshal(&goprotoMessage)
|
|
require.NoError(t, err)
|
|
|
|
req2 := &tempopb.TraceByIDRequest{}
|
|
err = c.Unmarshal(data2, req2)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, req1, req2)
|
|
}
|