mwalbeck/email2pdf
1
0
Fork 0
forked from archive/andrewferrier_email2pdf
Code Pull requests Activity

Fix tests to handle .jfifs being generated.

Browse source
This commit is contained in:
Andrew Ferrier 2020-09-06 10:11:39 +01:00
parent 49ad58d942
commit 5dbebcbf39
1 changed files with 7 additions and 5 deletions

12
tests/Direct/test_Direct_CID.py
View file

@ -1,6 +1,7 @@
from email.mime.multipart import MIMEMultipart
import os
import glob
from tests.BaseTestClasses import Email2PDFTestCase
@ -261,11 +262,12 @@ class Direct_CID(Email2PDFTestCase):
self.assertEqual('', error)
self.assertTrue(self.existsByTime())
self.assertLess(Email2PDFTestCase.JPG_SIZE, os.path.getsize(self.getTimedFilename()))
self.assertFalse(os.path.exists(os.path.join(self.workingDir, 'myid.jpg')))
self.assertTrue(os.path.exists(os.path.join(self.workingDir, 'myid2.jpg')))
self.assertTrue(os.path.exists(os.path.join(self.workingDir, 'myid3.jpg')))
self.assertTrue(os.path.exists(os.path.join(self.workingDir, 'floating_attachment.jpg')))
self.assertTrue(os.path.exists(os.path.join(self.workingDir, 'floating_attachment_1.jpg')))
# These use globs because they might generate .jpg or they might generate .jfif
self.assertFalse(glob.glob(os.path.join(self.workingDir, 'myid.*')))
self.assertTrue(glob.glob(os.path.join(self.workingDir, 'myid2.*')))
self.assertTrue(glob.glob(os.path.join(self.workingDir, 'myid3.*')))
self.assertTrue(glob.glob(os.path.join(self.workingDir, 'floating_attachment.*')))
self.assertTrue(glob.glob(os.path.join(self.workingDir, 'floating_attachment_1.*')))
self.assertFalse(self.existsByTimeWarning())
self.assertFalse(self.existsByTimeOriginal())