73 lines
3.1 KiB
Python
73 lines
3.1 KiB
Python
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)
|
|
|