mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
a57a446995
<!--Ex. Fixing a bug - Describe the bug and how this fixes the issue. Ex. Adding a feature - Explain what this achieves.--> #### Description Adds support for Profiles to Profies in the forward connector. <!--Describe what testing was performed and which tests were added.--> #### Testing Produce -> consume unit test <!--Describe the documentation added.--> #### Documentation <!--Please delete paragraphs that you did not use before submitting.--> --------- Co-authored-by: Alex Boten <[email protected]>
76 lines
2.6 KiB
Go
76 lines
2.6 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package forwardconnector
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/collector/component/componenttest"
|
|
"go.opentelemetry.io/collector/connector/connectortest"
|
|
"go.opentelemetry.io/collector/consumer/consumertest"
|
|
"go.opentelemetry.io/collector/pdata/plog"
|
|
"go.opentelemetry.io/collector/pdata/pmetric"
|
|
"go.opentelemetry.io/collector/pdata/pprofile"
|
|
"go.opentelemetry.io/collector/pdata/ptrace"
|
|
)
|
|
|
|
func TestForward(t *testing.T) {
|
|
f := NewFactory()
|
|
cfg := f.CreateDefaultConfig()
|
|
assert.Equal(t, &Config{}, cfg)
|
|
|
|
ctx := context.Background()
|
|
set := connectortest.NewNopSettings(f.Type())
|
|
host := componenttest.NewNopHost()
|
|
|
|
tracesSink := new(consumertest.TracesSink)
|
|
tracesToTraces, err := f.CreateTracesToTraces(ctx, set, cfg, tracesSink)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, tracesToTraces)
|
|
|
|
metricsSink := new(consumertest.MetricsSink)
|
|
metricsToMetrics, err := f.CreateMetricsToMetrics(ctx, set, cfg, metricsSink)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, metricsToMetrics)
|
|
|
|
logsSink := new(consumertest.LogsSink)
|
|
logsToLogs, err := f.CreateLogsToLogs(ctx, set, cfg, logsSink)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, logsToLogs)
|
|
|
|
profilesSink := new(consumertest.ProfilesSink)
|
|
profilesToProfiles, err := f.CreateProfilesToProfiles(ctx, set, cfg, profilesSink)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, profilesToProfiles)
|
|
|
|
assert.NoError(t, tracesToTraces.Start(ctx, host))
|
|
assert.NoError(t, metricsToMetrics.Start(ctx, host))
|
|
assert.NoError(t, logsToLogs.Start(ctx, host))
|
|
assert.NoError(t, profilesToProfiles.Start(ctx, host))
|
|
|
|
assert.NoError(t, tracesToTraces.ConsumeTraces(ctx, ptrace.NewTraces()))
|
|
|
|
assert.NoError(t, metricsToMetrics.ConsumeMetrics(ctx, pmetric.NewMetrics()))
|
|
assert.NoError(t, metricsToMetrics.ConsumeMetrics(ctx, pmetric.NewMetrics()))
|
|
|
|
assert.NoError(t, logsToLogs.ConsumeLogs(ctx, plog.NewLogs()))
|
|
assert.NoError(t, logsToLogs.ConsumeLogs(ctx, plog.NewLogs()))
|
|
assert.NoError(t, logsToLogs.ConsumeLogs(ctx, plog.NewLogs()))
|
|
|
|
assert.NoError(t, profilesToProfiles.ConsumeProfiles(ctx, pprofile.NewProfiles()))
|
|
|
|
assert.NoError(t, tracesToTraces.Shutdown(ctx))
|
|
assert.NoError(t, metricsToMetrics.Shutdown(ctx))
|
|
assert.NoError(t, logsToLogs.Shutdown(ctx))
|
|
assert.NoError(t, profilesToProfiles.Shutdown(ctx))
|
|
|
|
assert.Len(t, tracesSink.AllTraces(), 1)
|
|
assert.Len(t, metricsSink.AllMetrics(), 2)
|
|
assert.Len(t, logsSink.AllLogs(), 3)
|
|
assert.Len(t, profilesSink.AllProfiles(), 1)
|
|
}
|