Update tests

Author: DarcyRaynerDD

tests/test_wrapper.py

@@ -89,7 +89,7 @@ def lambda_handler(event, context):
         self.mock_wrapper_lambda_stats.flush.assert_called()
         self.mock_extract_dd_trace_context.assert_called_with(lambda_event)
         self.mock_set_correlation_ids.assert_called()
-        self.mock_inject_correlation_ids.assert_not_called()
+        self.mock_inject_correlation_ids.assert_called()
         self.mock_patch_all.assert_called()
 
     def test_datadog_lambda_wrapper_flush_to_log(self):