Explicitly remove test directories in FTP tests

This commit is contained in:
Paul Tremberth 2017-02-20 14:50:05 +01:00
parent d35a01a103
commit f2e200120b
1 changed files with 7 additions and 0 deletions

View File

@ -5,6 +5,7 @@ try:
from unittest import mock
except ImportError:
import mock
import shutil
from twisted.trial import unittest
from twisted.protocols.policies import WrappingFactory
@ -710,6 +711,9 @@ class BaseFTPTestCase(unittest.TestCase):
self.download_handler = FTPDownloadHandler(Settings())
self.addCleanup(self.port.stopListening)
def tearDown(self):
shutil.rmtree(self.directory)
def _add_test_callbacks(self, deferred, callback=None, errback=None):
def _clean(data):
self.download_handler.client.transport.loseConnection()
@ -817,3 +821,6 @@ class AnonymousFTPTestCase(BaseFTPTestCase):
self.portNum = self.port.getHost().port
self.download_handler = FTPDownloadHandler(Settings())
self.addCleanup(self.port.stopListening)
def tearDown(self):
shutil.rmtree(self.directory)