from tests.base_test import BaseTest
from crc.scripts.reset_workflow import ResetWorkflow
from crc.api.common import ApiError


class TestWorkflowReset(BaseTest):

    def test_workflow_reset_validation(self):
        self.load_test_spec('empty_workflow', master_spec=True)
        self.create_reference_document()
        spec_model = self.load_test_spec('reset_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_workflow_reset(self):
        workflow = self.create_workflow('two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        first_task = workflow_api.next_task
        self.assertEqual('Task_GetName', first_task.name)

        self.complete_form(workflow, first_task, {'name': 'Mona'})
        workflow_api = self.get_workflow_api(workflow)
        second_task = workflow_api.next_task
        self.assertEqual('Task_GetAge', second_task.name)

        ResetWorkflow().do_task(second_task, workflow.study_id, workflow.id, workflow_spec_id='two_user_tasks')

        workflow_api = self.get_workflow_api(workflow)
        task = workflow_api.next_task
        self.assertEqual('Task_GetName', task.name)

    def test_workflow_reset_missing_name(self):
        workflow = self.create_workflow('two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        first_task = workflow_api.next_task

        with self.assertRaises(ApiError):
            ResetWorkflow().do_task(first_task, workflow.study_id, workflow.id)

    def test_workflow_reset_bad_name(self):
        workflow = self.create_workflow('two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        first_task = workflow_api.next_task

        with self.assertRaises(ApiError):
            ResetWorkflow().do_task(first_task, workflow.study_id, workflow.id, workflow_spec_id='bad_workflow_name')

    def test_workflow_reset_no_start(self):
        """Sometimes we want to reset the workflow, but not start it up (don't do the engine steps etc...)"""
        workflow = self.create_workflow('two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        task = workflow_api.next_task

        ResetWorkflow().do_task(task, workflow.study_id, workflow.id, workflow_spec_id='two_user_tasks')

    def test_reset_workflow_clear_data(self):
        workflow = self.create_workflow('two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        task = workflow_api.next_task
        self.assertEqual('Task_GetName', task.name)

        workflow_api = self.complete_form(workflow, task, {'name': 'Mona'})
        task = workflow_api.next_task
        self.assertEqual('Task_GetAge', task.name)
        self.assertEqual('Mona', task.data['name'])

        # Reset the workflow, do not clear the data
        ResetWorkflow().do_task(task, workflow.study_id, workflow.id,
                                workflow_spec_id='two_user_tasks')
        workflow_api = self.get_workflow_api(workflow)
        task = workflow_api.next_task
        self.assertEqual('Task_GetName', task.name)
        self.assertEqual('Mona', task.data['name'])

        # Reset the workflow and clear the data
        ResetWorkflow().do_task(task, workflow.study_id, workflow.id,
                                workflow_spec_id='two_user_tasks', clear_data=True)
        workflow_api = self.get_workflow_api(workflow)
        task = workflow_api.next_task
        self.assertEqual('Task_GetName', task.name)
        self.assertNotIn('name', task.data)