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)