Merge pull request #142 from danielquinn/issue/131

Account for thumbnail deletion as well

src/documents/signals/handlers.py

@@ -85,7 +85,8 @@ def run_post_consume_script(sender, document, **kwargs):
 
 
 def cleanup_document_deletion(sender, instance, using, **kwargs):
-    try:
-        os.unlink(instance.source_path)
-    except FileNotFoundError:
-        pass  # The file's already gone, so we're cool with it.
+    for f in (instance.source_path, instance.thumbnail_path):
+        try:
+            os.unlink(f)
+        except FileNotFoundError:
+            pass  # The file's already gone, so we're cool with it.

src/documents/tests/test_document_model.py

@@ -15,6 +15,9 @@ class TestDocument(TestCase):
             checksum="checksum",
         )
         file_path = document.source_path
+        thumb_path = document.thumbnail_path
         with mock.patch("documents.signals.handlers.os.unlink") as mock_unlink:
             document.delete()
-            mock_unlink.assert_called_with(file_path)
+            mock_unlink.assert_any_call(file_path)
+            mock_unlink.assert_any_call(thumb_path)
+            self.assertEqual(mock_unlink.call_count, 2)