@@ -68,7 +68,9 @@ def test_get_trace_data_with_span_and_trace():
6868 parent_context = {}
6969
7070 span_processor = SentrySpanProcessor ()
71- sentry_trace_data = span_processor ._get_trace_data (otel_span , parent_context )
71+ sentry_trace_data = span_processor ._get_trace_data (
72+ otel_span .get_span_context (), otel_span .parent , parent_context
73+ )
7274 assert sentry_trace_data ["trace_id" ] == "1234567890abcdef1234567890abcdef"
7375 assert sentry_trace_data ["span_id" ] == "1234567890abcdef"
7476 assert sentry_trace_data ["parent_span_id" ] is None
@@ -90,7 +92,9 @@ def test_get_trace_data_with_span_and_trace_and_parent():
9092 parent_context = {}
9193
9294 span_processor = SentrySpanProcessor ()
93- sentry_trace_data = span_processor ._get_trace_data (otel_span , parent_context )
95+ sentry_trace_data = span_processor ._get_trace_data (
96+ otel_span .get_span_context (), otel_span .parent , parent_context
97+ )
9498 assert sentry_trace_data ["trace_id" ] == "1234567890abcdef1234567890abcdef"
9599 assert sentry_trace_data ["span_id" ] == "1234567890abcdef"
96100 assert sentry_trace_data ["parent_span_id" ] == "abcdef1234567890"
@@ -121,7 +125,9 @@ def test_get_trace_data_with_sentry_trace():
121125 ],
122126 ):
123127 span_processor = SentrySpanProcessor ()
124- sentry_trace_data = span_processor ._get_trace_data (otel_span , parent_context )
128+ sentry_trace_data = span_processor ._get_trace_data (
129+ otel_span .get_span_context (), otel_span .parent , parent_context
130+ )
125131 assert sentry_trace_data ["trace_id" ] == "1234567890abcdef1234567890abcdef"
126132 assert sentry_trace_data ["span_id" ] == "1234567890abcdef"
127133 assert sentry_trace_data ["parent_span_id" ] == "abcdef1234567890"
@@ -138,7 +144,9 @@ def test_get_trace_data_with_sentry_trace():
138144 ],
139145 ):
140146 span_processor = SentrySpanProcessor ()
141- sentry_trace_data = span_processor ._get_trace_data (otel_span , parent_context )
147+ sentry_trace_data = span_processor ._get_trace_data (
148+ otel_span .get_span_context (), otel_span .parent , parent_context
149+ )
142150 assert sentry_trace_data ["trace_id" ] == "1234567890abcdef1234567890abcdef"
143151 assert sentry_trace_data ["span_id" ] == "1234567890abcdef"
144152 assert sentry_trace_data ["parent_span_id" ] == "abcdef1234567890"
@@ -175,7 +183,9 @@ def test_get_trace_data_with_sentry_trace_and_baggage():
175183 ],
176184 ):
177185 span_processor = SentrySpanProcessor ()
178- sentry_trace_data = span_processor ._get_trace_data (otel_span , parent_context )
186+ sentry_trace_data = span_processor ._get_trace_data (
187+ otel_span .get_span_context (), otel_span .parent , parent_context
188+ )
179189 assert sentry_trace_data ["trace_id" ] == "1234567890abcdef1234567890abcdef"
180190 assert sentry_trace_data ["span_id" ] == "1234567890abcdef"
181191 assert sentry_trace_data ["parent_span_id" ] == "abcdef1234567890"
0 commit comments