diff --git a/src/documents/tests/test_mail_inline.py b/src/documents/tests/test_mail_inline.py deleted file mode 100644 index 78b20e5f9..000000000 --- a/src/documents/tests/test_mail_inline.py +++ /dev/null @@ -1,57 +0,0 @@ -import base64 -import os -import magic - -from hashlib import md5 -from unittest import mock - -from django.conf import settings -from django.test import TestCase - -from ..mail import Message, Attachment - - -class TestMessage(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): - data = base64.encodebytes(b"0") - self.assertEqual(Attachment(data, "application/pdf").suffix, "pdf") - self.assertEqual(Attachment(data, "image/png").suffix, "png") - self.assertEqual(Attachment(data, "image/jpeg").suffix, "jpeg") - self.assertEqual(Attachment(data, "image/gif").suffix, "gif") - self.assertEqual(Attachment(data, "image/tiff").suffix, "tiff") - self.assertEqual(Attachment(data, "image/png").read(), data)