def test_check_draft_state_approved(self):
try:
with mock.patch.object(DraftRegistration, 'requires_approval', mock.PropertyMock(return_value=True)), mock.patch.object(DraftRegistration, 'is_approved', mock.PropertyMock(return_value=True)):
draft_views.check_draft_state(self.draft)
except HTTPError as e:
assert_equal(e.code, http.FORBIDDEN)
else:
self.fail()
def test_check_draft_state_pending_review(self):
self.draft.submit_for_review(self.user, self.immediate_payload, save=True)
try:
with mock.patch.object(DraftRegistration, 'requires_approval', mock.PropertyMock(return_value=True)):
draft_views.check_draft_state(self.draft)
except HTTPError as e:
assert_equal(e.code, http.FORBIDDEN)
else:
self.fail()
def test_check_draft_state_registered_but_deleted(self):
reg = RegistrationFactory()
self.draft.registered_node = reg
reg.is_deleted = True
self.draft.save()
try:
draft_views.check_draft_state(self.draft)
except Exception:
self.fail()
def test_check_draft_state_registered(self):
reg = RegistrationFactory()
self.draft.registered_node = reg
self.draft.save()
try:
draft_views.check_draft_state(self.draft)
except HTTPError as e:
assert_equal(e.code, http.FORBIDDEN)
else:
self.fail()
def test_check_draft_state_registered_and_deleted_and_approved(self):
reg = RegistrationFactory()
self.draft.registered_node = reg
self.draft.save()
reg.is_deleted = True
reg.save()
with mock.patch('osf.models.DraftRegistration.is_approved', mock.PropertyMock(return_value=True)):
try:
draft_views.check_draft_state(self.draft)
except HTTPError:
self.fail()
def test_check_draft_state_ok(self):
try:
draft_views.check_draft_state(self.draft)
except Exception:
self.fail()
