package zipkin_test
import (
"context"
"testing"
"github.com/openzipkin/zipkin-go"
"github.com/openzipkin/zipkin-go/reporter/recorder"
"github.com/go-kit/kit/endpoint"
zipkinkit "github.com/go-kit/kit/tracing/zipkin"
)
const spanName = "test"
func TestTraceEndpoint(t *testing.T) {
rec := recorder.NewReporter()
tr, _ := zipkin.NewTracer(rec)
mw := zipkinkit.TraceEndpoint(tr, spanName)
mw(endpoint.Nop)(context.Background(), nil)
spans := rec.Flush()
if want, have := 1, len(spans); want != have {
t.Fatalf("incorrect number of spans, wanted %d, got %d", want, have)
}
if want, have := spanName, spans[0].Name; want != have {
t.Fatalf("incorrect span name, wanted %s, got %s", want, have)
}
}