#30 - fixes a bug related to multipart mail with empty part_map

1 files changed, 15 insertions, 0 deletions

diff --git a/service/test/unit/adapter/soledad_querier_test.py b/service/test/unit/adapter/soledad_querier_test.py
index 84f2bd5e..d07fdb00 100644
--- a/service/test/unit/adapter/soledad_querier_test.py
+++ b/service/test/unit/adapter/soledad_querier_test.py
@@ -67,6 +67,21 @@ class SoledadQuerierTest(unittest.TestCase):
 
         self.assertEquals(bdoc.content['raw'], parts['alternatives'][0]['content'])
 
+    def test_extract_handles_missing_part_map(self):
+        soledad = mock()
+        hdoc = {u'multi': True,
+                u'ctype': u'message/delivery-status',
+                u'headers': [[u'Content-Description', u'Delivery report'], [u'Content-Type', u'message/delivery-status']],
+                u'parts': 2,
+                u'phash': None,
+                u'size': 554}
+        querier = SoledadQuerier(soledad)
+
+        parts = querier._extract_parts(hdoc)
+
+        self.assertEquals(0, len(parts['alternatives']))
+        self.assertEquals(0, len(parts['attachments']))
+
     def test_attachment_base64(self):
         soledad = mock()
         bdoc = mock()