Use storages['DEFAULT'] instead of the depreacted get_storage_class

vintasoftware · Sep 21, 2023 · eb732ae · eb732ae
1 parent b687cb0
commit eb732ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/backends/test_vanilla_django_backend.py b/tests/backends/test_vanilla_django_backend.py
@@ -5,7 +5,7 @@
 
 from django.test import TestCase, override_settings
 from django.core.mail import EmailMessage, EmailMultiAlternatives
-from django.core.files.storage import get_storage_class
+from django.core.files.storage import storages
 from django.template import TemplateDoesNotExist
 from django.core import mail
 
@@ -69,7 +69,7 @@ class TemplateBackendTestCase(MockedNetworkTestCaseMixin,
     template_backend_klass = TemplateBackend
 
     def setUp(self):
-        self.storage_mock = Mock(wraps=get_storage_class())()
+        self.storage_mock = Mock(wraps=storages["default"])()
         self.storage_mock.save = Mock()
         self.backend = self.template_backend_klass()
         self.context = {'username': 'vintasoftware',