better file removal
This commit is contained in:
parent
b7669c53ff
commit
1b34aa2b0d
1 changed files with 11 additions and 6 deletions
|
@ -1,20 +1,25 @@
|
|||
import unittest
|
||||
import os
|
||||
from md_pdf.consts import TEMPLATES_DIR
|
||||
from md_pdf.consts import TEMPLATES_DIR, STATIC_DIR
|
||||
|
||||
|
||||
class BaseTestCase(unittest.TestCase):
|
||||
def removeFile(self, file):
|
||||
try:
|
||||
os.remove(file)
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
def deleteTemplates(self):
|
||||
for template in [
|
||||
'header.html',
|
||||
'footer.html',
|
||||
'cover.html',
|
||||
'toc-xml',
|
||||
'toc.xsl',
|
||||
]:
|
||||
try:
|
||||
os.remove(os.path.join(TEMPLATES_DIR, template))
|
||||
except OSError:
|
||||
pass
|
||||
self.removeFile(os.path.join(TEMPLATES_DIR, template))
|
||||
|
||||
def tearDown(self):
|
||||
self.deleteTemplates()
|
||||
self.removeFile(os.path.join(STATIC_DIR, 'style.css'))
|
||||
|
||||
|
|
Reference in a new issue