Skip to content

Commit 41576c1

Browse files
committed
Use default_storage directly
1 parent 511773a commit 41576c1

4 files changed

+5
-6
lines changed

tests/backends/test_vanilla_django_backend.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,12 @@
55

66
from django.test import TestCase, override_settings
77
from django.core.mail import EmailMessage, EmailMultiAlternatives
8-
from django.core.files.storage import storages
8+
from django.core.files.storage import default_storage
99
from django.template import TemplateDoesNotExist
1010
from django.core import mail
1111

1212
import pytest
13-
from mock import patch, Mock
13+
from unittest.mock import patch, Mock
1414
from anymail.message import AnymailMessage
1515

1616
from templated_email.backends.vanilla_django import TemplateBackend, EmailRenderException
@@ -69,7 +69,7 @@ class TemplateBackendTestCase(MockedNetworkTestCaseMixin,
6969
template_backend_klass = TemplateBackend
7070

7171
def setUp(self):
72-
self.storage_mock = Mock(wraps=storages["default"])()
72+
self.storage_mock = Mock(wraps=default_storage)()
7373
self.storage_mock.save = Mock()
7474
self.backend = self.template_backend_klass()
7575
self.context = {'username': 'vintasoftware',

tests/test_get_templated_mail.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from django.test import TestCase
22

3-
from mock import patch
3+
from unittest.mock import patch
44

55
from templated_email import get_templated_mail
66

tests/test_send_templated_mail.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from django.test import TestCase
22

3-
from mock import patch, Mock
3+
from unittest.mock import patch, Mock
44

55
from templated_email import send_templated_mail
66

tox-requirements.txt

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ django-anymail
22
django-pytest
33
django-render-block
44
html2text
5-
mock
65
pytest
76
pytest-django
87
pytest-pythonpath

0 commit comments

Comments
 (0)