fix based on comments

aws-observability · Jan 30, 2024 · 99ff24c · 99ff24c
1 parent 61d26e8
commit 99ff24c
Showing 1 changed file with 5 additions and 3 deletions.
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
@@ -367,9 +367,11 @@ def attributes_side_effect(key):
 
         self.assertFalse(should_generate_service_metric_attributes(self.span_data_mock))
         self.assertFalse(should_generate_dependency_metric_attributes(self.span_data_mock))
-        self.attributes_mock.get.side_effect = (
-            lambda key: MessagingOperationValues.RECEIVE if key == SpanAttributes.MESSAGING_OPERATION else None
-        )
+        def get_side_effect(key):
+            if key == SpanAttributes.MESSAGING_OPERATION:
+                return MessagingOperationValues.RECEIVE
+            return None
 
+        self.attributes_mock.get.side_effect = get_side_effect
         self.assertTrue(should_generate_service_metric_attributes(self.span_data_mock))
         self.assertTrue(should_generate_dependency_metric_attributes(self.span_data_mock))