From 6b62849aa40023fd9eba666560984e0edfedc55d Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Thu, 11 Sep 2014 10:26:18 -0300 Subject: generating unique ident for PixelatedMail --- service/test/adapter/pixelated_mailbox_test.py | 5 +++-- service/test/adapter/test_helper.py | 3 ++- 2 files changed, 5 insertions(+), 3 deletions(-) (limited to 'service/test/adapter') diff --git a/service/test/adapter/pixelated_mailbox_test.py b/service/test/adapter/pixelated_mailbox_test.py index 9bc65de0..6b84616f 100644 --- a/service/test/adapter/pixelated_mailbox_test.py +++ b/service/test/adapter/pixelated_mailbox_test.py @@ -58,12 +58,13 @@ class TestPixelatedMailbox(unittest.TestCase): self.assertIsNone(mailbox.tag_index.get('one_tag')) def test_mailbox_tag_is_added_when_new_mail_arrives(self): - mail_one = test_helper.leap_mail(uid=0) + mail_one = test_helper.leap_mail(uid=0, mbox='SENT') leap_mailbox = test_helper.leap_mailbox(messages=[mail_one], mailbox_name='SENT') mailbox = PixelatedMailbox(leap_mailbox, self.db_file_path) - mail = mailbox.mail(0) + from pixelated.support.id_gen import gen_pixelated_uid + mail = mailbox.mail(gen_pixelated_uid('SENT', 0)) self.assertIn('sent', mail.tags) def test_index_is_initialized_with_mail_tags_if_empty(self): diff --git a/service/test/adapter/test_helper.py b/service/test/adapter/test_helper.py index a904a041..ad39f343 100644 --- a/service/test/adapter/test_helper.py +++ b/service/test/adapter/test_helper.py @@ -41,9 +41,10 @@ def mail_dict(): } -def leap_mail(uid=0, flags=LEAP_FLAGS, headers=DEFAULT_HEADERS, extra_headers={}): +def leap_mail(uid=0, flags=LEAP_FLAGS, headers=DEFAULT_HEADERS, extra_headers={}, mbox='INBOX'): headers = dict(headers.items() + extra_headers.items()) return Mock(getUID=Mock(return_value=uid), + _mbox=mbox, getFlags=Mock(return_value=flags), bdoc=Mock(content={'raw': 'test'}), hdoc=Mock(content={'headers': headers})) -- cgit v1.2.3