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]>
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
// Code generated by mdatagen. DO NOT EDIT.
|
|
|
|
package otlpreceiver
|
|
|
|
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/consumer/consumertest"
|
|
"go.opentelemetry.io/collector/receiver"
|
|
"go.opentelemetry.io/collector/receiver/receivertest"
|
|
"go.opentelemetry.io/collector/receiver/xreceiver"
|
|
)
|
|
|
|
var typ = component.MustNewType("otlp")
|
|
|
|
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 receiver.Settings, cfg component.Config) (component.Component, error)
|
|
name string
|
|
}{
|
|
|
|
{
|
|
name: "logs",
|
|
createFn: func(ctx context.Context, set receiver.Settings, cfg component.Config) (component.Component, error) {
|
|
return factory.CreateLogs(ctx, set, cfg, consumertest.NewNop())
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "metrics",
|
|
createFn: func(ctx context.Context, set receiver.Settings, cfg component.Config) (component.Component, error) {
|
|
return factory.CreateMetrics(ctx, set, cfg, consumertest.NewNop())
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "traces",
|
|
createFn: func(ctx context.Context, set receiver.Settings, cfg component.Config) (component.Component, error) {
|
|
return factory.CreateTraces(ctx, set, cfg, consumertest.NewNop())
|
|
},
|
|
},
|
|
|
|
{
|
|
name: "profiles",
|
|
createFn: func(ctx context.Context, set receiver.Settings, cfg component.Config) (component.Component, error) {
|
|
return factory.(xreceiver.Factory).CreateProfiles(ctx, set, cfg, consumertest.NewNop())
|
|
},
|
|
},
|
|
}
|
|
|
|
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(), receivertest.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) {
|
|
firstRcvr, err := tt.createFn(context.Background(), receivertest.NewNopSettings(typ), cfg)
|
|
require.NoError(t, err)
|
|
host := newMdatagenNopHost()
|
|
require.NoError(t, err)
|
|
require.NoError(t, firstRcvr.Start(context.Background(), host))
|
|
require.NoError(t, firstRcvr.Shutdown(context.Background()))
|
|
secondRcvr, err := tt.createFn(context.Background(), receivertest.NewNopSettings(typ), cfg)
|
|
require.NoError(t, err)
|
|
require.NoError(t, secondRcvr.Start(context.Background(), host))
|
|
require.NoError(t, secondRcvr.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
|
|
}
|