add logger check in exception test cases

Author: michael-zhao459

tests/test_tracing.py

@@ -7,7 +7,7 @@
 import os
 import unittest
 
-from unittest.mock import Mock, patch, call
+from unittest.mock import Mock, patch, call, ANY
 
 import ddtrace
 
@@ -2605,8 +2605,9 @@ def test_sqs_no_DSM_context_in_message_attribute(self, mock_dsm_set_checkpoint):
         assert carrier_get("dd-pathway-ctx-base64") is None
 
     @patch("datadog_lambda.tracing._dsm_set_checkpoint")
+    @patch("datadog_lambda.tracing.logger")
     def test_sqs_invalid_datadog_message_attribute_raises_exception(
-        self, mock_dsm_set_checkpoint
+        self, mock_logger, mock_dsm_set_checkpoint
     ):
         event = {
             "Records": [
@@ -2626,6 +2627,11 @@ def test_sqs_invalid_datadog_message_attribute_raises_exception(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
+        # Exception handled by logger
+        mock_logger.debug.assert_any_call(
+            "The trace extractor returned with error %s",
+            ANY,
+        )
         # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
@@ -2829,8 +2835,9 @@ def test_sns_no_DSM_context_in_message_attribute(self, mock_dsm_set_checkpoint):
         assert carrier_get("dd-pathway-ctx-base64") is None
 
     @patch("datadog_lambda.tracing._dsm_set_checkpoint")
+    @patch("datadog_lambda.tracing.logger")
     def test_sns_invalid_datadog_message_attribute_raises_exception(
-        self, mock_dsm_set_checkpoint
+        self, mock_logger, mock_dsm_set_checkpoint
     ):
         event = {
             "Records": [
@@ -2852,6 +2859,11 @@ def test_sns_invalid_datadog_message_attribute_raises_exception(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
+        # Exception handled by logger
+        mock_logger.debug.assert_any_call(
+            "The trace extractor returned with error %s",
+            ANY,
+        )
         # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sns", "arn:aws:sns:us-east-1:123456789012:test-topic"
@@ -3088,8 +3100,9 @@ def test_sns_to_sqs_no_DSM_context_in_message_attribute(
         assert carrier_get("dd-pathway-ctx-base64") is None
 
     @patch("datadog_lambda.tracing._dsm_set_checkpoint")
+    @patch("datadog_lambda.tracing.logger")
     def test_sns_to_sqs_invalid_datadog_message_attribute_raises_exception(
-        self, mock_dsm_set_checkpoint
+        self, mock_logger, mock_dsm_set_checkpoint
     ):
         sns_notification = {
             "Type": "Notification",
@@ -3113,6 +3126,11 @@ def test_sns_to_sqs_invalid_datadog_message_attribute_raises_exception(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
+        # Exception handled by logger
+        mock_logger.debug.assert_any_call(
+            "The trace extractor returned with error %s",
+            ANY,
+        )
         # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
@@ -3247,8 +3265,9 @@ def test_kinesis_empty_message_attribute(self, mock_dsm_set_checkpoint):
         )
 
     @patch("datadog_lambda.tracing._dsm_set_checkpoint")
+    @patch("datadog_lambda.tracing.logger")
     def test_kinesis_invalid_datadog_message_attribute_raises_exception(
-        self, mock_dsm_set_checkpoint
+        self, mock_logger, mock_dsm_set_checkpoint
     ):
         event = {
             "Records": [
@@ -3260,6 +3279,11 @@ def test_kinesis_invalid_datadog_message_attribute_raises_exception(
         }
 
         extract_context_from_kinesis_event(event, self.lambda_context)
+        # Exception handled by logger
+        mock_logger.debug.assert_any_call(
+            "The trace extractor returned with error %s",
+            ANY,
+        )
         # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "kinesis", "arn:aws:kinesis:us-east-1:123456789012:stream/test-stream"