cr-connect-workflow/tests/scripts/test_start_workflow.py
mike cullerton ed0499a573 Make sure we have a workflow.
Raise an ApiError with the workflow_spec_id if we cannot find a workflow
2022-02-24 12:37:25 -05:00

48 lines
2.0 KiB
Python

from tests.base_test import BaseTest
from crc import session
from crc.models.study import StudyModel
from crc.models.workflow import WorkflowModel
from crc.services.study_service import StudyService
from crc.services.workflow_spec_service import WorkflowSpecService
class TestStartWorkflow(BaseTest):
def setup_test_start_workflow(self):
self.add_users()
self.create_workflow('hello_world')
workflow_spec_to_start = WorkflowSpecService().get_spec('hello_world')
workflow = self.create_workflow('start_workflow')
study_id = workflow.study_id
study = session.query(StudyModel).filter(StudyModel.id==study_id).first()
StudyService.add_all_workflow_specs_to_study(study, [workflow_spec_to_start])
return workflow
def test_start_workflow_validation(self):
spec_model = self.load_test_spec('start_workflow')
rv = self.app.get('/v1.0/workflow-specification/%s/validate' % spec_model.id, headers=self.logged_in_headers())
self.assertEqual([], rv.json)
def test_start_workflow(self):
workflow = self.setup_test_start_workflow()
workflow_before = session.query(WorkflowModel).filter(WorkflowModel.workflow_spec_id=='hello_world').first()
self.assertEqual('not_started', workflow_before.status.value)
workflow_api = self.get_workflow_api(workflow)
task = workflow_api.next_task
self.complete_form(workflow, task, {'workflow_spec_to_start': 'hello_world'})
workflow_after = session.query(WorkflowModel).filter(WorkflowModel.workflow_spec_id=='hello_world').first()
self.assertEqual('user_input_required', workflow_after.status.value)
def test_bad_workflow_spec_id(self):
workflow = self.setup_test_start_workflow()
workflow_api = self.get_workflow_api(workflow)
task = workflow_api.next_task
with self.assertRaises(AssertionError) as e:
self.complete_form(workflow, task, {'workflow_spec_to_start': 'bad_spec_id'})