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

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)