Переглянути джерело

test: add missing helper function

Mikael Koli 3 роки тому
батько
коміт
5b19363f34
1 змінених файлів з 6 додано та 0 видалено
  1. 6 0
      redmail/test/helpers/convert.py

+ 6 - 0
redmail/test/helpers/convert.py

@@ -10,6 +10,9 @@ def remove_email_extra(s:str):
     s = remove_extra_lines(s)
     return s.replace("=20", "").replace('"3D', "").replace("=\n", "")
 
+def remove_email_content_id(s:str, repl="<ID>"):
+    return re.sub(r"(?<================)[0-9]+(?===)", repl, s)
+
 def payloads_to_dict(*parts):
     data = {}
     for part in parts:
@@ -25,6 +28,9 @@ def payloads_to_dict(*parts):
             key = new_key
         if isinstance(payload, str):
             data[key] = payload
+        elif payload is None:
+            # Most likely empty message
+            pass
         else:
             data[key] = payloads_to_dict(*payload)
     return data