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

Use more elegant way of handling temporary files and directories.

Browse source
This commit is contained in:
Andrew Ferrier 2015-01-12 21:00:46 +00:00
parent 2047dbe7b1
commit a1839255b0
3 changed files with 25 additions and 29 deletions

23
tests/test_Subprocess_Basic.py
View file

@ -101,24 +101,21 @@ class TestBasic(Email2PDFTestCase):
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!")
with tempfile.TemporaryDirectory() as pathname:
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)
with tempfile.NamedTemporaryFile() as tmpfile:
(rc, output, error) = self.invokeAsSubprocess(outputFile=tmpfile.name)
self.assertEqual(2, rc)
self.assertRegex(error, "file.*exist")
finally:
os.unlink(f_path)
def test_verbose(self):
self.setPlainContent("Hello!")