forked from archive/andrewferrier_email2pdf
128 lines
5 KiB
Python
128 lines
5 KiB
Python
from email.message import Message
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
from tests.BaseTestClasses import Email2PDFTestCase
|
|
|
|
|
|
class TestBasic(Email2PDFTestCase):
|
|
def setUp(self):
|
|
super(TestBasic, self).setUp()
|
|
self.msg = Message()
|
|
|
|
def test_help(self):
|
|
(rc, output, error) = self.invokeAsSubprocess(extraParams=['--help'], expectOutput=True)
|
|
self.assertEqual(0, rc)
|
|
self.assertRegex(output, 'usage:')
|
|
self.assertEqual(error, '')
|
|
|
|
def test_invalid_option(self):
|
|
(rc, output, error) = self.invokeAsSubprocess(extraParams=['--invalid-option'])
|
|
self.assertEqual(2, rc)
|
|
self.assertRegex(error, 'ERROR: unrecognized.*')
|
|
|
|
def test_dontPrintBody(self):
|
|
(rc, output, error) = self.invokeAsSubprocess(extraParams=['--no-body'])
|
|
self.assertEqual(1, rc)
|
|
self.assertFalse(self.existsByTime())
|
|
self.assertRegex(error, "body.*or.*attachments")
|
|
|
|
def test_noheaders(self):
|
|
(rc, output, error) = self.invokeAsSubprocess()
|
|
self.assertEqual(0, rc)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertEqual('', error)
|
|
|
|
def test_simple(self):
|
|
self.addHeaders()
|
|
(rc, output, error) = self.invokeAsSubprocess()
|
|
self.assertEqual(0, rc)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertEqual('', error)
|
|
|
|
def test_withinputfile(self):
|
|
self.addHeaders()
|
|
(rc, output, error) = self.invokeAsSubprocess(inputFile=True)
|
|
self.assertEqual(0, rc)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertEqual('', error)
|
|
|
|
def test_nosubject(self):
|
|
self.addHeaders(Email2PDFTestCase.DEFAULT_FROM, Email2PDFTestCase.DEFAULT_TO, None)
|
|
(rc, output, error) = self.invokeAsSubprocess()
|
|
self.assertEqual(0, rc)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertEqual('', error)
|
|
|
|
def test_plaincontent(self):
|
|
self.addHeaders()
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess()
|
|
self.assertEqual(0, rc)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertEqual('', error)
|
|
self.assertRegex(self.getPDFText(self.getTimedFilename()), "Hello!")
|
|
|
|
def test_plaincontent_poundsign_iso88591(self):
|
|
self.addHeaders()
|
|
path = os.path.join(self.examineDir, "plaincontent_poundsign_iso88591.pdf")
|
|
self.setPlainContent("Hello - this email costs \xa35!", charset="ISO-8859-1")
|
|
(rc, output, error) = self.invokeAsSubprocess(outputFile=path)
|
|
self.assertEqual(0, rc)
|
|
self.assertEqual('', error)
|
|
self.assertTrue(os.path.exists(path))
|
|
self.assertRegex(self.getPDFText(path), "Hello - this email costs \xa35!")
|
|
|
|
def test_plaincontent_notrailingslash(self):
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess(outputDirectory="/tmp")
|
|
self.assertEqual(0, rc)
|
|
self.assertEqual('', error)
|
|
self.assertTrue(self.existsByTime("/tmp"))
|
|
self.assertRegex(self.getPDFText(self.getTimedFilename("/tmp/")), "Hello!")
|
|
|
|
def test_plaincontent_trailingslash(self):
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess(outputDirectory="/tmp")
|
|
self.assertEqual(0, rc)
|
|
self.assertEqual('', error)
|
|
self.assertTrue(self.existsByTime("/tmp/"))
|
|
self.assertRegex(self.getPDFText(self.getTimedFilename("/tmp/")), "Hello!")
|
|
|
|
def test_plaincontent_outputfileoverrides(self):
|
|
filename = os.path.join(self.examineDir, "outputFileOverrides.pdf")
|
|
pathname = tempfile.mkdtemp(dir='/tmp')
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess(outputDirectory=pathname, outputFile=filename)
|
|
self.assertEqual(0, rc)
|
|
self.assertEqual('', error)
|
|
self.assertFalse(self.existsByTime(pathname))
|
|
self.assertTrue(os.path.exists(filename))
|
|
self.assertRegex(self.getPDFText(filename), "Hello!")
|
|
|
|
def test_plaincontent_fileexist(self):
|
|
self.setPlainContent("Hello!")
|
|
unused_f_handle, f_path = tempfile.mkstemp()
|
|
try:
|
|
(rc, output, error) = self.invokeAsSubprocess(outputFile=f_path)
|
|
self.assertEqual(2, rc)
|
|
self.assertRegex(error, "file.*exist")
|
|
finally:
|
|
os.unlink(f_path)
|
|
|
|
def test_verbose(self):
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess(extraParams=['-v'])
|
|
self.assertEqual(0, rc)
|
|
self.assertNotEqual('', error)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertRegex(self.getPDFText(self.getTimedFilename()), "Hello!")
|
|
|
|
def test_veryverbose(self):
|
|
self.setPlainContent("Hello!")
|
|
(rc, output, error) = self.invokeAsSubprocess(extraParams=['-vv'])
|
|
self.assertEqual(0, rc)
|
|
self.assertNotEqual('', error)
|
|
self.assertTrue(self.existsByTime())
|
|
self.assertRegex(self.getPDFText(self.getTimedFilename()), "Hello!")
|