Clean up after test_ssl.WriteLoadTestCase tests.

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/test_smime.py b/tests/test_smime.py
index d0cdd81..678204b 100644
--- a/tests/test_smime.py
+++ b/tests/test_smime.py
@@ -5,6 +5,7 @@
 Copyright (C) 2006 Open Source Applications Foundation. All Rights Reserved.
 """
 
+import os.path
 try:
     import unittest2 as unittest
 except ImportError:
@@ -63,7 +64,8 @@ class SMIMETestCase(unittest.TestCase):
 
     def test_crlf(self):
         self.assertEqual(SMIME.text_crlf(b'foobar'), b'Content-Type: text/plain\r\n\r\nfoobar')
-        self.assertEqual(SMIME.text_crlf_bio(BIO.MemoryBuffer(b'foobar')).read(), b'Content-Type: text/plain\r\n\r\nfoobar')
+        self.assertEqual(SMIME.text_crlf_bio(
+            BIO.MemoryBuffer(b'foobar')).read(), b'Content-Type: text/plain\r\n\r\nfoobar')
 
     def test_sign(self):
         buf = BIO.MemoryBuffer(self.cleartext)
@@ -171,7 +173,6 @@ class SMIMETestCase(unittest.TestCase):
         v = s.verify(p7, data)
         self.assertEqual(v, self.cleartext)
 
-
     def verify_cb_dummy_method(self, ok, store):
         return verify_cb_dummy_function(ok, store)
 
@@ -337,6 +338,10 @@ class WriteLoadTestCase(unittest.TestCase):
         with BIO.openfile(self.filenameSmime, 'wb') as f:
             self.assertEqual(s.write(f, p7, BIO.MemoryBuffer(b'some text')), 1)
 
+    def tearDown(self):
+        if os.path.exists(self.filename_der):
+            os.unlink(self.filename_der)
+
     def test_load_pkcs7(self):
         self.assertEqual(SMIME.load_pkcs7(self.filename).type(), SMIME.PKCS7_SIGNED)