Fix ids_generator references from moving from api to sdk (#283)

This commit is contained in:
Leighton Chen
2021-01-11 12:40:38 -08:00
committed by GitHub
parent 57b8106edc
commit d12f67fc31
4 changed files with 13 additions and 7 deletions

View File

@ -18,6 +18,7 @@ from unittest.mock import Mock, patch
from opentelemetry import trace as trace_api
from opentelemetry.exporter.datadog import constants, propagator
from opentelemetry.sdk import trace
from opentelemetry.sdk.trace.ids_generator import RandomIdsGenerator
from opentelemetry.trace import get_current_span, set_span_in_context
from opentelemetry.trace.propagation.textmap import DictGetter
@ -29,7 +30,7 @@ carrier_getter = DictGetter()
class TestDatadogFormat(unittest.TestCase):
@classmethod
def setUpClass(cls):
ids_generator = trace_api.RandomIdsGenerator()
ids_generator = RandomIdsGenerator()
cls.serialized_trace_id = propagator.format_trace_id(
ids_generator.generate_trace_id()
)
@ -107,7 +108,7 @@ class TestDatadogFormat(unittest.TestCase):
"child",
trace_api.SpanContext(
parent_span_context.trace_id,
trace_api.RandomIdsGenerator().generate_span_id(),
RandomIdsGenerator().generate_span_id(),
is_remote=False,
trace_flags=parent_span_context.trace_flags,
trace_state=parent_span_context.trace_state,
@ -154,7 +155,7 @@ class TestDatadogFormat(unittest.TestCase):
"child",
trace_api.SpanContext(
parent_span_context.trace_id,
trace_api.RandomIdsGenerator().generate_span_id(),
RandomIdsGenerator().generate_span_id(),
is_remote=False,
trace_flags=parent_span_context.trace_flags,
trace_state=parent_span_context.trace_state,