from tests.base_test import BaseTest from crc import db from crc.services.file_service import FileService from crc.services.workflow_processor import WorkflowProcessor class TestFileService(BaseTest): """Largely tested via the test_file_api, and time is tight, but adding new tests here.""" def test_add_file_from_task_increments_version_and_replaces_on_subsequent_add(self): self.load_example_data() self.create_reference_document() workflow = self.create_workflow('file_upload_form') processor = WorkflowProcessor(workflow) task = processor.next_task() irb_code = "UVACompl_PRCAppr" # The first file referenced in pb required docs. FileService.add_workflow_file(workflow_id=workflow.id, name="anything.png", content_type="text", binary_data=b'1234', irb_doc_code=irb_code) # Add the file again with different data FileService.add_workflow_file(workflow_id=workflow.id, name="anything.png", content_type="text", binary_data=b'5678', irb_doc_code=irb_code) file_models = FileService.get_workflow_files(workflow_id=workflow.id) self.assertEqual(1, len(file_models)) file_data = FileService.get_workflow_data_files(workflow_id=workflow.id) self.assertEqual(1, len(file_data)) self.assertEqual(2, file_data[0].version) def test_add_file_from_form_increments_version_and_replaces_on_subsequent_add_with_same_name(self): self.load_example_data() self.create_reference_document() workflow = self.create_workflow('file_upload_form') processor = WorkflowProcessor(workflow) task = processor.next_task() irb_code = "UVACompl_PRCAppr" # The first file referenced in pb required docs. FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="anything.png", content_type="text", binary_data=b'1234') # Add the file again with different data FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="anything.png", content_type="text", binary_data=b'5678') def test_replace_archive_file_unarchives_the_file_and_updates(self): self.load_example_data() self.create_reference_document() workflow = self.create_workflow('file_upload_form') processor = WorkflowProcessor(workflow) task = processor.next_task() irb_code = "UVACompl_PRCAppr" # The first file referenced in pb required docs. FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="anything.png", content_type="text", binary_data=b'1234') # Archive the file file_models = FileService.get_workflow_files(workflow_id=workflow.id) self.assertEquals(1, len(file_models)) file_model = file_models[0] file_model.archived = True db.session.add(file_model) # Assure that the file no longer comes back. file_models = FileService.get_workflow_files(workflow_id=workflow.id) self.assertEquals(0, len(file_models)) # Add the file again with different data FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="anything.png", content_type="text", binary_data=b'5678') file_models = FileService.get_workflow_files(workflow_id=workflow.id) self.assertEqual(1, len(file_models)) file_data = FileService.get_workflow_data_files(workflow_id=workflow.id) self.assertEqual(1, len(file_data)) self.assertEqual(2, file_data[0].version) self.assertEqual(b'5678', file_data[0].data) def test_add_file_from_form_allows_multiple_files_with_different_names(self): self.load_example_data() self.create_reference_document() workflow = self.create_workflow('file_upload_form') processor = WorkflowProcessor(workflow) task = processor.next_task() irb_code = "UVACompl_PRCAppr" # The first file referenced in pb required docs. FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="anything.png", content_type="text", binary_data=b'1234') # Add the file again with different data FileService.add_workflow_file(workflow_id=workflow.id, irb_doc_code=irb_code, name="a_different_thing.png", content_type="text", binary_data=b'5678') file_models = FileService.get_workflow_files(workflow_id=workflow.id) self.assertEqual(2, len(file_models))