Refs #54, #55 -- check payload type before trimming

voiio · Sep 25, 2023 · 0d166ce · 0d166ce
1 parent 3689edf
commit 0d166ce
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/emark/backends.py b/emark/backends.py
@@ -72,8 +72,9 @@ class ConsoleEmailBackendMixin:
 
     def write_message(self, message):
         msg = message.message()
-        payload_count = len(msg.get_payload())
-        if payload_count > 1:
+        original_payload = msg.get_payload()
+        payload_count = len(original_payload)
+        if isinstance(original_payload, list):
             msg.set_payload(msg.get_payload(0))
         msg_data = msg.as_bytes()
         charset = (

diff --git a/tests/test_backends.py b/tests/test_backends.py
@@ -11,6 +11,13 @@
 
 class TestConsoleEmailBackend:
     def test_write_message(self):
+        msg = EmailMessage(body="foo\nbar")
+        with io.StringIO() as stream:
+            backends.ConsoleEmailBackend(stream=stream).write_message(msg)
+            stdout = stream.getvalue()
+            assert "foo\nbar" in stdout
+
+    def test_write_message__multi_alternatives(self):
         msg = EmailMultiAlternatives(body="foo")
         msg.attach_alternative("<html></html>", "text/html")
         with io.StringIO() as stream: