def test_on_post_raises_when_db_replace_session_raises(self):
self.mock_db.replace_session.side_effect = exceptions.AccessForbidden(
'regular test failure')
session = common.get_fake_session_0()
self.mock_json_body.return_value = session
self.assertRaises(exceptions.AccessForbidden, self.resource.on_post,
self.mock_req, self.mock_req,
common.fake_session_0['session_id'])
def enforce(self,
rule,
action,
ctxt,
do_raise=True,
exc=exceptions.AccessForbidden):
if self.rules.get(rule) is False:
raise exceptions.AccessForbidden()
def test_on_post_raises_when_db_replace_job_raises(self):
self.mock_db.replace_job.side_effect = exceptions.AccessForbidden(
'regular test failure')
job = common.get_fake_job_0()
self.mock_req.stream.read.return_value = json.dumps(job)
self.assertRaises(exceptions.AccessForbidden, self.resource.on_post,
self.mock_req, self.mock_req,
common.fake_job_0_job_id)
def get(self, project_id, doc_id, user_id=None):
try:
res = self.es.get(index=self.index,
doc_type=self.doc_type,
id=doc_id)
doc = res['_source']
except elasticsearch.TransportError:
raise freezer_api_exc.DocumentNotFound(
message='No document found with ID:{0}'.format(doc_id))
except Exception as e:
raise freezer_api_exc.StorageEngineError(
message='Get operation failed: {}'.format(e))
if doc['project_id'] != project_id:
raise freezer_api_exc.AccessForbidden("You are not allowed to"
" access")
if user_id:
if doc['user_id'] != user_id:
raise freezer_api_exc.AccessForbidden(
"Document access forbidden")
if '_version' in res:
doc['_version'] = res['_version']
return doc
def get(self, user_id, doc_id):
try:
res = self.es.get(index=self.index,
doc_type=self.doc_type,
id=doc_id)
doc = res['_source']
except elasticsearch.TransportError:
raise freezer_api_exc.DocumentNotFound(
message=_i18n._('No document found with ID %s') % doc_id)
except Exception as e:
raise freezer_api_exc.StorageEngineError(
message=_i18n._('Get operation failed: %s') % e)
if doc['user_id'] != user_id:
raise freezer_api_exc.AccessForbidden(
_i18n._("Document access forbidden"))
if '_version' in res:
doc['_version'] = res['_version']
return doc
def test_AccessForbidden(self):
e = exceptions.AccessForbidden(message='testing')
self.assertRaises(falcon.HTTPForbidden, e.handle, self.ex,
self.mock_req, self.mock_req, None)
