from tests.base_test import BaseTest

from crc import app, session
from crc.services.protocol_builder import ProtocolBuilderService

from unittest.mock import patch


class TestIRBInfo(BaseTest):

    @patch('crc.services.protocol_builder.requests.get')
    def test_irb_info_script(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('irb_info.json')
        workflow = self.create_workflow('irb_info_script')
        irb_info = ProtocolBuilderService.get_irb_info(workflow.study_id)
        workflow_api = self.get_workflow_api(workflow)
        first_task = workflow_api.next_task
        self.assertEqual('Task_PrintInfo', first_task.name)
        # The API returns a list, but the get_irb_info script returns the first element in the list
        self.assertEqual('Downloaded', irb_info[0]['STATUS'])
        self.assertEqual('IRB Event 1', irb_info[0]['IRBEVENT'])
        self.assertEqual('IRB Status 1', irb_info[0]['IRB_STATUS'])
        self.assertEqual(f'IRB Info: {irb_info[0]}', first_task.documentation)

    @patch('crc.services.protocol_builder.requests.get')
    def test_irb_info_script_no_result(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('irb_info_error.json')
        workflow = self.create_workflow('irb_info_script')
        irb_info = ProtocolBuilderService.get_irb_info(workflow.study_id)
        workflow_api = self.get_workflow_api(workflow)
        first_task = workflow_api.next_task
        self.assertIn(irb_info['DETAIL'], first_task.documentation)
        self.assertIn(irb_info['STATUS'], first_task.documentation)
        self.assertEqual(irb_info['DETAIL'], 'Study not downloaded to IRB Online.')
        self.assertEqual(irb_info['STATUS'], 'Error')