opentelemetry-go/attribute/filter_test.go

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

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)
			}
		}
	})
}