mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-17 10:13:56 -05:00
removing duplicate code
This commit is contained in:
parent
859440d8bd
commit
5a07c72f1e
@ -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)
|
|
Loading…
x
Reference in New Issue
Block a user