mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
3b50717ab5
We currently don't test profiles in mdatagen code generation. That's not really an issue so far, as we only have components that support other signals as well. But with https://github.com/open-telemetry/opentelemetry-ebpf-profiler/pull/1075, we try to import `consumertest` and never use it. --------- Co-authored-by: Dmitry Anoshin <[email protected]>
119 lines
4.1 KiB
Go
119 lines
4.1 KiB
Go
// Code generated by mdatagen. DO NOT EDIT.
|
|
|
|
package forwardconnector
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/collector/component"
|
|
"go.opentelemetry.io/collector/component/componenttest"
|
|
"go.opentelemetry.io/collector/confmap/confmaptest"
|
|
"go.opentelemetry.io/collector/connector"
|
|
"go.opentelemetry.io/collector/connector/connectortest"
|
|
"go.opentelemetry.io/collector/connector/xconnector"
|
|
"go.opentelemetry.io/collector/consumer"
|
|
"go.opentelemetry.io/collector/consumer/consumertest"
|
|
"go.opentelemetry.io/collector/consumer/xconsumer"
|
|
"go.opentelemetry.io/collector/pipeline"
|
|
"go.opentelemetry.io/collector/pipeline/xpipeline"
|
|
)
|
|
|
|
var typ = component.MustNewType("forward")
|
|
|
|
func TestComponentFactoryType(t *testing.T) {
|
|
require.Equal(t, typ, NewFactory().Type())
|
|
}
|
|
|
|
func TestComponentConfigStruct(t *testing.T) {
|
|
require.NoError(t, componenttest.CheckConfigStruct(NewFactory().CreateDefaultConfig()))
|
|
}
|
|
|
|
func TestComponentLifecycle(t *testing.T) {
|
|
factory := NewFactory()
|
|
|
|
tests := []struct {
|
|
createFn func(ctx context.Context, set connector.Settings, cfg component.Config) (component.Component, error)
|
|
name string
|
|
}{
|
|
|
|
{
|
|
name: "logs_to_logs",
|
|
createFn: func(ctx context.Context, set connector.Settings, cfg component.Config) (component.Component, error) {
|
|
router := connector.NewLogsRouter(map[pipeline.ID]consumer.Logs{pipeline.NewID(pipeline.SignalLogs): consumertest.NewNop()})
|
|
return factory.CreateLogsToLogs(ctx, set, cfg, router)
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "metrics_to_metrics",
|
|
createFn: func(ctx context.Context, set connector.Settings, cfg component.Config) (component.Component, error) {
|
|
router := connector.NewMetricsRouter(map[pipeline.ID]consumer.Metrics{pipeline.NewID(pipeline.SignalMetrics): consumertest.NewNop()})
|
|
return factory.CreateMetricsToMetrics(ctx, set, cfg, router)
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "traces_to_traces",
|
|
createFn: func(ctx context.Context, set connector.Settings, cfg component.Config) (component.Component, error) {
|
|
router := connector.NewTracesRouter(map[pipeline.ID]consumer.Traces{pipeline.NewID(pipeline.SignalTraces): consumertest.NewNop()})
|
|
return factory.CreateTracesToTraces(ctx, set, cfg, router)
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "profiles_to_profiles",
|
|
createFn: func(ctx context.Context, set connector.Settings, cfg component.Config) (component.Component, error) {
|
|
router := xconnector.NewProfilesRouter(map[pipeline.ID]xconsumer.Profiles{pipeline.NewID(xpipeline.SignalProfiles): consumertest.NewNop()})
|
|
return factory.(xconnector.Factory).CreateProfilesToProfiles(ctx, set, cfg, router)
|
|
},
|
|
},
|
|
}
|
|
|
|
cm, err := confmaptest.LoadConf("metadata.yaml")
|
|
require.NoError(t, err)
|
|
cfg := factory.CreateDefaultConfig()
|
|
sub, err := cm.Sub("tests::config")
|
|
require.NoError(t, err)
|
|
require.NoError(t, sub.Unmarshal(&cfg))
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name+"-shutdown", func(t *testing.T) {
|
|
c, err := tt.createFn(context.Background(), connectortest.NewNopSettings(typ), cfg)
|
|
require.NoError(t, err)
|
|
err = c.Shutdown(context.Background())
|
|
require.NoError(t, err)
|
|
})
|
|
t.Run(tt.name+"-lifecycle", func(t *testing.T) {
|
|
firstConnector, err := tt.createFn(context.Background(), connectortest.NewNopSettings(typ), cfg)
|
|
require.NoError(t, err)
|
|
host := newMdatagenNopHost()
|
|
require.NoError(t, err)
|
|
require.NoError(t, firstConnector.Start(context.Background(), host))
|
|
require.NoError(t, firstConnector.Shutdown(context.Background()))
|
|
secondConnector, err := tt.createFn(context.Background(), connectortest.NewNopSettings(typ), cfg)
|
|
require.NoError(t, err)
|
|
require.NoError(t, secondConnector.Start(context.Background(), host))
|
|
require.NoError(t, secondConnector.Shutdown(context.Background()))
|
|
})
|
|
}
|
|
}
|
|
|
|
var _ component.Host = (*mdatagenNopHost)(nil)
|
|
|
|
type mdatagenNopHost struct{}
|
|
|
|
func newMdatagenNopHost() component.Host {
|
|
return &mdatagenNopHost{}
|
|
}
|
|
|
|
func (mnh *mdatagenNopHost) GetExtensions() map[component.ID]component.Component {
|
|
return nil
|
|
}
|
|
|
|
func (mnh *mdatagenNopHost) GetFactory(_ component.Kind, _ component.Type) component.Factory {
|
|
return nil
|
|
}
|