|
8 | 8 | from amazon.opentelemetry.distro.always_record_sampler import AlwaysRecordSampler
|
9 | 9 | from amazon.opentelemetry.distro.attribute_propagating_span_processor import AttributePropagatingSpanProcessor
|
10 | 10 | from amazon.opentelemetry.distro.aws_batch_unsampled_span_processor import BatchUnsampledSpanProcessor
|
| 11 | +from amazon.opentelemetry.distro.aws_lambda_span_processor import AwsLambdaSpanProcessor |
11 | 12 | from amazon.opentelemetry.distro.aws_metric_attributes_span_exporter import AwsMetricAttributesSpanExporter
|
12 | 13 | from amazon.opentelemetry.distro.aws_opentelemetry_configurator import (
|
13 | 14 | LAMBDA_SPAN_EXPORT_BATCH_SIZE,
|
@@ -320,10 +321,12 @@ def test_customize_span_processors_lambda(self):
|
320 | 321 | _customize_span_processors(mock_tracer_provider, Resource.get_empty())
|
321 | 322 | self.assertEqual(mock_tracer_provider.add_span_processor.call_count, 3)
|
322 | 323 | first_processor: SpanProcessor = mock_tracer_provider.add_span_processor.call_args_list[0].args[0]
|
323 |
| - self.assertIsInstance(first_processor, AttributePropagatingSpanProcessor) |
| 324 | + self.assertIsInstance(first_processor, AwsLambdaSpanProcessor) |
324 | 325 | second_processor: SpanProcessor = mock_tracer_provider.add_span_processor.call_args_list[1].args[0]
|
325 |
| - self.assertIsInstance(second_processor, BatchUnsampledSpanProcessor) |
326 |
| - self.assertEqual(second_processor.max_export_batch_size, LAMBDA_SPAN_EXPORT_BATCH_SIZE) |
| 326 | + self.assertIsInstance(second_processor, AttributePropagatingSpanProcessor) |
| 327 | + third_processor: SpanProcessor = mock_tracer_provider.add_span_processor.call_args_list[2].args[0] |
| 328 | + self.assertIsInstance(third_processor, BatchUnsampledSpanProcessor) |
| 329 | + self.assertEqual(third_processor.max_export_batch_size, LAMBDA_SPAN_EXPORT_BATCH_SIZE) |
327 | 330 | os.environ.pop("OTEL_AWS_APPLICATION_SIGNALS_ENABLED", None)
|
328 | 331 | os.environ.pop("AWS_LAMBDA_FUNCTION_NAME", None)
|
329 | 332 |
|
|
0 commit comments