@@ -26,18 +26,18 @@ class TempfileMixin(unittest.TestCase):
2626 temp_directory = None
2727 default_storage = None
2828 _original_media_root = None
29- _original_file_storage = None
29+ _original_storages_default = None
3030
3131 @classmethod
3232 def set_up_temp_directory (cls ):
3333 """
3434 Creates a temp directory and fixes Django settings
3535 """
3636 cls ._original_media_root = settings .MEDIA_ROOT
37- cls ._original_file_storage = settings .DEFAULT_FILE_STORAGE
37+ cls ._original_storages_default = settings .STORAGES . get ( "default" , {}). copy ()
3838 cls .temp_directory = mkdtemp ()
3939 settings .MEDIA_ROOT = cls .temp_directory
40- settings .DEFAULT_FILE_STORAGE = " django.core.files.storage.FileSystemStorage"
40+ settings .STORAGES [ "default" ] = { "BACKEND" : " django.core.files.storage.FileSystemStorage"}
4141 cls .default_storage = default_storage
4242
4343 @classmethod
@@ -47,9 +47,9 @@ def tear_down_temp_directory(cls):
4747 """
4848 shutil .rmtree (cls .temp_directory , ignore_errors = True )
4949 settings .MEDIA_ROOT = cls ._original_media_root
50- settings .DEFAULT_FILE_STORAGE = cls ._original_file_storage
50+ settings .STORAGES [ "default" ] = cls ._original_storages_default
5151 del cls ._original_media_root
52- del cls ._original_file_storage
52+ del cls ._original_storages_default
5353
5454 @contextmanager
5555 def dummy_upload (self , filename , data = b"some information" ):
0 commit comments