Added pytest and broke up the consumer into file and mail

src/documents/tests/consumers/mail.py

@@ -0,0 +1,43 @@
+import os
+import magic
+
+from hashlib import md5
+
+from django.conf import settings
+from django.test import TestCase
+
+from ...consumers.mail import MailConsumer
+
+
+class TestMailConsumer(TestCase):
+
+    def __init__(self, *args, **kwargs):
+
+        TestCase.__init__(self, *args, **kwargs)
+        self.sample = os.path.join(
+            settings.BASE_DIR,
+            "documents",
+            "tests",
+            "consumers",
+            "samples",
+            "mail.txt"
+        )
+
+    def test_parse(self):
+        consumer = MailConsumer()
+        with open(self.sample) as f:
+
+            messages = consumer._parse_message(f.read())
+
+            self.assertTrue(len(messages), 1)
+            self.assertEqual(messages[0]["subject"], "Test 0")
+
+            attachment = messages[0]["attachment"]
+            data = attachment.read()
+
+            self.assertEqual(
+                md5(data).hexdigest(), "7c89655f9e9eb7dd8cde8568e8115d59")
+
+            self.assertEqual(attachment.content_type, "application/pdf")
+            with magic.Magic(flags=magic.MAGIC_MIME_TYPE) as m:
+                self.assertEqual(m.id_buffer(data), "application/pdf")