fix(appsec): perform _after extractions on the blocking_response

Author: florentinl

datadog_lambda/wrapper.py

@@ -177,6 +177,7 @@ def __call__(self, event, context, **kwargs):
             return self.response
         except BlockingException:
             self.blocking_response = get_asm_blocked_response(self.event_source)
+            return self.blocking_response
         except Exception:
             from datadog_lambda.metric import submit_errors_metric
 
@@ -187,8 +188,6 @@ def __call__(self, event, context, **kwargs):
             raise
         finally:
             self._after(event, context)
-            if self.blocking_response:
-                return self.blocking_response
 
     def _inject_authorizer_span_headers(self, request_id):
         reference_span = self.inferred_span if self.inferred_span else self.span
@@ -283,6 +282,8 @@ def _before(self, event, context):
 
     def _after(self, event, context):
         try:
+            if self.blocking_response:
+                self.response = self.blocking_response
             status_code = extract_http_status_code_tag(self.trigger_tags, self.response)
             if status_code:
                 self.trigger_tags["http.status_code"] = status_code