fix tests

SmileyChris · Jul 27, 2024 · d455b42 · d455b42
1 parent da9382a
commit d455b42
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/easy_thumbnails/tests/settings.py b/easy_thumbnails/tests/settings.py
@@ -31,7 +31,7 @@
 
 STORAGES = {
     "easy_thumbnails": {
-        "BACKEND": "django.core.files.storage.FileSystemStorage",
+        "BACKEND": "easy_thumbnails.tests.utils.TemporaryStorage",
     },
     "default": {
         "BACKEND": "django.core.files.storage.FileSystemStorage",

diff --git a/easy_thumbnails/tests/test_thumbnail_cleanup.py b/easy_thumbnails/tests/test_thumbnail_cleanup.py
@@ -110,3 +110,18 @@ def test_cleanup_last_n_days(self):
         # Verify the source reference has been deleted
         with self.assertRaises(Source.DoesNotExist):
             Source.objects.get(id=self.source.id)
+
+    def test_source_storage_hash_not_found(self):
+        self.assertTrue(os.path.exists(self.source_image_path))
+        self.assertTrue(os.path.exists(self.thumbnail_path))
+
+        # Change the source's storage_hash to simulate an unknown storage hash
+        self.source.storage_hash = "unknown_storage_hash"
+        self.source.save()
+
+        # Run the thumbnail cleanup command
+        call_command("thumbnail_cleanup", verbosity=2)
+
+        # Verify the thumbnail and source still exist
+        self.assertTrue(os.path.exists(self.thumbnail_path))
+        self.assertIsNotNone(Source.objects.get(id=self.source.id))