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()