// TestTeeTracerSpanRefs verifies that ChildOf/FollowsFrom relations are // reflected correctly in the underlying spans. func TestTeeTracerSpanRefs(t *testing.T) { r1 := basictracer.NewInMemoryRecorder() t1 := basictracer.NewWithOptions(basictracer.Options{ Recorder: r1, ShouldSample: func(traceID uint64) bool { return true }, // always sample }) r2 := basictracer.NewInMemoryRecorder() t2 := basictracer.NewWithOptions(basictracer.Options{ Recorder: r2, ShouldSample: func(traceID uint64) bool { return true }, // always sample }) tr := NewTeeTracer(t1, t2) root := tr.StartSpan("x") child := tr.StartSpan("x", opentracing.ChildOf(root.Context())) child.Finish() root.Finish() for _, spans := range [][]basictracer.RawSpan{r1.GetSpans(), r2.GetSpans()} { if e, a := 2, len(spans); a != e { t.Fatalf("expected %d, got %d", e, a) } if e, a := spans[0].Context.TraceID, spans[1].Context.TraceID; a != e { t.Errorf("expected %d, got %d", e, a) } if e, a := spans[1].Context.SpanID, spans[0].ParentSpanID; a != e { t.Errorf("expected %d, got %d", e, a) } } }
func TestTeeTracer(t *testing.T) { r1 := basictracer.NewInMemoryRecorder() t1 := basictracer.NewWithOptions(basictracer.Options{ Recorder: r1, ShouldSample: func(traceID uint64) bool { return true }, // always sample }) r2 := basictracer.NewInMemoryRecorder() t2 := basictracer.NewWithOptions(basictracer.Options{ Recorder: r2, ShouldSample: func(traceID uint64) bool { return true }, // always sample }) tr := NewTeeTracer(t1, t2) span := tr.StartSpan("x") span.LogKV("k1", "v1", "k2", "v2") span.SetTag("tag", "value") span.SetBaggageItem("baggage", "baggage-value") if e, a := "baggage-value", span.BaggageItem("baggage"); a != e { t.Errorf("expected %s, got %s", e, a) } spanCtx := span.Context() var ctxBuffer bytes.Buffer if err := tr.Inject(spanCtx, opentracing.Binary, &ctxBuffer); err != nil { t.Fatal(err) } decodedCtx, err := tr.Extract(opentracing.Binary, &ctxBuffer) if err != nil { t.Fatal(err) } span2 := tr.StartSpan("y", opentracing.FollowsFrom(decodedCtx)) span2.LogEvent("event2") if e, a := "baggage-value", span2.BaggageItem("baggage"); a != e { t.Errorf("expected %s, got %s", e, a) } span.Finish() span2.Finish() for _, spans := range [][]basictracer.RawSpan{r1.GetSpans(), r2.GetSpans()} { if e, a := 2, len(spans); a != e { t.Fatalf("expected %d, got %d", e, a) } if e, a := "x", spans[0].Operation; a != e { t.Errorf("expected %s, got %s", e, a) } if e, a := (opentracing.Tags{"tag": "value"}), spans[0].Tags; !reflect.DeepEqual(a, e) { t.Errorf("expected %s, got %s", e, a) } if e, a := "k1:v1", spans[0].Logs[0].Fields[0].String(); a != e { t.Errorf("expected %s, got %s", e, a) } if e, a := "k2:v2", spans[0].Logs[0].Fields[1].String(); a != e { t.Errorf("expected %s, got %s", e, a) } if e, a := 1, len(spans[0].Context.Baggage); a != e { t.Errorf("expected %d, got %d", e, a) } if e, a := "y", spans[1].Operation; a != e { t.Errorf("expected %s, got %s", e, a) } if e, a := opentracing.Tags(nil), spans[1].Tags; !reflect.DeepEqual(a, e) { t.Errorf("expected %s, got %s", e, a) } if e, a := "event:event2", spans[1].Logs[0].Fields[0].String(); a != e { t.Errorf("expected %s, got %s", e, a) } if e, a := 1, len(spans[1].Context.Baggage); a != e { t.Errorf("expected %d, got %d", e, a) } } }