mirror of
				https://github.com/paperless-ngx/paperless-ngx.git
				synced 2025-11-02 18:47:10 -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