|
18 | 18 | from opentelemetry import trace as trace_api
|
19 | 19 | from opentelemetry.exporter.datadog import constants, propagator
|
20 | 20 | from opentelemetry.sdk import trace
|
21 |
| -from opentelemetry.sdk.trace.ids_generator import RandomIdsGenerator |
| 21 | +from opentelemetry.sdk.trace.id_generator import RandomIdGenerator |
22 | 22 | from opentelemetry.trace import get_current_span, set_span_in_context
|
23 | 23 | from opentelemetry.trace.propagation.textmap import DictGetter
|
24 | 24 |
|
|
30 | 30 | class TestDatadogFormat(unittest.TestCase):
|
31 | 31 | @classmethod
|
32 | 32 | def setUpClass(cls):
|
33 |
| - ids_generator = RandomIdsGenerator() |
| 33 | + id_generator = RandomIdGenerator() |
34 | 34 | cls.serialized_trace_id = propagator.format_trace_id(
|
35 |
| - ids_generator.generate_trace_id() |
| 35 | + id_generator.generate_trace_id() |
36 | 36 | )
|
37 | 37 | cls.serialized_parent_id = propagator.format_span_id(
|
38 |
| - ids_generator.generate_span_id() |
| 38 | + id_generator.generate_span_id() |
39 | 39 | )
|
40 | 40 | cls.serialized_origin = "origin-service"
|
41 | 41 |
|
@@ -108,7 +108,7 @@ def test_context_propagation(self):
|
108 | 108 | "child",
|
109 | 109 | trace_api.SpanContext(
|
110 | 110 | parent_span_context.trace_id,
|
111 |
| - RandomIdsGenerator().generate_span_id(), |
| 111 | + RandomIdGenerator().generate_span_id(), |
112 | 112 | is_remote=False,
|
113 | 113 | trace_flags=parent_span_context.trace_flags,
|
114 | 114 | trace_state=parent_span_context.trace_state,
|
@@ -155,7 +155,7 @@ def test_sampling_priority_auto_reject(self):
|
155 | 155 | "child",
|
156 | 156 | trace_api.SpanContext(
|
157 | 157 | parent_span_context.trace_id,
|
158 |
| - RandomIdsGenerator().generate_span_id(), |
| 158 | + RandomIdGenerator().generate_span_id(), |
159 | 159 | is_remote=False,
|
160 | 160 | trace_flags=parent_span_context.trace_flags,
|
161 | 161 | trace_state=parent_span_context.trace_state,
|
|
0 commit comments