63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
|
from tests.base_test import BaseTest
|
||
|
|
from crc import session
|
||
|
|
from crc.models.study import StudyModel, StudyStatus, StudySchema
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
def test_hold_study(self):
|
||
|
|
self.load_example_data()
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
def test_abandon_study(self):
|
||
|
|
self.load_example_data()
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
def test_open_enrollment_study(self):
|
||
|
|
self.load_example_data()
|
||
|
|
|
||
|
|
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)
|
||
|
|
|