60 | 60 |
}
|
61 | 61 |
}
|
62 | 62 |
|
63 | |
func TestToGRPCContext(t *testing.T) {
|
64 | |
const (
|
65 | |
hostport = "5.5.5.5:5555"
|
66 | |
serviceName = "foo-service"
|
67 | |
methodName = "foo-method"
|
68 | |
traceID int64 = 12
|
69 | |
spanID int64 = 34
|
70 | |
parentSpanID int64 = 56
|
71 | |
)
|
72 | |
|
73 | |
md := metadata.MD{
|
74 | |
"x-b3-traceid": []string{strconv.FormatInt(traceID, 16)},
|
75 | |
"x-b3-spanid": []string{strconv.FormatInt(spanID, 16)},
|
76 | |
"x-b3-parentspanid": []string{strconv.FormatInt(parentSpanID, 16)},
|
77 | |
}
|
78 | |
|
79 | |
newSpan := zipkin.MakeNewSpanFunc(hostport, serviceName, methodName)
|
80 | |
toContext := zipkin.ToGRPCContext(newSpan, log.NewLogfmtLogger(ioutil.Discard))
|
81 | |
|
82 | |
ctx := toContext(context.Background(), &md)
|
83 | |
val := ctx.Value(zipkin.SpanContextKey)
|
84 | |
if val == nil {
|
85 | |
t.Fatalf("%s returned no value", zipkin.SpanContextKey)
|
86 | |
}
|
87 | |
span, ok := val.(*zipkin.Span)
|
|
63 |
func TestFromContext(t *testing.T) {
|
|
64 |
const (
|
|
65 |
hostport = "5.5.5.5:5555"
|
|
66 |
serviceName = "foo-service"
|
|
67 |
methodName = "foo-method"
|
|
68 |
traceID int64 = 14
|
|
69 |
spanID int64 = 36
|
|
70 |
parentSpanID int64 = 58
|
|
71 |
)
|
|
72 |
|
|
73 |
ctx := context.WithValue(
|
|
74 |
context.Background(),
|
|
75 |
zipkin.SpanContextKey,
|
|
76 |
zipkin.NewSpan(hostport, serviceName, methodName, traceID, spanID, parentSpanID),
|
|
77 |
)
|
|
78 |
|
|
79 |
span, ok := zipkin.FromContext(ctx)
|
88 | 80 |
if !ok {
|
89 | |
t.Fatalf("%s was not a Span object", zipkin.SpanContextKey)
|
90 | |
}
|
91 | |
|
|
81 |
t.Fatalf("expected a context value in %q", zipkin.SpanContextKey)
|
|
82 |
}
|
|
83 |
if span == nil {
|
|
84 |
t.Fatal("expected a Zipkin span object")
|
|
85 |
}
|
92 | 86 |
for want, haveFunc := range map[int64]func() int64{
|
93 | 87 |
traceID: span.TraceID,
|
94 | 88 |
spanID: span.SpanID,
|
|
104 | 98 |
}
|
105 | 99 |
}
|
106 | 100 |
|
|
101 |
func TestToGRPCContext(t *testing.T) {
|
|
102 |
const (
|
|
103 |
hostport = "5.5.5.5:5555"
|
|
104 |
serviceName = "foo-service"
|
|
105 |
methodName = "foo-method"
|
|
106 |
traceID int64 = 12
|
|
107 |
spanID int64 = 34
|
|
108 |
parentSpanID int64 = 56
|
|
109 |
)
|
|
110 |
|
|
111 |
md := metadata.MD{
|
|
112 |
"x-b3-traceid": []string{strconv.FormatInt(traceID, 16)},
|
|
113 |
"x-b3-spanid": []string{strconv.FormatInt(spanID, 16)},
|
|
114 |
"x-b3-parentspanid": []string{strconv.FormatInt(parentSpanID, 16)},
|
|
115 |
}
|
|
116 |
|
|
117 |
newSpan := zipkin.MakeNewSpanFunc(hostport, serviceName, methodName)
|
|
118 |
toContext := zipkin.ToGRPCContext(newSpan, log.NewLogfmtLogger(ioutil.Discard))
|
|
119 |
|
|
120 |
ctx := toContext(context.Background(), &md)
|
|
121 |
val := ctx.Value(zipkin.SpanContextKey)
|
|
122 |
if val == nil {
|
|
123 |
t.Fatalf("%s returned no value", zipkin.SpanContextKey)
|
|
124 |
}
|
|
125 |
span, ok := val.(*zipkin.Span)
|
|
126 |
if !ok {
|
|
127 |
t.Fatalf("%s was not a Span object", zipkin.SpanContextKey)
|
|
128 |
}
|
|
129 |
|
|
130 |
for want, haveFunc := range map[int64]func() int64{
|
|
131 |
traceID: span.TraceID,
|
|
132 |
spanID: span.SpanID,
|
|
133 |
parentSpanID: span.ParentSpanID,
|
|
134 |
} {
|
|
135 |
if have := haveFunc(); want != have {
|
|
136 |
name := runtime.FuncForPC(reflect.ValueOf(haveFunc).Pointer()).Name()
|
|
137 |
name = strings.Split(name, "ยท")[0]
|
|
138 |
toks := strings.Split(name, ".")
|
|
139 |
name = toks[len(toks)-1]
|
|
140 |
t.Errorf("%s: want %d, have %d", name, want, have)
|
|
141 |
}
|
|
142 |
}
|
|
143 |
}
|
|
144 |
|
107 | 145 |
func TestToRequest(t *testing.T) {
|
108 | 146 |
const (
|
109 | 147 |
hostport = "5.5.5.5:5555"
|