Card #30 - does no crash when a part does not have headers

2 files changed, 13 insertions, 1 deletions

diff --git a/service/pixelated/adapter/soledad_querier.py b/service/pixelated/adapter/soledad_querier.py
index 2ccbcd72..4123c71d 100644
--- a/service/pixelated/adapter/soledad_querier.py
+++ b/service/pixelated/adapter/soledad_querier.py
@@ -123,7 +123,7 @@ class SoledadQuerier:
             for part_key in hdoc['part_map'].keys():
                 self._extract_parts(hdoc['part_map'][part_key], parts)
         else:
-            headers_dict = {elem[0]: elem[1] for elem in hdoc['headers']}
+            headers_dict = {elem[0]: elem[1] for elem in hdoc.get('headers', [])}
             if 'attachment' in headers_dict.get('Content-Disposition', ''):
                 parts['attachments'].append(self._extract_attachment(hdoc, headers_dict))
             else:
diff --git a/service/test/unit/adapter/soledad_querier_test.py b/service/test/unit/adapter/soledad_querier_test.py
index 87ed5512..c63a3962 100644
--- a/service/test/unit/adapter/soledad_querier_test.py
+++ b/service/test/unit/adapter/soledad_querier_test.py
@@ -52,3 +52,15 @@ class SoledadQuerierTest(unittest.TestCase):
             self.assertIn('headers', attachment)
             self.assertIn('ident', attachment)
             self.assertIn('name', attachment)
+
+    def test_extract_part_without_headers(self):
+        soledad = mock()
+        bdoc = mock()
+        bdoc.content = {'raw': 'esse papo seu ta qualquer coisa'}
+        when(soledad).get_from_index('by-type-and-payloadhash', 'cnt', any(unicode)).thenReturn([bdoc])
+        hdoc = {'multi': True, 'part_map': {'1': {'multi': False, 'phash': u'0400BEBACAFE'}}}
+        querier = SoledadQuerier(soledad)
+
+        parts = querier._extract_parts(hdoc)
+
+        self.assertEquals(bdoc.content['raw'], parts['alternatives'][0]['content'])