Delete templates after test

2017-05-27 16:43:23 +01:00 · 2017-05-27 16:43:23 +01:00 · 8ce8a6f940
commit 8ce8a6f940
parent 6cefbfb20d
1 changed files with 16 additions and 1 deletions
--- a/tests/init.py
+++ b/tests/init.py
@ -1,5 +1,20 @@
 from unittest import TestCase
+import os
+from md_pdf.consts import TEMPLATES_DIR


 class BaseTestCase(TestCase):
+    def deleteTemplates(self):
+        for template in [
+            'header.html',
+            'footer.html',
+            'cover.html',
+            'toc-xml',
+        ]:
+            try:
+                os.remove(os.path.join(TEMPLATES_DIR, template))
+            except OSError:
                pass
+
+    def tearDown(self):
+        self.deleteTemplates()