130 lines
5.5 KiB
Python
130 lines
5.5 KiB
Python
from tests.base_test import BaseTest
|
|
|
|
from crc import mail
|
|
|
|
from crc.models.file import FileModel
|
|
from crc.services.user_file_service import UserFileService
|
|
|
|
|
|
class TestEmailAttachments(BaseTest):
|
|
|
|
irb_code_1 = 'Study_App_Doc'
|
|
irb_code_2 = 'Study_Protocol_Document'
|
|
|
|
def setup_attachments(self):
|
|
self.create_reference_document()
|
|
|
|
workflow = self.create_workflow('email_script')
|
|
workflow_api = self.get_workflow_api(workflow)
|
|
first_task = workflow_api.next_task
|
|
|
|
UserFileService.add_workflow_file(workflow_id=workflow.id,
|
|
task_spec_name=first_task.name,
|
|
name="something.png", content_type="text",
|
|
binary_data=b'1234', irb_doc_code=self.irb_code_1)
|
|
UserFileService.add_workflow_file(workflow_id=workflow.id,
|
|
task_spec_name=first_task.name,
|
|
name="another.png", content_type="text",
|
|
binary_data=b'67890', irb_doc_code=self.irb_code_1)
|
|
UserFileService.add_workflow_file(workflow_id=workflow.id,
|
|
task_spec_name=first_task.name,
|
|
name="anything.png", content_type="text",
|
|
binary_data=b'5678', irb_doc_code=self.irb_code_2)
|
|
return workflow, first_task
|
|
|
|
def test_email_attachments_one_code(self):
|
|
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': self.irb_code_1
|
|
}
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
self.assertEqual(2, len(outbox[0].attachments))
|
|
self.assertEqual('image/png', outbox[0].attachments[0].content_type)
|
|
self.assertEqual('something.png', outbox[0].attachments[0].filename)
|
|
self.assertEqual(b'1234', outbox[0].attachments[0].data)
|
|
|
|
def test_email_attachments_list_of_codes(self):
|
|
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': [self.irb_code_1, self.irb_code_2]
|
|
}
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
self.assertEqual(3, len(outbox[0].attachments))
|
|
|
|
def test_email_attachments_list_of_codes_with_empty_filters(self):
|
|
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': [(self.irb_code_1, []), (self.irb_code_2, [])]
|
|
}
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
self.assertEqual(3, len(outbox[0].attachments))
|
|
|
|
def test_email_attachments_list_of_codes_with_filters(self):
|
|
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
doc_code_tuple_1 = (self.irb_code_1, ['another.png'])
|
|
doc_code_tuple_2 = (self.irb_code_2, [])
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': [doc_code_tuple_1, doc_code_tuple_2]
|
|
}
|
|
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
# We should only get 2 files because we filtered one out
|
|
self.assertEqual(2, len(outbox[0].attachments))
|
|
|
|
def test_email_attachments_list_of_codes_some_with_filters(self):
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
doc_code_tuple = (self.irb_code_1, ['another.png'])
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': [doc_code_tuple, self.irb_code_2]
|
|
}
|
|
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
# We should only get 2 files because we filtered one out
|
|
self.assertEqual(2, len(outbox[0].attachments))
|
|
|
|
def test_email_attachments_no_archived(self):
|
|
"""Update a user file.
|
|
Make sure we don't add the archived file as an attachment.
|
|
"""
|
|
|
|
workflow, first_task = self.setup_attachments()
|
|
|
|
file_model = FileModel.query.first()
|
|
new_data = b'NewData'
|
|
|
|
UserFileService().update_file(file_model, new_data, file_model.content_type)
|
|
UserFileService().add_workflow_file(file_model.workflow_id, file_model.irb_doc_code, file_model.task_spec, file_model.name, file_model.content_type, new_data)
|
|
|
|
form_data = {'subject': 'My Test Subject',
|
|
'recipients': 'user@example.com',
|
|
'doc_codes': [self.irb_code_1, self.irb_code_2]
|
|
}
|
|
with mail.record_messages() as outbox:
|
|
self.complete_form(workflow, first_task, form_data)
|
|
self.assertEqual(1, len(outbox))
|
|
self.assertEqual(3, len(outbox[0].attachments))
|