You've already forked opentelemetry-go
mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2025-11-25 22:41:46 +02:00
80cb909774
Based on the Go version we currently use, the dependency already supports 1.24+, which allows using `t.Context()` and `b.Context()` in unit tests and benchmarks respectively. - Enable `context-background` and `context-todo` in [`usetesting`](https://golangci-lint.run/docs/linters/configuration/#usetesting) - Adjust the code to support linter detection --------- Co-authored-by: Tyler Yahn <MrAlias@users.noreply.github.com> Co-authored-by: Tyler Yahn <codingalias@gmail.com> Co-authored-by: Damien Mathieu <42@dmathieu.com>
125 lines
2.7 KiB
Go
125 lines
2.7 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package propagation_test
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"go.opentelemetry.io/otel/propagation"
|
|
)
|
|
|
|
type ctxKeyType uint
|
|
|
|
var ctxKey ctxKeyType
|
|
|
|
type carrier []string
|
|
|
|
func (*carrier) Keys() []string { return nil }
|
|
|
|
func (*carrier) Get(string) string { return "" }
|
|
|
|
func (c *carrier) Set(setter, _ string) {
|
|
*c = append(*c, setter)
|
|
}
|
|
|
|
type propagator struct {
|
|
Name string
|
|
}
|
|
|
|
func (p propagator) Inject(_ context.Context, carrier propagation.TextMapCarrier) {
|
|
carrier.Set(p.Name, "")
|
|
}
|
|
|
|
func (p propagator) Extract(ctx context.Context, _ propagation.TextMapCarrier) context.Context {
|
|
v := ctx.Value(ctxKey)
|
|
if v == nil {
|
|
ctx = context.WithValue(ctx, ctxKey, []string{p.Name})
|
|
} else {
|
|
orig := v.([]string)
|
|
ctx = context.WithValue(ctx, ctxKey, append(orig, p.Name))
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
func (p propagator) Fields() []string { return []string{p.Name} }
|
|
|
|
func TestCompositeTextMapPropagatorFields(t *testing.T) {
|
|
a, b1, b2 := propagator{"a"}, propagator{"b"}, propagator{"b"}
|
|
|
|
want := map[string]struct{}{
|
|
"a": {},
|
|
"b": {},
|
|
}
|
|
got := propagation.NewCompositeTextMapPropagator(a, b1, b2).Fields()
|
|
if len(got) != len(want) {
|
|
t.Fatalf("invalid fields from composite: %v (want %v)", got, want)
|
|
}
|
|
for _, v := range got {
|
|
if _, ok := want[v]; !ok {
|
|
t.Errorf("invalid field returned from composite: %q", v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCompositeTextMapPropagatorInject(t *testing.T) {
|
|
a, b := propagator{"a"}, propagator{"b"}
|
|
|
|
c := make(carrier, 0, 2)
|
|
propagation.NewCompositeTextMapPropagator(a, b).Inject(t.Context(), &c)
|
|
|
|
if got := strings.Join([]string(c), ","); got != "a,b" {
|
|
t.Errorf("invalid inject order: %s", got)
|
|
}
|
|
}
|
|
|
|
func TestCompositeTextMapPropagatorExtract(t *testing.T) {
|
|
a, b := propagator{"a"}, propagator{"b"}
|
|
|
|
ctx := t.Context()
|
|
ctx = propagation.NewCompositeTextMapPropagator(a, b).Extract(ctx, nil)
|
|
|
|
v := ctx.Value(ctxKey)
|
|
if v == nil {
|
|
t.Fatal("no composite extraction")
|
|
}
|
|
if got := strings.Join(v.([]string), ","); got != "a,b" {
|
|
t.Errorf("invalid extract order: %s", got)
|
|
}
|
|
}
|
|
|
|
func TestMapCarrierGet(t *testing.T) {
|
|
carrier := propagation.MapCarrier{
|
|
"foo": "bar",
|
|
"baz": "qux",
|
|
}
|
|
|
|
assert.Equal(t, "bar", carrier.Get("foo"))
|
|
assert.Equal(t, "qux", carrier.Get("baz"))
|
|
}
|
|
|
|
func TestMapCarrierSet(t *testing.T) {
|
|
carrier := make(propagation.MapCarrier)
|
|
carrier.Set("foo", "bar")
|
|
carrier.Set("baz", "qux")
|
|
|
|
assert.Equal(t, "bar", carrier["foo"])
|
|
assert.Equal(t, "qux", carrier["baz"])
|
|
}
|
|
|
|
func TestMapCarrierKeys(t *testing.T) {
|
|
carrier := propagation.MapCarrier{
|
|
"foo": "bar",
|
|
"baz": "qux",
|
|
}
|
|
|
|
keys := carrier.Keys()
|
|
slices.Sort(keys)
|
|
assert.Equal(t, []string{"baz", "foo"}, keys)
|
|
}
|