Fix failed tests

communityconnectlabs · Jul 25, 2023 · acee342 · acee342
1 parent 1dd8bfa
commit acee342
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/temba/channels/models.py b/temba/channels/models.py
@@ -1285,8 +1285,8 @@ def basic_reduction():
             return result
 
         redacted = basic_reduction()
-        if force_redact_keys and self.msg_id:
-            redacted = redact.http_trace(redacted, self.msg.contact_urn.path, mask, force_redact_keys)
+        if force_redact_keys:
+            redacted = redact.http_trace(redacted, "********", mask, force_redact_keys)
 
         return redacted
 

diff --git a/temba/utils/redact.py b/temba/utils/redact.py
@@ -46,9 +46,10 @@ def replace_headers(raw_data, headers_to_replace):
     try:
         headers = OrderedDict()
         raw_headers, raw_body = raw_data.split(HTTP_BODY_BOUNDARY)
+        split_headers = str(raw_headers).split("\r\n")
 
         # parse headers
-        for line in str(raw_headers).split("\r\n")[1:]:
+        for line in split_headers[1:]:
             key, value = line.split(": ")
             headers[key] = value
 
@@ -58,7 +59,8 @@ def replace_headers(raw_data, headers_to_replace):
                 headers[key] = value
 
         # rebuild http message
-        raw_headers = "\r\n".join([": ".join((key, value)) for key, value in headers.items()])
+        first_header = split_headers[0].strip()
+        raw_headers = "\r\n".join([first_header] + [": ".join((key, value)) for key, value in headers.items()])
         raw_message = HTTP_BODY_BOUNDARY.join((raw_headers, raw_body))
         return raw_message
     except (KeyError, ValueError):