Move MockBackend numSpansReceived addition after ConsumeTraces

Signed-off-by: James Ryans <james.ryans2012@gmail.com>
open-telemetry · Jan 14, 2024 · 43dc5f4 · 43dc5f4
1 parent 6a61430
commit 43dc5f4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/testbed/testbed/mock_backend.go b/testbed/testbed/mock_backend.go
@@ -195,8 +195,6 @@ func (tc *MockTraceConsumer) ConsumeTraces(_ context.Context, td ptrace.Traces)
 		return err
 	}
 
-	tc.numSpansReceived.Add(uint64(td.SpanCount()))
-
 	rs := td.ResourceSpans()
 	for i := 0; i < rs.Len(); i++ {
 		ils := rs.At(i).ScopeSpans()
@@ -226,6 +224,7 @@ func (tc *MockTraceConsumer) ConsumeTraces(_ context.Context, td ptrace.Traces)
 	}
 
 	tc.backend.ConsumeTrace(td)
+	tc.numSpansReceived.Add(uint64(td.SpanCount()))
 
 	return nil
 }