From af3f421579cf983da7f91eb7182061a4d2ff4ac3 Mon Sep 17 00:00:00 2001
From: Wolf-Bastian Poettner <bastian@poettner.de>
Date: Mon, 13 Apr 2020 07:13:54 +0000
Subject: [PATCH] Add: make unit test add their temporary directory to the
 deletion list

---
 src/documents/tests/test_file_handling.py | 26 +++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/src/documents/tests/test_file_handling.py b/src/documents/tests/test_file_handling.py
index b8d2b20f0..75ce80905 100644
--- a/src/documents/tests/test_file_handling.py
+++ b/src/documents/tests/test_file_handling.py
@@ -54,6 +54,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}")
     def test_file_renaming(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -100,6 +102,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}")
     def test_document_delete(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -124,6 +128,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}")
     def test_directory_not_empty(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -157,6 +163,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[type]}")
     def test_tags_with_underscore(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -180,6 +188,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[type]}")
     def test_tags_with_dash(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -203,6 +213,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[type]}")
     def test_tags_malformed(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -226,6 +238,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[0]}")
     def test_tags_all(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -248,6 +262,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[0]}")
     def test_tags_out_of_bounds_0(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -266,6 +282,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[10000000]}")
     def test_tags_out_of_bounds_10000000(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -284,6 +302,8 @@ class TestDate(TestCase):
                        format(str(uuid4())[:8]))
     @override_settings(PAPERLESS_FILENAME_FORMAT="{tags[99]}")
     def test_tags_out_of_bounds_99(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -303,6 +323,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}/{correspondent}")
     def test_nested_directory_cleanup(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -345,6 +367,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}")
     def test_document_renamed(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_UNENCRYPTED
@@ -392,6 +416,8 @@ class TestDate(TestCase):
     @override_settings(PAPERLESS_FILENAME_FORMAT="{correspondent}/" +
                        "{correspondent}")
     def test_document_renamed_encrypted(self):
+        self.add_to_deletion_list(settings.MEDIA_ROOT)
+
         document = Document()
         document.file_type = "pdf"
         document.storage_type = Document.STORAGE_TYPE_GPG