1
0
Fork 0

Use generator in Mailbox.get_mail to limit memory leak (#149)

* Use generator in Mailbox.get_mail to limit memory leak

* Fix generator support in tests/base.py
This commit is contained in:
Adam Dobrawy 2018-03-09 19:09:45 +01:00 committed by Adam Coddington
parent a299a00506
commit 2b2bdb9825
4 changed files with 24 additions and 14 deletions

View file

@ -64,7 +64,8 @@ class TestImap(EmailMessageTestCase):
self.mailbox,
condition=lambda m: m['subject'] == self.arbitrary_identifier
)
message = next(messages)
self.assertEqual(1, len(messages))
self.assertEqual(messages[0].subject, self.arbitrary_identifier)
self.assertEqual(messages[0].text, text_content)
self.assertEqual(message.subject, self.arbitrary_identifier)
self.assertEqual(message.text, text_content)
self.assertEqual(0, len(list(messages)))