mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Chao WengandGitHub
7d7cac5ce3
<!--Ex. Fixing a bug - Describe the bug and how this fixes the issue. Ex. Adding a feature - Explain what this achieves.--> #### Description Add `component.Type` parameter to `NewNopSettings` And deprecate `NewNopSettingsWithType` cc @mx-psi <!-- Issue number if applicable --> #### Link to tracking issue Relevant to #12305 <!--Describe what testing was performed and which tests were added.--> #### Testing Updated <!--Describe the documentation added.--> #### Documentation Added <!--Please delete paragraphs that you did not use before submitting.-->
100 lines
3.3 KiB
Go
100 lines
3.3 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package builders
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.opentelemetry.io/collector/component"
|
|
"go.opentelemetry.io/collector/component/componenttest"
|
|
"go.opentelemetry.io/collector/extension"
|
|
"go.opentelemetry.io/collector/extension/extensiontest"
|
|
"go.opentelemetry.io/collector/otelcol"
|
|
"go.opentelemetry.io/collector/service/internal/builders"
|
|
)
|
|
|
|
func TestExtensionBuilder(t *testing.T) {
|
|
testType := component.MustNewType("test")
|
|
defaultCfg := struct{}{}
|
|
testID := component.NewID(testType)
|
|
unknownID := component.MustNewID("unknown")
|
|
|
|
factories, err := otelcol.MakeFactoryMap([]extension.Factory{
|
|
extension.NewFactory(
|
|
testType,
|
|
func() component.Config { return &defaultCfg },
|
|
func(_ context.Context, settings extension.Settings, _ component.Config) (extension.Extension, error) {
|
|
return nopExtension{Settings: settings}, nil
|
|
},
|
|
component.StabilityLevelDevelopment),
|
|
}...)
|
|
require.NoError(t, err)
|
|
|
|
cfgs := map[component.ID]component.Config{testID: defaultCfg, unknownID: defaultCfg}
|
|
b := builders.NewExtension(cfgs, factories)
|
|
|
|
e, err := b.Create(context.Background(), createExtensionSettings(testID))
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, e)
|
|
|
|
// Check that the extension has access to the resource attributes.
|
|
nop, ok := e.(nopExtension)
|
|
assert.True(t, ok)
|
|
assert.Equal(t, 0, nop.Settings.Resource.Attributes().Len())
|
|
|
|
missingType, err := b.Create(context.Background(), createExtensionSettings(unknownID))
|
|
require.EqualError(t, err, "extension factory not available for: \"unknown\"")
|
|
assert.Nil(t, missingType)
|
|
|
|
missingCfg, err := b.Create(context.Background(), createExtensionSettings(component.NewIDWithName(testType, "foo")))
|
|
require.EqualError(t, err, "extension \"test/foo\" is not configured")
|
|
assert.Nil(t, missingCfg)
|
|
}
|
|
|
|
func TestExtensionBuilderFactory(t *testing.T) {
|
|
factories, err := otelcol.MakeFactoryMap([]extension.Factory{extension.NewFactory(component.MustNewType("foo"), nil, nil, component.StabilityLevelDevelopment)}...)
|
|
require.NoError(t, err)
|
|
|
|
cfgs := map[component.ID]component.Config{component.MustNewID("foo"): struct{}{}}
|
|
b := builders.NewExtension(cfgs, factories)
|
|
|
|
assert.NotNil(t, b.Factory(component.MustNewID("foo").Type()))
|
|
assert.Nil(t, b.Factory(component.MustNewID("bar").Type()))
|
|
}
|
|
|
|
func TestNewNopExtensionConfigsAndFactories(t *testing.T) {
|
|
configs, factories := builders.NewNopExtensionConfigsAndFactories()
|
|
builder := builders.NewExtension(configs, factories)
|
|
require.NotNil(t, builder)
|
|
|
|
factory := extensiontest.NewNopFactory()
|
|
cfg := factory.CreateDefaultConfig()
|
|
set := extensiontest.NewNopSettings(factory.Type())
|
|
set.ID = component.NewID(builders.NopType)
|
|
|
|
ext, err := factory.Create(context.Background(), set, cfg)
|
|
require.NoError(t, err)
|
|
bExt, err := builder.Create(context.Background(), set)
|
|
require.NoError(t, err)
|
|
assert.IsType(t, ext, bExt)
|
|
}
|
|
|
|
type nopExtension struct {
|
|
component.StartFunc
|
|
component.ShutdownFunc
|
|
extension.Settings
|
|
}
|
|
|
|
func createExtensionSettings(id component.ID) extension.Settings {
|
|
return extension.Settings{
|
|
ID: id,
|
|
TelemetrySettings: componenttest.NewNopTelemetrySettings(),
|
|
BuildInfo: component.NewDefaultBuildInfo(),
|
|
}
|
|
}
|