242 lines
7.0 KiB
Go
242 lines
7.0 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package fanoutconsumer
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/collector/consumer"
|
|
"go.opentelemetry.io/collector/consumer/consumertest"
|
|
"go.opentelemetry.io/collector/pdata/testdata"
|
|
)
|
|
|
|
func TestTracesNotMultiplexing(t *testing.T) {
|
|
nop := consumertest.NewNop()
|
|
tfc := NewTraces([]consumer.Traces{nop})
|
|
assert.Same(t, nop, tfc)
|
|
}
|
|
|
|
func TestTracesNotMultiplexingMutating(t *testing.T) {
|
|
p := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
lfc := NewTraces([]consumer.Traces{p})
|
|
assert.True(t, lfc.Capabilities().MutatesData)
|
|
}
|
|
|
|
func TestTracesMultiplexingNonMutating(t *testing.T) {
|
|
p1 := new(consumertest.TracesSink)
|
|
p2 := new(consumertest.TracesSink)
|
|
p3 := new(consumertest.TracesSink)
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
assert.False(t, tfc.Capabilities().MutatesData)
|
|
td := testdata.GenerateTraces(1)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
err := tfc.ConsumeTraces(context.Background(), td)
|
|
if err != nil {
|
|
t.Errorf("Wanted nil got error")
|
|
return
|
|
}
|
|
}
|
|
|
|
assert.Equal(t, td, p1.AllTraces()[0])
|
|
assert.Equal(t, td, p1.AllTraces()[1])
|
|
assert.Equal(t, td, p1.AllTraces()[0])
|
|
assert.Equal(t, td, p1.AllTraces()[1])
|
|
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
|
|
// The data should be marked as read only.
|
|
assert.True(t, td.IsReadOnly())
|
|
}
|
|
|
|
func TestTracesMultiplexingMutating(t *testing.T) {
|
|
p1 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p2 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p3 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
assert.True(t, tfc.Capabilities().MutatesData)
|
|
td := testdata.GenerateTraces(1)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
err := tfc.ConsumeTraces(context.Background(), td)
|
|
if err != nil {
|
|
t.Errorf("Wanted nil got error")
|
|
return
|
|
}
|
|
}
|
|
|
|
assert.NotSame(t, &td, &p1.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p1.AllTraces()[1])
|
|
assert.Equal(t, td, p1.AllTraces()[0])
|
|
assert.Equal(t, td, p1.AllTraces()[1])
|
|
|
|
assert.NotSame(t, &td, &p2.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p2.AllTraces()[1])
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
|
|
// For this consumer, will receive the initial data.
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
|
|
// The data should not be marked as read only.
|
|
assert.False(t, td.IsReadOnly())
|
|
}
|
|
|
|
func TestReadOnlyTracesMultiplexingMutating(t *testing.T) {
|
|
p1 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p2 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p3 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
assert.True(t, tfc.Capabilities().MutatesData)
|
|
|
|
tdOrig := testdata.GenerateTraces(1)
|
|
td := testdata.GenerateTraces(1)
|
|
td.MarkReadOnly()
|
|
|
|
for i := 0; i < 2; i++ {
|
|
err := tfc.ConsumeTraces(context.Background(), td)
|
|
if err != nil {
|
|
t.Errorf("Wanted nil got error")
|
|
return
|
|
}
|
|
}
|
|
|
|
// All consumers should receive the cloned data.
|
|
|
|
assert.NotEqual(t, td, p1.AllTraces()[0])
|
|
assert.NotEqual(t, td, p1.AllTraces()[1])
|
|
assert.Equal(t, tdOrig, p1.AllTraces()[0])
|
|
assert.Equal(t, tdOrig, p1.AllTraces()[1])
|
|
|
|
assert.NotEqual(t, td, p2.AllTraces()[0])
|
|
assert.NotEqual(t, td, p2.AllTraces()[1])
|
|
assert.Equal(t, tdOrig, p2.AllTraces()[0])
|
|
assert.Equal(t, tdOrig, p2.AllTraces()[1])
|
|
|
|
assert.NotEqual(t, td, p3.AllTraces()[0])
|
|
assert.NotEqual(t, td, p3.AllTraces()[1])
|
|
assert.Equal(t, tdOrig, p3.AllTraces()[0])
|
|
assert.Equal(t, tdOrig, p3.AllTraces()[1])
|
|
}
|
|
|
|
func TestTracesMultiplexingMixLastMutating(t *testing.T) {
|
|
p1 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p2 := new(consumertest.TracesSink)
|
|
p3 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
assert.False(t, tfc.Capabilities().MutatesData)
|
|
td := testdata.GenerateTraces(1)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
err := tfc.ConsumeTraces(context.Background(), td)
|
|
if err != nil {
|
|
t.Errorf("Wanted nil got error")
|
|
return
|
|
}
|
|
}
|
|
|
|
assert.NotSame(t, &td, &p1.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p1.AllTraces()[1])
|
|
assert.Equal(t, td, p1.AllTraces()[0])
|
|
assert.Equal(t, td, p1.AllTraces()[1])
|
|
|
|
// For this consumer, will receive the initial data.
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
|
|
// For this consumer, will clone the initial data.
|
|
assert.NotSame(t, &td, &p3.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p3.AllTraces()[1])
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
|
|
// The data should not be marked as read only.
|
|
assert.False(t, td.IsReadOnly())
|
|
}
|
|
|
|
func TestTracesMultiplexingMixLastNonMutating(t *testing.T) {
|
|
p1 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p2 := &mutatingTracesSink{TracesSink: new(consumertest.TracesSink)}
|
|
p3 := new(consumertest.TracesSink)
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
assert.False(t, tfc.Capabilities().MutatesData)
|
|
td := testdata.GenerateTraces(1)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
err := tfc.ConsumeTraces(context.Background(), td)
|
|
if err != nil {
|
|
t.Errorf("Wanted nil got error")
|
|
return
|
|
}
|
|
}
|
|
|
|
assert.NotSame(t, &td, &p1.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p1.AllTraces()[1])
|
|
assert.Equal(t, td, p1.AllTraces()[0])
|
|
assert.Equal(t, td, p1.AllTraces()[1])
|
|
|
|
assert.NotSame(t, &td, &p2.AllTraces()[0])
|
|
assert.NotSame(t, &td, &p2.AllTraces()[1])
|
|
assert.Equal(t, td, p2.AllTraces()[0])
|
|
assert.Equal(t, td, p2.AllTraces()[1])
|
|
|
|
// For this consumer, will receive the initial data.
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
|
|
// The data should not be marked as read only.
|
|
assert.False(t, td.IsReadOnly())
|
|
}
|
|
|
|
func TestTracesWhenErrors(t *testing.T) {
|
|
p1 := mutatingErr{Consumer: consumertest.NewErr(errors.New("my error"))}
|
|
p2 := consumertest.NewErr(errors.New("my error"))
|
|
p3 := new(consumertest.TracesSink)
|
|
|
|
tfc := NewTraces([]consumer.Traces{p1, p2, p3})
|
|
td := testdata.GenerateTraces(1)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
require.Error(t, tfc.ConsumeTraces(context.Background(), td))
|
|
}
|
|
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
assert.Equal(t, td, p3.AllTraces()[0])
|
|
assert.Equal(t, td, p3.AllTraces()[1])
|
|
}
|
|
|
|
type mutatingTracesSink struct {
|
|
*consumertest.TracesSink
|
|
}
|
|
|
|
func (mts *mutatingTracesSink) Capabilities() consumer.Capabilities {
|
|
return consumer.Capabilities{MutatesData: true}
|
|
}
|