def test_bucket_get_by_id_raises_exception_if_object_does_not_exist(
self, mock_get_by_id):
# Arrange
mock_absent_id = ObjectId()
mock_get_by_id.side_effect = exceptions.DoesNotExist('')
# Act + Assert
with self.assertRaises(exceptions.DoesNotExist):
bucket_api.get_by_id(mock_absent_id)
def validate_id(self, id):
"""Validate id field
Args:
id:
Returns:
"""
try:
bucket_api.get_by_id(id)
return id
except DoesNotExist:
raise Http404
def test_bucket_get_by_id_returns_bucket(self, mock_get_by_id):
# Arrange
mock_bucket = _create_mock_bucket()
mock_get_by_id.return_value = mock_bucket
# Act
result = bucket_api.get_by_id(mock_bucket.id)
# Assert
self.assertIsInstance(result, Bucket)
def get_object(self, pk):
""" Get bucket from db
Args:
pk: ObjectId
Returns:
Bucket
"""
try:
return bucket_api.get_by_id(pk)
except DoesNotExist:
raise Http404
def delete_bucket(request):
"""Delete a bucket.
Args:
request:
Returns:
"""
try:
bucket_id = request.POST['bucket_id']
bucket = bucket_api.get_by_id(bucket_id)
bucket_api.delete(bucket)
except Exception, e:
return HttpResponseBadRequest(e.message)
def delete_bucket(request):
"""Delete a bucket.
Args:
request:
Returns:
"""
try:
bucket_id = request.POST["bucket_id"]
bucket = bucket_api.get_by_id(bucket_id)
bucket_api.delete(bucket)
except Exception as e:
return HttpResponseBadRequest(escape(str(e)))
return HttpResponse(json.dumps({}), content_type="application/javascript")
