From 3395e1ca4cd6fe2d3d4770257242e0dc470dc11f Mon Sep 17 00:00:00 2001 From: Tulio Casagrande Date: Mon, 10 Apr 2017 17:51:30 -0300 Subject: [#927] Refactor tests with @deniscostadsc --- service/test/unit/test_account_recovery.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'service/test/unit/test_account_recovery.py') diff --git a/service/test/unit/test_account_recovery.py b/service/test/unit/test_account_recovery.py index cd6835a9..96dbc402 100644 --- a/service/test/unit/test_account_recovery.py +++ b/service/test/unit/test_account_recovery.py @@ -47,7 +47,7 @@ class AccountRecoveryTest(unittest.TestCase): @defer.inlineCallbacks def test_update_recovery_code(self): when(self.account_recovery)._send_mail(ANY).thenReturn(defer.succeed(None)) - response = yield self.account_recovery.update_recovery_code() + yield self.account_recovery.update_recovery_code() self.mock_bonafide_session.update_recovery_code.assert_called_once_with(self.generated_code) @defer.inlineCallbacks @@ -58,10 +58,8 @@ class AccountRecoveryTest(unittest.TestCase): msg['From'] = sender msg['To'] = self.backup_email - result = MagicMock() - deferred_sendmail = defer.succeed(result) - with patch.object(smtp, 'sendmail', return_value=deferred_sendmail) as mock_sendmail: - response = yield self.account_recovery._send_mail(self.generated_code, self.backup_email) + with patch.object(smtp, 'sendmail', return_value=defer.succeed(None)) as mock_sendmail: + yield self.account_recovery._send_mail(self.generated_code, self.backup_email) mock_sendmail.assert_called_with( self.mock_smtp_config.remote_smtp_host, -- cgit v1.2.3