1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2024-12-02 08:52:21 +02:00

Fix TestWithIDGenerator (#4894)

This commit is contained in:
Robert Pająk 2024-02-09 09:01:32 +01:00 committed by GitHub
parent c921815474
commit 15b3f4d776
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1954,25 +1954,29 @@ var _ IDGenerator = (*testIDGenerator)(nil)
func TestWithIDGenerator(t *testing.T) {
const (
startTraceID = 1
startSpanID = 1
numSpan = 10
startSpanID = 10
numSpan = 5
)
gen := &testIDGenerator{traceID: startSpanID, spanID: startSpanID}
for i := 0; i < numSpan; i++ {
gen := &testIDGenerator{traceID: startTraceID, spanID: startSpanID}
te := NewTestExporter()
tp := NewTracerProvider(
WithSyncer(te),
WithIDGenerator(gen),
)
span := startSpan(tp, "TestWithIDGenerator")
got, err := strconv.ParseUint(span.SpanContext().SpanID().String(), 16, 64)
for i := 0; i < numSpan; i++ {
func() {
_, span := tp.Tracer(t.Name()).Start(context.Background(), strconv.Itoa(i))
defer span.End()
gotSpanID, err := strconv.ParseUint(span.SpanContext().SpanID().String(), 16, 64)
require.NoError(t, err)
want := uint64(startSpanID + i)
assert.Equal(t, got, want)
_, err = endSpan(te, span)
assert.Equal(t, uint64(startSpanID+i), gotSpanID)
gotTraceID, err := strconv.ParseUint(span.SpanContext().TraceID().String(), 16, 64)
require.NoError(t, err)
assert.Equal(t, uint64(startTraceID+i), gotTraceID)
}()
}
}