cr-connect-workflow/tests/test_request_approval_script.py

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 = {"study": {"approvals": ['dhf8r', 'lb3dp']}}

        script = RequestApproval()
        script.do_task(task, workflow.study_id, workflow.id, "study.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())