kit-eco.social/consumocuidado-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added object count to my_products

Browse Source
This commit is contained in:
Sam
2021-03-05 10:13:31 +00:00
parent e7366226ea
commit f81bbee4e6
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
products/tests.py
View File

@@ -947,7 +947,8 @@ class MyProductsViewTest(APITestCase):
# Assert forbidden code # Assert forbidden code
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEquals(len(user_instances), len(payload)) self.assertEquals(payload['count'], len(payload['results']))
self.assertEquals(len(user_instances), payload['count'])
def test_auth_user_can_paginate_instances(self): def test_auth_user_can_paginate_instances(self):
"""authenticated user can paginate instances """authenticated user can paginate instances
@@ -967,7 +968,8 @@ class MyProductsViewTest(APITestCase):
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)
# assert only 2 instances in response # assert only 2 instances in response
payload = response.json() payload = response.json()
self.assertEquals(2, len(payload)) self.assertEquals(payload['count'], len(payload['results']))
self.assertEquals(2, payload['count'])
def test_anon_user_cannot_access(self): def test_anon_user_cannot_access(self):
# send in request # send in request

6
products/views.py
View File

@@ -68,7 +68,11 @@ def my_products(request):
elif limit is not None: elif limit is not None:
limit = int(limit) limit = int(limit)
data = data[:limit] data = data[:limit]
return Response(data=data) # prepare response payload
payload = {}
payload['results'] = data
payload['count'] = len(payload['results'])
return Response(data=payload)
@api_view(['POST',]) @api_view(['POST',])