diff options
| -rw-r--r-- | service/pixelated/adapter/mail_service.py | 4 | ||||
| -rw-r--r-- | service/test/adapter/mail_service_test.py | 3 |
2 files changed, 4 insertions, 3 deletions
diff --git a/service/pixelated/adapter/mail_service.py b/service/pixelated/adapter/mail_service.py index f5bfd1c8..5a60d1bb 100644 --- a/service/pixelated/adapter/mail_service.py +++ b/service/pixelated/adapter/mail_service.py @@ -57,8 +57,8 @@ class MailService: mails = [] if query['tags']: tags = [Tag(tag) for tag in query['tags']] - for leap_mailbox in self.account.mailboxes: - mailbox = PixelatedMailbox(leap_mailbox) + for leap_mailbox_name in self.account.mailboxes: + mailbox = PixelatedMailbox(self.account.getMailbox(leap_mailbox_name)) if len(mailbox.all_tags().intersection(tags)): # mailbox has at least one mail with tag for mail in mailbox.mails(): diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py index b351df1c..d736e0d0 100644 --- a/service/test/adapter/mail_service_test.py +++ b/service/test/adapter/mail_service_test.py @@ -65,7 +65,8 @@ class TestMailService(unittest.TestCase): mailbox.messages = [matching_mail, not_matching_mail] mailbox.all_tags.return_value = set() account = MagicMock() - account.mailboxes = [mailbox] + account.mailboxes = ['inbox'] + account.getMailbox.return_value = mailbox leap_session = MagicMock(account=account) |