Skip to content

Commit 20fc935

Browse files
committed
remove spec mocking
1 parent 41d76f4 commit 20fc935

File tree

2 files changed

+12
-8
lines changed

2 files changed

+12
-8
lines changed

lib/datadog/tracing/correlation.rb

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,15 @@ def identifier_from_digest(digest)
9494
end
9595

9696
def format_trace_id(trace_id)
97-
if Datadog.configuration.tracing.trace_id_128_bit_logging_enabled &&
98-
!Tracing::Utils::TraceId.to_high_order(trace_id).zero?
97+
if Datadog.configuration.tracing.trace_id_128_bit_logging_enabled
98+
format_trace_id_128(trace_id)
99+
else
100+
Tracing::Utils::TraceId.to_low_order(trace_id).to_s
101+
end
102+
end
103+
104+
def format_trace_id_128(trace_id)
105+
if !Tracing::Utils::TraceId.to_high_order(trace_id).zero?
99106
Kernel.format('%032x', trace_id)
100107
else
101108
Tracing::Utils::TraceId.to_low_order(trace_id).to_s

spec/support/tracer_helpers.rb

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
require 'datadog/tracing/tracer'
2+
require 'datadog/tracing/correlation'
23
require 'datadog/tracing/trace_operation'
34
require 'support/faux_writer'
45
require 'datadog/tracing/utils'
@@ -124,13 +125,9 @@ def low_order_trace_id(trace_id)
124125
Datadog::Tracing::Utils::TraceId.to_low_order(trace_id)
125126
end
126127

127-
# Ensures the given trace ID is always formatted using the 128-bit logging format
128-
# by wrapping Tracing::Correlation.format_trace_id(trace_id)
128+
## Wraps call to Datadog::Tracing::Correlation.format_trace_id_128 for better test readability
129129
def trace_id_128_log_injection(trace_id)
130-
RSpec::Mocks.with_temporary_scope do
131-
allow(Datadog.configuration.tracing).to receive(:trace_id_128_bit_logging_enabled).and_return(true)
132-
Datadog::Tracing::Correlation.format_trace_id(trace_id)
133-
end
130+
Datadog::Tracing::Correlation.format_trace_id_128(trace_id)
134131
end
135132

136133
# Wraps call to Tracing::Utils::TraceId.to_high_order and converts to hex

0 commit comments

Comments
 (0)