67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
from tests.base_test import BaseTest
|
|
|
|
from crc import session
|
|
from crc.models.email import EmailModel
|
|
from crc.services.email_service import EmailService
|
|
from unittest.mock import patch
|
|
|
|
|
|
class TestEmailService(BaseTest):
|
|
|
|
def test_add_email(self):
|
|
self.load_example_data()
|
|
study = self.create_study()
|
|
|
|
subject = 'Email Subject'
|
|
sender = 'sender@sartography.com'
|
|
recipients = ['recipient@sartography.com', 'back@sartography.com']
|
|
content = 'Content for this email'
|
|
content_html = '<p>Hypertext Markup Language content for this email</p>'
|
|
|
|
EmailService.add_email(subject=subject, sender=sender, recipients=recipients,
|
|
content=content, content_html=content_html, study_id=study.id)
|
|
|
|
email_model = EmailModel.query.first()
|
|
|
|
self.assertEqual(email_model.subject, subject)
|
|
self.assertEqual(email_model.sender, sender)
|
|
self.assertEqual(email_model.recipients, str(recipients))
|
|
self.assertEqual(email_model.content, content)
|
|
self.assertEqual(email_model.content_html, content_html)
|
|
self.assertEqual(email_model.study, study)
|
|
|
|
subject = 'Email Subject - Empty study'
|
|
EmailService.add_email(subject=subject, sender=sender, recipients=recipients,
|
|
content=content, content_html=content_html)
|
|
|
|
email_model = EmailModel.query.order_by(EmailModel.id.desc()).first()
|
|
|
|
self.assertEqual(email_model.subject, subject)
|
|
self.assertEqual(email_model.sender, sender)
|
|
self.assertEqual(email_model.recipients, str(recipients))
|
|
self.assertEqual(email_model.content, content)
|
|
self.assertEqual(email_model.content_html, content_html)
|
|
self.assertEqual(email_model.study, None)
|
|
|
|
@patch('crc.services.email_service.EmailService.add_email')
|
|
def test_add_email_with_error(self, mock_response):
|
|
|
|
self.load_example_data()
|
|
mock_response.return_value.ok = True
|
|
mock_response.side_effect = Exception("This is my exception!")
|
|
|
|
study = self.create_study()
|
|
|
|
subject = 'Email Subject'
|
|
sender = 'sender@sartography.com'
|
|
recipients = ['recipient@sartography.com', 'back@sartography.com']
|
|
content = 'Content for this email'
|
|
content_html = '<p>Hypertext Markup Language content for this email</p>'
|
|
|
|
# Make sure we generate an error
|
|
with self.assertRaises(Exception) as e:
|
|
EmailService.add_email(subject=subject, sender=sender, recipients=recipients,
|
|
content=content, content_html=content_html, study_id=study.id)
|
|
# Make sure it's the error we want
|
|
self.assertEqual('This is my exception!', e.exception.args[0])
|