forked from archive/andrewferrier_email2pdf
Add assert for outputFile existence in invokeDirectly.
This commit is contained in:
parent
c7a9403b3a
commit
c8139788ed
2 changed files with 4 additions and 2 deletions
tests
|
@ -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)
|
||||
|
||||
|
|
|
@ -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!")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue