49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from tests.base_test import BaseTest
|
|
|
|
from crc.scripts.request_approval import RequestApproval
|
|
from crc.services.workflow_processor import WorkflowProcessor
|
|
from crc.api.common import ApiError
|
|
|
|
from crc import db
|
|
from crc.models.approval import ApprovalModel
|
|
|
|
|
|
class TestRequestApprovalScript(BaseTest):
|
|
|
|
def test_do_task(self):
|
|
self.load_example_data()
|
|
self.create_reference_document()
|
|
workflow = self.create_workflow('empty_workflow')
|
|
processor = WorkflowProcessor(workflow)
|
|
task = processor.next_task()
|
|
task.data = {"approvals": ['dhf8r', 'lb3dp']}
|
|
|
|
script = RequestApproval()
|
|
script.do_task(task, workflow.study_id, workflow.id, "approvals")
|
|
self.assertEquals(2, db.session.query(ApprovalModel).count())
|
|
|
|
def test_do_task_with_incorrect_argument(self):
|
|
"""This script should raise an error if it can't figure out the approvers."""
|
|
self.load_example_data()
|
|
self.create_reference_document()
|
|
workflow = self.create_workflow('empty_workflow')
|
|
processor = WorkflowProcessor(workflow)
|
|
task = processor.next_task()
|
|
task.data = {"approvals": {'dhf8r':"invalid", 'lb3dp':"invalid"}}
|
|
script = RequestApproval()
|
|
with self.assertRaises(ApiError):
|
|
script.do_task(task, workflow.study_id, workflow.id, "approvals")
|
|
|
|
def test_do_task_validate_only(self):
|
|
self.load_example_data()
|
|
self.create_reference_document()
|
|
workflow = self.create_workflow('empty_workflow')
|
|
processor = WorkflowProcessor(workflow)
|
|
task = processor.next_task()
|
|
task.data = {"approvals": ['dhf8r', 'lb3dp']}
|
|
|
|
script = RequestApproval()
|
|
script.do_task_validate_only(task, workflow.study_id, workflow.id, "approvals")
|
|
self.assertEquals(0, db.session.query(ApprovalModel).count())
|
|
|