p15693087
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Used mock.Mock() in SMTPBackendTests. 

Using bool caused mypy typecheck failures.
This commit is contained in:
Konstantin Alekseev 2019-12-01 18:01:51 +03:00 committed by Mariusz Felisiak
parent 6e9c5ee88f
commit e5118b545b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/mail/tests.py
View File

@ -1493,7 +1493,7 @@ class SMTPBackendTests(BaseEmailBackendTests, SMTPBackendTestsBase):
def test_reopen_connection(self): def test_reopen_connection(self):
backend = smtp.EmailBackend() backend = smtp.EmailBackend()
# Simulate an already open connection. # Simulate an already open connection.
backend.connection = True backend.connection = mock.Mock(spec=object())
self.assertIs(backend.open(), False) self.assertIs(backend.open(), False)
def test_server_login(self): def test_server_login(self):
@ -1653,20 +1653,20 @@ class SMTPBackendTests(BaseEmailBackendTests, SMTPBackendTestsBase):
backend = smtp.EmailBackend() backend = smtp.EmailBackend()
# Simulate connection initialization success and a subsequent # Simulate connection initialization success and a subsequent
# connection exception. # connection exception.
backend.connection = True backend.connection = mock.Mock(spec=object())
backend.open = lambda: None backend.open = lambda: None
email = EmailMessage('Subject', 'Content', 'from@example.com', ['to@example.com']) email = EmailMessage('Subject', 'Content', 'from@example.com', ['to@example.com'])
self.assertEqual(backend.send_messages([email]), 0) self.assertEqual(backend.send_messages([email]), 0)
def test_send_messages_empty_list(self): def test_send_messages_empty_list(self):
backend = smtp.EmailBackend() backend = smtp.EmailBackend()
backend.connection = True backend.connection = mock.Mock(spec=object())
self.assertEqual(backend.send_messages([]), 0) self.assertEqual(backend.send_messages([]), 0)
def test_send_messages_zero_sent(self): def test_send_messages_zero_sent(self):
"""A message isn't sent if it doesn't have any recipients.""" """A message isn't sent if it doesn't have any recipients."""
backend = smtp.EmailBackend() backend = smtp.EmailBackend()
backend.connection = True backend.connection = mock.Mock(spec=object())
email = EmailMessage('Subject', 'Content', 'from@example.com', to=[]) email = EmailMessage('Subject', 'Content', 'from@example.com', to=[])
sent = backend.send_messages([email]) sent = backend.send_messages([email])
self.assertEqual(sent, 0) self.assertEqual(sent, 0)