1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2024-12-12 10:04:29 +02:00
opentelemetry-go/attribute/filter_test.go
Tyler Yahn 69611bd01a
Switch Stream back to having an AttributeFilter field and add New*Filter functions (#4444)
* Add allow/deny attr filters

* Revert "Replace `Stream.AttributeFilter` with `AllowAttributeKeys` (#4288)"

This reverts commit 1633c74aea.

* Rename new attr filter funcs

Do not include the term "Attribute" in a creation function of the
"attribute" pkg.

* Update the AttributeFilter field documentation

* Add tests for filter creation funcs

* Add change to changelog

* Apply feedback

* Use NewDenyKeysFilter for allow-all and deny-list filters

* Remove links from field docs

These links do not render.

---------

Co-authored-by: Chester Cheung <cheung.zhy.csu@gmail.com>
2023-08-25 07:39:43 -07:00

88 lines
2.4 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 attribute
import "testing"
func TestNewAllowKeysFilter(t *testing.T) {
keys := []string{"zero", "one", "two"}
attrs := []KeyValue{Int(keys[0], 0), Int(keys[1], 1), Int(keys[2], 2)}
t.Run("Empty", func(t *testing.T) {
empty := NewAllowKeysFilter()
for _, kv := range attrs {
if empty(kv) {
t.Errorf("empty NewAllowKeysFilter filter accepted %v", kv)
}
}
})
t.Run("Partial", func(t *testing.T) {
partial := NewAllowKeysFilter(Key(keys[0]), Key(keys[1]))
for _, kv := range attrs[:2] {
if !partial(kv) {
t.Errorf("partial NewAllowKeysFilter filter denied %v", kv)
}
}
if partial(attrs[2]) {
t.Errorf("partial NewAllowKeysFilter filter accepted %v", attrs[2])
}
})
t.Run("Full", func(t *testing.T) {
full := NewAllowKeysFilter(Key(keys[0]), Key(keys[1]), Key(keys[2]))
for _, kv := range attrs {
if !full(kv) {
t.Errorf("full NewAllowKeysFilter filter denied %v", kv)
}
}
})
}
func TestNewDenyKeysFilter(t *testing.T) {
keys := []string{"zero", "one", "two"}
attrs := []KeyValue{Int(keys[0], 0), Int(keys[1], 1), Int(keys[2], 2)}
t.Run("Empty", func(t *testing.T) {
empty := NewDenyKeysFilter()
for _, kv := range attrs {
if !empty(kv) {
t.Errorf("empty NewDenyKeysFilter filter denied %v", kv)
}
}
})
t.Run("Partial", func(t *testing.T) {
partial := NewDenyKeysFilter(Key(keys[0]), Key(keys[1]))
for _, kv := range attrs[:2] {
if partial(kv) {
t.Errorf("partial NewDenyKeysFilter filter accepted %v", kv)
}
}
if !partial(attrs[2]) {
t.Errorf("partial NewDenyKeysFilter filter denied %v", attrs[2])
}
})
t.Run("Full", func(t *testing.T) {
full := NewDenyKeysFilter(Key(keys[0]), Key(keys[1]), Key(keys[2]))
for _, kv := range attrs {
if full(kv) {
t.Errorf("full NewDenyKeysFilter filter accepted %v", kv)
}
}
})
}