From 859440d8bd48f500df6a0de9d38e3f54d127478a Mon Sep 17 00:00:00 2001
From: Daniel Quinn <code@danielquinn.org>
Date: Sun, 17 Apr 2016 14:39:36 +0100
Subject: [PATCH] Merging the duplicate code for tests

---
 src/documents/tests/test_mail.py | 34 ++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/documents/tests/test_mail.py b/src/documents/tests/test_mail.py
index 256c77231..3e78cd23b 100644
--- a/src/documents/tests/test_mail.py
+++ b/src/documents/tests/test_mail.py
@@ -45,6 +45,40 @@ class TestMessage(TestCase):
                 self.assertEqual(m.id_buffer(data), "application/pdf")
 
 
+class TestInlineMessage(TestCase):
+
+    def __init__(self, *args, **kwargs):
+
+        TestCase.__init__(self, *args, **kwargs)
+        self.sample = os.path.join(
+            settings.BASE_DIR,
+            "documents",
+            "tests",
+            "samples",
+            "inline_mail.txt"
+        )
+
+    def test_init(self):
+
+        with open(self.sample, "rb") as f:
+
+            with mock.patch("logging.StreamHandler.emit") as __:
+                message = Message(f.read())
+
+            self.assertTrue(message)
+            self.assertEqual(message.subject, "Paperless Inline Image")
+
+            data = message.attachment.read()
+
+            self.assertEqual(
+                md5(data).hexdigest(), "30c00a7b42913e65f7fdb0be40b9eef3")
+
+            self.assertEqual(
+                message.attachment.content_type, "image/png")
+            with magic.Magic(flags=magic.MAGIC_MIME_TYPE) as m:
+                self.assertEqual(m.id_buffer(data), "image/png")
+
+
 class TestAttachment(TestCase):
 
     def test_init(self):