From 377ffe8bc17bbcf00b6342428bb91eb3f2e41a20 Mon Sep 17 00:00:00 2001 From: Tulio Casagrande Date: Thu, 1 Jun 2017 17:36:53 -0300 Subject: Fix _attach_key mock to reflect bitmask-dev changes --- service/test/unit/adapter/services/test_mail_sender.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/service/test/unit/adapter/services/test_mail_sender.py b/service/test/unit/adapter/services/test_mail_sender.py index 23951214..93703c8e 100644 --- a/service/test/unit/adapter/services/test_mail_sender.py +++ b/service/test/unit/adapter/services/test_mail_sender.py @@ -24,8 +24,7 @@ from pixelated.adapter.model.mail import InputMail from pixelated.bitmask_libraries.smtp import LeapSMTPConfig from pixelated.support.functional import flatten from test.support.test_helper import mail_dict -from twisted.internet import reactor, defer -from twisted.internet.defer import Deferred +from twisted.internet import defer from mockito.matchers import Matcher @@ -108,7 +107,7 @@ class MailSenderTest(unittest.TestCase): def test_keymanager_encrypt_problem_raises_exception(self): input_mail = InputMail.from_dict(mail_dict(), from_address='pixelated@org') - when(OutgoingMail)._maybe_attach_key(any(), any(), any()).thenReturn( + when(OutgoingMail)._attach_key(any(), any()).thenReturn( defer.succeed(None)) when(OutgoingMail)._fix_headers(any(), any(), any()).thenReturn( defer.succeed((None, mock()))) -- cgit v1.2.3