108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package spanprocessor
|
|
|
|
import (
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"go.opentelemetry.io/collector/config"
|
|
"go.opentelemetry.io/collector/config/configmodels"
|
|
"go.opentelemetry.io/collector/internal/processor/filterset"
|
|
"go.opentelemetry.io/collector/internal/processor/filterspan"
|
|
)
|
|
|
|
func TestLoadConfig(t *testing.T) {
|
|
factories, err := config.ExampleComponents()
|
|
assert.NoError(t, err)
|
|
|
|
factory := &Factory{}
|
|
factories.Processors[typeStr] = factory
|
|
|
|
config, err := config.LoadConfigFile(t, path.Join(".", "testdata", "config.yaml"), factories)
|
|
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, config)
|
|
|
|
p0 := config.Processors["span/custom"]
|
|
assert.Equal(t, p0, &Config{
|
|
ProcessorSettings: configmodels.ProcessorSettings{
|
|
TypeVal: typeStr,
|
|
NameVal: "span/custom",
|
|
},
|
|
Rename: Name{
|
|
FromAttributes: []string{"db.svc", "operation", "id"},
|
|
Separator: "::",
|
|
},
|
|
})
|
|
|
|
p1 := config.Processors["span/no-separator"]
|
|
assert.Equal(t, p1, &Config{
|
|
ProcessorSettings: configmodels.ProcessorSettings{
|
|
TypeVal: typeStr,
|
|
NameVal: "span/no-separator",
|
|
},
|
|
Rename: Name{
|
|
FromAttributes: []string{"db.svc", "operation", "id"},
|
|
Separator: "",
|
|
},
|
|
})
|
|
|
|
p2 := config.Processors["span/to_attributes"]
|
|
assert.Equal(t, p2, &Config{
|
|
ProcessorSettings: configmodels.ProcessorSettings{
|
|
TypeVal: typeStr,
|
|
NameVal: "span/to_attributes",
|
|
},
|
|
Rename: Name{
|
|
ToAttributes: &ToAttributes{
|
|
Rules: []string{`^\/api\/v1\/document\/(?P<documentId>.*)\/update$`},
|
|
},
|
|
},
|
|
})
|
|
|
|
p3 := config.Processors["span/includeexclude"]
|
|
assert.Equal(t, p3, &Config{
|
|
ProcessorSettings: configmodels.ProcessorSettings{
|
|
TypeVal: typeStr,
|
|
NameVal: "span/includeexclude",
|
|
},
|
|
MatchConfig: filterspan.MatchConfig{
|
|
Include: &filterspan.MatchProperties{
|
|
Config: *createMatchConfig(filterset.Regexp),
|
|
Services: []string{`banks`},
|
|
SpanNames: []string{"^(.*?)/(.*?)$"},
|
|
},
|
|
Exclude: &filterspan.MatchProperties{
|
|
Config: *createMatchConfig(filterset.Strict),
|
|
SpanNames: []string{`donot/change`},
|
|
},
|
|
},
|
|
Rename: Name{
|
|
ToAttributes: &ToAttributes{
|
|
Rules: []string{`(?P<operation_website>.*?)$`},
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func createMatchConfig(matchType filterset.MatchType) *filterset.Config {
|
|
return &filterset.Config{
|
|
MatchType: matchType,
|
|
}
|
|
}
|