better file removal

This commit is contained in:
Jake Howard 2017-05-28 14:11:35 +01:00
parent b7669c53ff
commit 1b34aa2b0d

View file

@ -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'))