import json from crc import session from crc.models.file import FileModelSchema from crc.models.study import StudyModel from crc.models.workflow import WorkflowSpecModel, WorkflowSpecModelSchema, WorkflowModel, \ WorkflowApiSchema, WorkflowStatus, Task from tests.base_test import BaseTest class TestTasksApi(BaseTest): def create_workflow(self, workflow_name): study = session.query(StudyModel).first() spec = session.query(WorkflowSpecModel).filter_by(id=workflow_name).first() rv = self.app.post( '/v1.0/study/%i/workflows' % study.id, headers=self.logged_in_headers(), content_type="application/json", data=json.dumps(WorkflowSpecModelSchema().dump(spec))) self.assert_success(rv) workflow = session.query(WorkflowModel).filter_by(study_id=study.id, workflow_spec_id=workflow_name).first() return workflow def get_workflow_api(self, workflow): rv = self.app.get('/v1.0/workflow/%i' % workflow.id, content_type="application/json") json_data = json.loads(rv.get_data(as_text=True)) workflow_api = WorkflowApiSchema().load(json_data) self.assertEqual(workflow.workflow_spec_id, workflow_api.workflow_spec_id) return workflow_api def complete_form(self, workflow, task, dict_data): rv = self.app.put('/v1.0/workflow/%i/task/%s/data' % (workflow.id, task.id), content_type="application/json", data=json.dumps(dict_data)) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) workflow = WorkflowApiSchema().load(json_data) return workflow def test_get_current_user_tasks(self): self.load_example_data() workflow = self.create_workflow('random_fact') tasks = self.get_workflow_api(workflow).user_tasks self.assertEqual("Task_User_Select_Type", tasks[0].name) self.assertEqual(3, len(tasks[0].form["fields"][0]["options"])) self.assertIsNotNone(tasks[0].documentation) expected_docs = """# h1 Heading 8-) ## h2 Heading ### h3 Heading #### h4 Heading ##### h5 Heading ###### h6 Heading """ self.assertTrue(str.startswith(tasks[0].documentation, expected_docs)) def test_two_forms_task(self): # Set up a new workflow self.load_example_data() workflow = self.create_workflow('two_forms') # get the first form in the two form workflow. workflow_api = self.get_workflow_api(workflow) self.assertEqual('two_forms', workflow_api.workflow_spec_id) self.assertEqual(2, len(workflow_api.user_tasks)) self.assertIsNotNone(workflow_api.user_tasks[0].form) self.assertEqual("UserTask", workflow_api.next_task['type']) self.assertEqual("StepOne", workflow_api.next_task['name']) self.assertEqual(1, len(workflow_api.next_task['form']['fields'])) # Complete the form for Step one and post it. self.complete_form(workflow, workflow_api.user_tasks[0], {"color": "blue"}) # Get the next Task workflow_api = self.get_workflow_api(workflow) self.assertEqual("StepTwo", workflow_api.next_task['name']) # Get all user Tasks and check that the data have been saved for task in workflow_api.user_tasks: self.assertIsNotNone(task.data) for val in task.data.values(): self.assertIsNotNone(val) def test_error_message_on_bad_gateway_expression(self): self.load_example_data() workflow = self.create_workflow('exclusive_gateway') # get the first form in the two form workflow. tasks = self.get_workflow_api(workflow).user_tasks self.complete_form(workflow, tasks[0], {"has_bananas": True}) def test_workflow_with_parallel_forms(self): self.load_example_data() workflow = self.create_workflow('exclusive_gateway') # get the first form in the two form workflow. tasks = self.get_workflow_api(workflow).user_tasks self.complete_form(workflow, tasks[0], {"has_bananas": True}) # Get the next Task workflow_api = self.get_workflow_api(workflow) self.assertEqual("Task_Num_Bananas", workflow_api.next_task['name']) def test_get_workflow_contains_details_about_last_task_data(self): self.load_example_data() workflow = self.create_workflow('exclusive_gateway') # get the first form in the two form workflow. tasks = self.get_workflow_api(workflow).user_tasks workflow_api = self.complete_form(workflow, tasks[0], {"has_bananas": True}) self.assertIsNotNone(workflow_api.last_task) self.assertEqual({"has_bananas": True}, workflow_api.last_task['data']) def test_get_workflow_contains_reference_to_last_task_and_next_task(self): self.load_example_data() workflow = self.create_workflow('exclusive_gateway') # get the first form in the two form workflow. tasks = self.get_workflow_api(workflow).user_tasks self.complete_form(workflow, tasks[0], {"has_bananas": True}) workflow_api = self.get_workflow_api(workflow) self.assertIsNotNone(workflow_api.last_task) self.assertIsNotNone(workflow_api.next_task) def test_document_added_to_workflow_shows_up_in_file_list(self): self.load_example_data() workflow = self.create_workflow('docx') # get the first form in the two form workflow. tasks = self.get_workflow_api(workflow).user_tasks data = { "full_name": "Buck of the Wild", "date": "5/1/2020", "title": "Leader of the Pack", "company": "In the company of wolves", "last_name": "Mr. Wolf" } workflow_api = self.complete_form(workflow, tasks[0], data) self.assertIsNotNone(workflow_api.next_task) self.assertEquals("EndEvent_0evb22x", workflow_api.next_task['name']) self.assertTrue(workflow_api.status == WorkflowStatus.complete) rv = self.app.get('/v1.0/file?workflow_id=%i' % workflow.id) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) files = FileModelSchema(many=True).load(json_data, session=session) self.assertTrue(len(files) == 1) def test_documentation_processing_handles_replacements(self): docs = "Some simple docs" task = Task(1, "bill", "bill", "", "started", {}, docs, {}) task.process_documentation(docs) self.assertEqual(docs, task.documentation) task.data = {"replace_me": "new_thing"} task.process_documentation("{{replace_me}}") self.assertEqual("new_thing", task.documentation) documentation = """ # Bigger Test * bullet one * bullet two has {{replace_me}} # other stuff. """ expected = """ # Bigger Test * bullet one * bullet two has new_thing # other stuff. """ task.process_documentation(documentation) self.assertEqual(expected, task.documentation) def test_get_documentation_populated_in_end(self): self.load_example_data() workflow = self.create_workflow('random_fact') workflow_api = self.get_workflow_api(workflow) tasks = workflow_api.user_tasks self.assertEqual("Task_User_Select_Type", tasks[0].name) self.assertEqual(3, len(tasks[0].form["fields"][0]["options"])) self.assertIsNotNone(tasks[0].documentation) self.complete_form(workflow, workflow_api.user_tasks[0], {"type": "norris"}) workflow_api = self.get_workflow_api(workflow) self.assertEqual("EndEvent_0u1cgrf", workflow_api.next_task['name']) self.assertIsNotNone(workflow_api.next_task['documentation']) self.assertTrue("norris" in workflow_api.next_task['documentation'])