1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
package wire_test
import (
"testing"
"github.com/opentracing/basictracer-go"
"github.com/opentracing/basictracer-go/wire"
)
func TestProtobufCarrier(t *testing.T) {
var carrier basictracer.DelegatingCarrier = &wire.ProtobufCarrier{}
var traceID, spanID uint64 = 1, 2
sampled := true
baggageKey, expVal := "key1", "val1"
carrier.SetState(traceID, spanID, sampled)
carrier.SetBaggageItem(baggageKey, expVal)
gotTraceID, gotSpanID, gotSampled := carrier.State()
if traceID != gotTraceID || spanID != gotSpanID || sampled != gotSampled {
t.Errorf("Wanted state %d %d %t, got %d %d %t", spanID, traceID, sampled,
gotTraceID, gotSpanID, gotSampled)
}
gotBaggage := map[string]string{}
f := func(k, v string) {
gotBaggage[k] = v
}
carrier.GetBaggage(f)
value, ok := gotBaggage[baggageKey]
if !ok {
t.Errorf("Expected baggage item %s to exist", baggageKey)
}
if value != expVal {
t.Errorf("Expected key %s to be %s, got %s", baggageKey, expVal, value)
}
}
|
