def test_attachment_base64(self):
soledad = mock()
bdoc = mock()
bdoc.content = {'raw': base64.encodestring('esse papo seu ta qualquer coisa'), 'content-type': 'text/plain'}
when(soledad).get_from_index('by-type-and-payloadhash', 'cnt', any(unicode)).thenReturn([bdoc])
querier = SoledadQuerier(soledad)
attachment = yield querier.attachment(u'0400BEBACAFE', 'base64')
self.assertEquals('esse papo seu ta qualquer coisa', attachment['content'])
def test_attachment_quoted_printable(self):
soledad = mock()
bdoc = mock()
bdoc.content = {'raw': quopri.encodestring('esse papo seu ta qualquer coisa'), 'content-type': 'text/plain'}
when(soledad).get_from_index('by-type-and-payloadhash', 'cnt', any(unicode)).thenReturn([bdoc])
querier = SoledadQuerier(soledad)
attachment = querier.attachment(u'0400BEBACAFE', 'quoted-printable')
self.assertEquals('esse papo seu ta qualquer coisa', attachment['content'])
