Add assert for outputFile existence in invokeDirectly.

This commit is contained in:
Andrew Ferrier 2015-03-17 00:03:26 +00:00
parent c7a9403b3a
commit c8139788ed
2 changed files with 4 additions and 2 deletions

View file

@ -161,7 +161,7 @@ class Email2PDFTestCase(unittest.TestCase):
return (p.returncode, output, error)
def invokeDirectly(self, outputDirectory=None, outputFile=None, extraParams=[], completeMessage=None):
def invokeDirectly(self, outputDirectory=None, outputFile=None, extraParams=[], completeMessage=None, okToExist=False):
import importlib.machinery
module_path = self.getOriginalScriptPath()
loader = importlib.machinery.SourceFileLoader('email2pdf', module_path)
@ -184,6 +184,8 @@ class Email2PDFTestCase(unittest.TestCase):
if outputFile:
options.extend(['-o', outputFile])
if not okToExist:
assert(not os.path.exists(outputFile))
options.extend(extraParams)

View file

@ -15,7 +15,7 @@ class Direct_Errors(BaseTestClasses.Email2PDFTestCase):
self.attachText("Hello!")
with tempfile.NamedTemporaryFile() as tmpfile:
with self.assertRaisesRegex(Exception, "file.*exist"):
self.invokeDirectly(outputFile=tmpfile.name)
self.invokeDirectly(outputFile=tmpfile.name, okToExist=True)
def test_plaincontent_dirnotexist(self):
self.attachText("Hello!")