mirror of
https://github.com/hanzoai/otel-collector.git
synced 2026-11-03 02:20:04 +00:00
zeekayandHanzo Dev
c16ed9fe26
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers; omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
140 lines
3.7 KiB
Go
140 lines
3.7 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package o11ykafkaexporter
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/goccy/go-json"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.opentelemetry.io/collector/pdata/pcommon"
|
|
"go.opentelemetry.io/collector/pdata/ptrace"
|
|
conventions "go.opentelemetry.io/collector/semconv/v1.6.1"
|
|
)
|
|
|
|
func TestDefaultTracesMarshalers(t *testing.T) {
|
|
expectedEncodings := []string{
|
|
"otlp_proto",
|
|
"otlp_json",
|
|
"jaeger_proto",
|
|
"jaeger_json",
|
|
}
|
|
marshalers := tracesMarshalers()
|
|
assert.Equal(t, len(expectedEncodings), len(marshalers))
|
|
for _, e := range expectedEncodings {
|
|
t.Run(e, func(t *testing.T) {
|
|
m, ok := marshalers[e]
|
|
require.True(t, ok)
|
|
assert.NotNil(t, m)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDefaultMetricsMarshalers(t *testing.T) {
|
|
expectedEncodings := []string{
|
|
"otlp_proto",
|
|
"otlp_json",
|
|
}
|
|
marshalers := metricsMarshalers()
|
|
assert.Equal(t, len(expectedEncodings), len(marshalers))
|
|
for _, e := range expectedEncodings {
|
|
t.Run(e, func(t *testing.T) {
|
|
m, ok := marshalers[e]
|
|
require.True(t, ok)
|
|
assert.NotNil(t, m)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDefaultLogsMarshalers(t *testing.T) {
|
|
expectedEncodings := []string{
|
|
"otlp_proto",
|
|
"otlp_json",
|
|
"raw",
|
|
}
|
|
marshalers := logsMarshalers()
|
|
assert.Equal(t, len(expectedEncodings), len(marshalers))
|
|
for _, e := range expectedEncodings {
|
|
t.Run(e, func(t *testing.T) {
|
|
m, ok := marshalers[e]
|
|
require.True(t, ok)
|
|
assert.NotNil(t, m)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestOTLPTracesJsonMarshaling(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
now := time.Unix(1, 0)
|
|
|
|
traces := ptrace.NewTraces()
|
|
traces.ResourceSpans().AppendEmpty()
|
|
|
|
rs := traces.ResourceSpans().At(0)
|
|
rs.SetSchemaUrl(conventions.SchemaURL)
|
|
rs.ScopeSpans().AppendEmpty()
|
|
|
|
ils := rs.ScopeSpans().At(0)
|
|
ils.SetSchemaUrl(conventions.SchemaURL)
|
|
ils.Spans().AppendEmpty()
|
|
|
|
span := ils.Spans().At(0)
|
|
span.SetKind(ptrace.SpanKindInternal)
|
|
span.SetName(t.Name())
|
|
span.SetStartTimestamp(pcommon.NewTimestampFromTime(now))
|
|
span.SetEndTimestamp(pcommon.NewTimestampFromTime(now.Add(time.Second)))
|
|
span.SetSpanID([8]byte{0, 1, 2, 3, 4, 5, 6, 7})
|
|
span.SetParentSpanID([8]byte{8, 9, 10, 11, 12, 13, 14})
|
|
|
|
marshaler, ok := tracesMarshalers()["otlp_json"]
|
|
require.True(t, ok, "Must have otlp json marshaller")
|
|
|
|
msg, err := marshaler.Marshal(traces, t.Name())
|
|
require.NoError(t, err, "Must have marshaled the data without error")
|
|
require.Len(t, msg, 1, "Must have one entry in the message")
|
|
|
|
data, err := msg[0].Value.Encode()
|
|
require.NoError(t, err, "Must not error when encoding value")
|
|
require.NotNil(t, data, "Must have valid data to test")
|
|
|
|
// Since marshaling json is not guaranteed to be in order
|
|
// within a string, using a map to compare that the expected values are there
|
|
expectedJSON := map[string]any{
|
|
"resourceSpans": []any{
|
|
map[string]any{
|
|
"resource": map[string]any{},
|
|
"scopeSpans": []any{
|
|
map[string]any{
|
|
"scope": map[string]any{},
|
|
"spans": []any{
|
|
map[string]any{
|
|
"spanId": "0001020304050607",
|
|
"parentSpanId": "08090a0b0c0d0e00",
|
|
"name": t.Name(),
|
|
"kind": float64(ptrace.SpanKindInternal),
|
|
"startTimeUnixNano": fmt.Sprint(now.UnixNano()),
|
|
"endTimeUnixNano": fmt.Sprint(now.Add(time.Second).UnixNano()),
|
|
"status": map[string]any{},
|
|
},
|
|
},
|
|
"schemaUrl": conventions.SchemaURL,
|
|
},
|
|
},
|
|
"schemaUrl": conventions.SchemaURL,
|
|
},
|
|
},
|
|
}
|
|
|
|
var final map[string]any
|
|
err = json.Unmarshal(data, &final)
|
|
require.NoError(t, err, "Must not error marshaling expected data")
|
|
|
|
assert.Equal(t, expectedJSON, final, "Must match the expected value")
|
|
}
|