From 5cf360add8cdbbdc4ca1c620d5c04a8af8956ace Mon Sep 17 00:00:00 2001 From: Folker Bernitt Date: Tue, 12 Jan 2016 15:25:42 +0100 Subject: Refactor AttachmentInfo: Remove knowledge about part map structure --- service/test/unit/adapter/mailstore/test_leap_mail.py | 5 ++--- service/test/unit/adapter/mailstore/test_leap_mailstore.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) (limited to 'service/test/unit/adapter') diff --git a/service/test/unit/adapter/mailstore/test_leap_mail.py b/service/test/unit/adapter/mailstore/test_leap_mail.py index 4b074082..91e960e1 100644 --- a/service/test/unit/adapter/mailstore/test_leap_mail.py +++ b/service/test/unit/adapter/mailstore/test_leap_mail.py @@ -85,9 +85,8 @@ class TestLeapMail(TestCase): self.assertEqual(body, mail.as_dict()['body']) def test_as_dict_with_attachments(self): - mail = LeapMail('doc id', 'INBOX', attachments=[AttachmentInfo({'phash': 'id', 'ctype': 'text/plain', 'size': 2}, - {'Content-Description': 'name', - 'Content-Transfer-Encoding': 'encoding'})]) + attachment_info = AttachmentInfo('id', 'name', 'encoding', ctype='text/plain', size=2) + mail = LeapMail('doc id', 'INBOX', attachments=[attachment_info]) self.assertEqual([{'ident': 'id', 'name': 'name', 'encoding': 'encoding', 'content-type': 'text/plain', 'size': 2}], mail.as_dict()['attachments']) diff --git a/service/test/unit/adapter/mailstore/test_leap_mailstore.py b/service/test/unit/adapter/mailstore/test_leap_mailstore.py index 1eae48df..0945e0c4 100644 --- a/service/test/unit/adapter/mailstore/test_leap_mailstore.py +++ b/service/test/unit/adapter/mailstore/test_leap_mailstore.py @@ -37,7 +37,7 @@ import pkg_resources from leap.mail.mail import Message from pixelated.adapter.mailstore import underscore_uuid -from pixelated.adapter.mailstore.leap_mailstore import LeapMailStore, LeapMail, AttachmentInfo +from pixelated.adapter.mailstore.leap_mailstore import LeapMailStore, LeapMail class TestLeapMailStore(TestCase): -- cgit v1.2.3