130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package otelcol
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"gopkg.in/yaml.v3"
|
|
|
|
"go.opentelemetry.io/collector/confmap"
|
|
"go.opentelemetry.io/collector/confmap/provider/fileprovider"
|
|
"go.opentelemetry.io/collector/confmap/provider/yamlprovider"
|
|
)
|
|
|
|
func newConfig(yamlBytes []byte, factories Factories) (*Config, error) {
|
|
var stringMap = map[string]interface{}{}
|
|
err := yaml.Unmarshal(yamlBytes, stringMap)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conf := confmap.NewFromStringMap(stringMap)
|
|
|
|
cfg, err := unmarshal(conf, factories)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Config{
|
|
Receivers: cfg.Receivers.Configs(),
|
|
Processors: cfg.Processors.Configs(),
|
|
Exporters: cfg.Exporters.Configs(),
|
|
Connectors: cfg.Connectors.Configs(),
|
|
Extensions: cfg.Extensions.Configs(),
|
|
Service: cfg.Service,
|
|
}, nil
|
|
}
|
|
|
|
func TestConfigProviderYaml(t *testing.T) {
|
|
yamlBytes, err := os.ReadFile(filepath.Join("testdata", "otelcol-nop.yaml"))
|
|
require.NoError(t, err)
|
|
|
|
uriLocation := "yaml:" + string(yamlBytes)
|
|
provider := yamlprovider.New()
|
|
set := ConfigProviderSettings{
|
|
ResolverSettings: confmap.ResolverSettings{
|
|
URIs: []string{uriLocation},
|
|
Providers: map[string]confmap.Provider{provider.Scheme(): provider},
|
|
},
|
|
}
|
|
|
|
cp, err := NewConfigProvider(set)
|
|
require.NoError(t, err)
|
|
|
|
factories, err := nopFactories()
|
|
require.NoError(t, err)
|
|
|
|
cfg, err := cp.Get(context.Background(), factories)
|
|
require.NoError(t, err)
|
|
|
|
configNop, err := newConfig(yamlBytes, factories)
|
|
require.NoError(t, err)
|
|
|
|
assert.EqualValues(t, configNop, cfg)
|
|
}
|
|
|
|
func TestConfigProviderFile(t *testing.T) {
|
|
uriLocation := "file:" + filepath.Join("testdata", "otelcol-nop.yaml")
|
|
provider := fileprovider.New()
|
|
set := ConfigProviderSettings{
|
|
ResolverSettings: confmap.ResolverSettings{
|
|
URIs: []string{uriLocation},
|
|
Providers: map[string]confmap.Provider{provider.Scheme(): provider},
|
|
},
|
|
}
|
|
|
|
cp, err := NewConfigProvider(set)
|
|
require.NoError(t, err)
|
|
|
|
factories, err := nopFactories()
|
|
require.NoError(t, err)
|
|
|
|
cfg, err := cp.Get(context.Background(), factories)
|
|
require.NoError(t, err)
|
|
|
|
yamlBytes, err := os.ReadFile(filepath.Join("testdata", "otelcol-nop.yaml"))
|
|
require.NoError(t, err)
|
|
|
|
configNop, err := newConfig(yamlBytes, factories)
|
|
require.NoError(t, err)
|
|
|
|
assert.EqualValues(t, configNop, cfg)
|
|
}
|
|
|
|
func TestGetConfmap(t *testing.T) {
|
|
uriLocation := "file:" + filepath.Join("testdata", "otelcol-nop.yaml")
|
|
provider := fileprovider.New()
|
|
set := ConfigProviderSettings{
|
|
ResolverSettings: confmap.ResolverSettings{
|
|
URIs: []string{uriLocation},
|
|
Providers: map[string]confmap.Provider{provider.Scheme(): provider},
|
|
},
|
|
}
|
|
|
|
configBytes, err := os.ReadFile(filepath.Join("testdata", "otelcol-nop.yaml"))
|
|
require.NoError(t, err)
|
|
|
|
yamlMap := map[string]any{}
|
|
err = yaml.Unmarshal(configBytes, yamlMap)
|
|
require.NoError(t, err)
|
|
|
|
cp, err := NewConfigProvider(set)
|
|
require.NoError(t, err)
|
|
|
|
cmp, ok := cp.(ConfmapProvider)
|
|
require.True(t, ok)
|
|
|
|
cmap, err := cmp.GetConfmap(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
assert.EqualValues(t, yamlMap, cmap.ToStringMap())
|
|
}
|