diff --git a/aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_span_processing_util.py b/aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_span_processing_util.py index 81e311c74..dffe49bf9 100644 --- a/aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_span_processing_util.py +++ b/aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_span_processing_util.py @@ -360,20 +360,20 @@ def test_no_metric_attributes_for_aws_sdk_sqs_consumer_process_span(self): self.span_data_mock.kind = SpanKind.CONSUMER self.span_data_mock.name = "Sqs.ReceiveMessage" - def attributes_get_side_effect(key): + def attributes_get_side_effect_process(key): if key == SpanAttributes.MESSAGING_OPERATION: return MessagingOperationValues.PROCESS - self.attributes_mock.get.side_effect = attributes_get_side_effect + self.attributes_mock.get.side_effect = attributes_get_side_effect_process self.span_data_mock.attributes = self.attributes_mock self.assertFalse(should_generate_service_metric_attributes(self.span_data_mock)) self.assertFalse(should_generate_dependency_metric_attributes(self.span_data_mock)) - def attributes_get_side_effect(key): + def attributes_get_side_effect_receive(key): if key == SpanAttributes.MESSAGING_OPERATION: return MessagingOperationValues.RECEIVE - self.attributes_mock.get.side_effect = attributes_get_side_effect + self.attributes_mock.get.side_effect = attributes_get_side_effect_receive self.assertTrue(should_generate_service_metric_attributes(self.span_data_mock)) self.assertTrue(should_generate_dependency_metric_attributes(self.span_data_mock))