def test_the_email_answer_can_have_attachments(self):
'''An email answer can have attachments'''
email_answer = EmailAnswer()
email_answer.subject = 'prueba4'
email_answer.content_text = 'prueba4lafieritaespeluda'
email_answer.add_attachment(self.photo_fiera)
email_answer.add_attachment(self.pdf_file)
self.assertTrue(email_answer.attachments)
self.assertIn(self.photo_fiera, email_answer.attachments)
self.assertIn(self.pdf_file, email_answer.attachments)
def test_save_attachments_on_save(self):
'''When saving it also calls the save an attachment'''
email_answer = EmailAnswer()
email_answer.subject = 'prueba4'
email_answer.content_text = 'prueba4lafieritaespeluda'
email_answer.add_attachment(self.photo_fiera)
email_answer.add_attachment(self.pdf_file)
any_answer = Answer.objects.first()
with patch('requests.Session.post') as post:
post.return_value = PostMock()
with patch('mailit.bin.handleemail.EmailAnswer.save_attachment') as save_attachment:
with patch('mailit.bin.handleemail.EmailAnswer.save') as save_answer:
save_answer.return_value = any_answer
email_answer.send_back()
save_attachment.assert_any_call(any_answer, self.photo_fiera)
save_attachment.assert_any_call(any_answer, self.pdf_file)
