cr-connect-workflow/tests/study/test_study_actions_status.py

73 lines
3.1 KiB
Python
Raw Normal View History

from tests.base_test import BaseTest
from crc import session
from crc.models.study import StudyModel, StudyStatus, StudySchema
import json
from unittest.mock import patch
class TestStudyActionsStatus(BaseTest):
def update_study_status(self, study, study_schema):
self.app.put('/v1.0/study/%i' % study.id,
content_type="application/json",
headers=self.logged_in_headers(),
data=json.dumps(study_schema))
# The error happened when the dashboard reloaded,
# in particular, when we got the studies for the user
api_response = self.app.get('/v1.0/study', headers=self.logged_in_headers(), content_type="application/json")
self.assert_success(api_response)
study_result = session.query(StudyModel).filter(StudyModel.id == study.id).first()
return study_result
@patch('crc.services.protocol_builder.ProtocolBuilderService.get_study_details') # mock_details
def test_hold_study(self, mock_details):
self.load_example_data()
details_response = self.protocol_builder_response('study_details.json')
mock_details.return_value = json.loads(details_response)
study = session.query(StudyModel).first()
self.assertEqual(study.status, StudyStatus.in_progress)
study_schema = StudySchema().dump(study)
study_schema['status'] = 'hold'
study_schema['comment'] = 'This is my hold comment'
study_result = self.update_study_status(study, study_schema)
self.assertEqual(StudyStatus.hold, study_result.status)
@patch('crc.services.protocol_builder.ProtocolBuilderService.get_study_details') # mock_details
def test_abandon_study(self, mock_details):
self.load_example_data()
details_response = self.protocol_builder_response('study_details.json')
mock_details.return_value = json.loads(details_response)
study = session.query(StudyModel).first()
self.assertEqual(study.status, StudyStatus.in_progress)
study_schema = StudySchema().dump(study)
study_schema['status'] = 'abandoned'
study_schema['comment'] = 'This is my abandon comment'
study_result = self.update_study_status(study, study_schema)
self.assertEqual(StudyStatus.abandoned, study_result.status)
@patch('crc.services.protocol_builder.ProtocolBuilderService.get_study_details') # mock_details
def test_open_enrollment_study(self, mock_details):
self.load_example_data()
details_response = self.protocol_builder_response('study_details.json')
mock_details.return_value = json.loads(details_response)
study = session.query(StudyModel).first()
self.assertEqual(study.status, StudyStatus.in_progress)
study_schema = StudySchema().dump(study)
study_schema['status'] = 'open_for_enrollment'
study_schema['comment'] = 'This is my open enrollment comment'
study_schema['enrollment_date'] = '2021-01-04T05:00:00.000Z'
study_result = self.update_study_status(study, study_schema)
self.assertEqual(StudyStatus.open_for_enrollment, study_result.status)