54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package opentracing
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/opentracing/opentracing-go/mocktracer"
|
|
"go.unistack.org/micro/v3/metadata"
|
|
"go.unistack.org/micro/v3/tracer"
|
|
)
|
|
|
|
func TestTraceID(t *testing.T) {
|
|
md := metadata.New(1)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
ctx = metadata.NewIncomingContext(ctx, md)
|
|
|
|
tr := NewTracer(Tracer(mocktracer.New()))
|
|
if err := tr.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var sp tracer.Span
|
|
|
|
ctx, sp = tr.Start(ctx, "test")
|
|
if v := sp.TraceID(); v != "43" {
|
|
t.Fatalf("invalid span trace id %#+v", v)
|
|
}
|
|
if v := sp.SpanID(); v != "44" {
|
|
t.Fatalf("invalid span span id %#+v", v)
|
|
}
|
|
}
|
|
|
|
func TestTraceTags(t *testing.T) {
|
|
md := metadata.New(1)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
ctx = metadata.NewIncomingContext(ctx, md)
|
|
|
|
mtr := mocktracer.New()
|
|
tr := NewTracer(Tracer(mtr))
|
|
if err := tr.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var sp tracer.Span
|
|
|
|
ctx, sp = tr.Start(ctx, "test", tracer.WithSpanLabels("key", "val", "odd"))
|
|
sp.Finish()
|
|
|
|
msp := mtr.FinishedSpans()[0]
|
|
t.Logf("mock span %#+v", msp.Tags())
|
|
}
|