From de8a7b14014abc895b7a9c2e8de19bc5bbc3c818 Mon Sep 17 00:00:00 2001 From: Patrick Maia and Victor Shyba Date: Mon, 27 Oct 2014 15:25:45 -0300 Subject: Card #30 - does no crash when a part does not have headers --- service/test/unit/adapter/soledad_querier_test.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'service/test/unit') 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']) -- cgit v1.2.3