From 9442be5c230e286073244451189ffc05ae6c8083 Mon Sep 17 00:00:00 2001 From: Jefferson Stachelski Date: Wed, 3 Feb 2016 14:31:53 -0200 Subject: Issue #549 - Implemented remove attachment --- .../unit/adapter/mailstore/test_leap_attachment_store.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'service/test/unit/adapter') diff --git a/service/test/unit/adapter/mailstore/test_leap_attachment_store.py b/service/test/unit/adapter/mailstore/test_leap_attachment_store.py index 4e9b56b1..f1bd8528 100644 --- a/service/test/unit/adapter/mailstore/test_leap_attachment_store.py +++ b/service/test/unit/adapter/mailstore/test_leap_attachment_store.py @@ -116,6 +116,21 @@ class TestLeapAttachmentStore(TestCase): except ValueError: pass + @defer.inlineCallbacks + def test_soledad_delete_doc_is_called_when_deleting_an_attachment(self): + attachment_id = '1B0A9AAD9E153D24265395203C53884506ABA276394B9FEC02B214BF9E77E48E' + doc = SoledadDocument(json=json.dumps({'content_type': 'foo/bar', 'raw': 'quoted-printable', + 'phash': attachment_id, + 'content_transfer_encoding': ''})) + + when(self.soledad).get_from_index('by-type-and-payloadhash', 'cnt', attachment_id).thenReturn(defer.succeed([doc])) + when(self.soledad).delete_doc(doc).thenReturn(defer.succeed(None)) + + store = LeapAttachmentStore(self.soledad) + yield store.delete_attachment(attachment_id) + + verify(self.soledad).delete_doc(doc) + def _mock_get_mailbox(self, mailbox_name, create_new_uuid=False): mbox_uuid = self.mbox_uuid if not create_new_uuid else str(uuid4()) when(self.soledad).list_indexes().thenReturn(defer.succeed(MAIL_INDEXES)).thenReturn( -- cgit v1.2.3