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 = yield querier._extract_parts(hdoc)
self.assertEquals(bdoc.content['raw'], parts['alternatives'][0]['content'])
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'])
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 = yield querier._extract_parts(hdoc)
self.assertEquals(0, len(parts['alternatives']))
self.assertEquals(0, len(parts['attachments']))
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_extract_parts(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])
multipart_attachment_file = pkg_resources.resource_filename('test.unit.fixtures', 'multipart_attachment.json')
with open(multipart_attachment_file) as f:
hdoc = json.loads(f.read())
querier = SoledadQuerier(soledad)
parts = yield querier._extract_parts(hdoc)
self.assertIn('alternatives', parts.keys())
self.assertIn('attachments', parts.keys())
self.assertEquals(2, len(parts['alternatives']))
self.assertEquals(1, len(parts['attachments']))
self.check_alternatives(parts)
self.check_attachments(parts)
def test_extract_parts(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])
multipart_attachment_file = os.path.join(os.path.dirname(__file__), '..', 'fixtures', 'multipart_attachment.json')
with open(multipart_attachment_file) as f:
hdoc = json.loads(f.read())
querier = SoledadQuerier(soledad)
parts = querier._extract_parts(hdoc)
self.assertIn('alternatives', parts.keys())
self.assertIn('attachments', parts.keys())
self.assertEquals(2, len(parts['alternatives']))
self.assertEquals(1, len(parts['attachments']))
self.check_alternatives(parts)
self.check_attachments(parts)
