def test_get_object_list_with_limit(self):
"""
test collecting object_list when there is limit set
"""
resource = mongoengine.CollectionResource(objects_class=None,
max_limit=2)
object_list = resource.get_object_list(queryset=[1, 2, 3, 4],
limit=3,
offset=None)
self.assertEqual([1, 2], object_list)
def test_get_object_list_limit_and_offset(self):
"""
test collecting object_list when there is limit and offset set in a same time
"""
resource = mongoengine.CollectionResource(objects_class=None,
max_limit=2)
object_list = resource.get_object_list(queryset=[1, 2, 3, 4],
limit=3,
offset=1)
self.assertEqual([2, 3], object_list)
def test_on_head(self):
"""
need to check if status of the response is set for 200 and
"""
env = create_environ(path='/')
req = Request(env)
req.context = {'doc': {}}
req.params[mongoengine.CollectionResource.PARAM_TOTAL_COUNT] = '1'
resp = Response()
resource = mongoengine.CollectionResource(
objects_class=Mock(return_value=[1, 2, 3]), max_limit=2)
resource.get_object_list = Mock(return_value=[1, 2])
resource.get_total_objects = Mock(return_value={'total_count': 3})
resource.on_head(req=req, resp=resp)
self.assertIsNot(resp.body, [1, 2, 3])
self.assertEqual(resp.get_header('x-api-total'), '3')
self.assertEqual(resp.get_header('x-api-returned'), '2')
def test_on_options(self):
"""
need to check if status of the response is set for 200 and
"""
env = create_environ(path='/')
req = Request(env)
req.context = {'doc': {}}
resp = Response()
resource = mongoengine.CollectionResource(objects_class=FakeObjectList,
max_limit=2)
resource.on_options(req=req, resp=resp)
self.assertEqual(resp.get_header('Allow'),
'GET, HEAD, OPTIONS, POST, PUT')
self.assertEqual(
resp.body, {
'name': 'FakeObjectList',
'description':
'Extend list to match interface of List resource',
})