From db6f423ca7348e973ec8654f3a95f9c341968d4c Mon Sep 17 00:00:00 2001 From: Wolf-Bastian Poettner Date: Mon, 13 Apr 2020 07:12:48 +0000 Subject: [PATCH] Add: Ability to specifically remove temporary directories that have been used by a unit test --- src/documents/tests/test_file_handling.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/src/documents/tests/test_file_handling.py b/src/documents/tests/test_file_handling.py index 1dbe1ca18..50f3fe485 100644 --- a/src/documents/tests/test_file_handling.py +++ b/src/documents/tests/test_file_handling.py @@ -15,18 +15,14 @@ from django.conf import settings class TestDate(TestCase): + deletion_list = [] + + def add_to_deletion_list(self, dirname): + self.deletion_list.append(dirname) + def tearDown(self): - # Delete all temporary directories from failed tests - root = os.path.normpath("/tmp") - - for filename in os.listdir(root): - fullname = os.path.join(root, filename) - - if not os.path.isdir(fullname): - continue - - if filename.startswith("paperless-tests-"): - shutil.rmtree(fullname, ignore_errors=True) + for dirname in self.deletion_list: + shutil.rmtree(dirname, ignore_errors=True) @override_settings(PAPERLESS_FILENAME_FORMAT="") def test_source_filename(self):