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)