cr-connect-workflow/tests/test_protocol_builder.py

import os

from tests.base_test import BaseTest
from unittest.mock import patch

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


class TestProtocolBuilder(BaseTest):
    test_uid = "dhf8r"
    test_study_id = 1

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_studies(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('user_studies.json')
        response = ProtocolBuilderService.get_studies(self.test_uid)
        self.assertIsNotNone(response)

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_investigators(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('investigators.json')
        response = ProtocolBuilderService.get_investigators(self.test_study_id)
        self.assertIsNotNone(response)
        self.assertEqual(5, len(response))
        self.assertEqual("DC", response[0]["INVESTIGATORTYPE"])
        self.assertEqual("Department Contact", response[0]["INVESTIGATORTYPEFULL"])
        self.assertEqual("asd3v", response[0]["NETBADGEID"])

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_required_docs(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('required_docs.json')
        response = ProtocolBuilderService.get_required_docs(self.test_study_id)
        auxdocs = response['AUXDOCS']
        self.assertIsNotNone(auxdocs)
        self.assertEqual(5, len(auxdocs))
        self.assertEqual(6, auxdocs[0]['SS_AUXILIARY_DOC_TYPE_ID'])

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_details(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('study_details.json')
        response = ProtocolBuilderService.get_study_details(self.test_study_id)[0]
        self.assertIsNotNone(response)
        self.assertEqual(65, len(response))
        self.assertEqual('1234', response['IND_1'])

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_study_sponsors(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('sponsors.json')
        response = ProtocolBuilderService.get_sponsors(self.test_study_id)
        self.assertIsNotNone(response)
        self.assertEqual(3, len(response))
        self.assertEqual(2, response[0]["SS_STUDY"])
        self.assertEqual(2453, response[0]["SPONSOR_ID"])
        self.assertEqual("Abbott Ltd", response[0]["SP_NAME"])

    @patch('crc.services.protocol_builder.requests.get')
    def test_get_irb_info(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')
        response = ProtocolBuilderService.get_irb_info(self.test_study_id)
        self.assertIsNotNone(response)
        # IRB Info should always return 1 record
        self.assertEqual(1, len(response))
        self.assertEqual('IRB Event 1', response[0]["IRBEVENT"])

    @patch('crc.services.protocol_builder.requests.get')
    def test_check_study(self, mock_get):
        app.config['PB_ENABLED'] = True
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('check_study.json')
        response = ProtocolBuilderService.check_study(self.test_study_id)
        self.assertIsNotNone(response)
        self.assertIn('DETAIL', response[0].keys())
        self.assertIn('STATUS', response[0].keys())

    @patch('crc.services.protocol_builder.requests.get')
    def test_pb_recent_creation_date_filter(self, mock_get):
        """filter by recent creation date
            should only get more recent studies (3) """
        app.config['PB_ENABLED'] = True
        app.config['PB_MIN_DATE'] = "2020-01-01T00:00:00.000Z"
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('user_studies.json')
        api_response = self.app.get('/v1.0/study', headers=self.logged_in_headers(), content_type="application/json")
        self.assertEqual(3, len(api_response.json))

    @patch('crc.services.protocol_builder.requests.get')
    def test_pb_older_creation_date_filter(self, mock_get):
        """filter by older creation date
            should get all studies back (4)"""
        app.config['PB_ENABLED'] = True
        app.config['PB_MIN_DATE'] = "2000-01-01T00:00:00.000Z"
        mock_get.return_value.ok = True
        mock_get.return_value.text = self.protocol_builder_response('user_studies.json')
        api_response = self.app.get('/v1.0/study', headers=self.logged_in_headers(), content_type="application/json")
        self.assertEqual(4, len(api_response.json))