def test_put_counts_in_listings_endpoint(self):
queryset = models.Listing.objects.all()
data = model_access.put_counts_in_listings_endpoint(queryset)
self.assertTrue(data['total'] > 5)
self.assertTrue(data['enabled'] > 2)
self.assertTrue(data['organizations']['1'] > 2)
self.assertTrue(data['IN_PROGRESS'] >= 0)
self.assertTrue(data['PENDING'] >= 0)
self.assertTrue(data['REJECTED'] >= 0)
self.assertTrue(data['APPROVED_ORG'] >= 0)
self.assertTrue(data['APPROVED'] >= 0)
def test_put_counts_in_listings_endpoint(self):
queryset = models.Listing.objects.all()
data = model_access.put_counts_in_listings_endpoint(queryset)
self.assertTrue(data['total'] > 5)
self.assertTrue(data['enabled'] > 2)
self.assertTrue(data['organizations']['1'] > 2)
self.assertTrue(data['IN_PROGRESS'] >= 0)
self.assertTrue(data['PENDING'] >= 0)
self.assertTrue(data['REJECTED'] >= 0)
self.assertTrue(data['APPROVED_ORG'] >= 0)
self.assertTrue(data['APPROVED'] >= 0)
self.assertTrue(data['DELETED'] >= 0)
def list(self, request):
queryset = self.get_queryset()
counts_data = model_access.put_counts_in_listings_endpoint(queryset)
# it appears that because we override the queryset here, we must
# manually invoke the pagination methods
page = self.paginate_queryset(queryset)
if page is not None:
serializer = serializers.ListingSerializer(page,
context={'request': request}, many=True)
r = self.get_paginated_response(serializer.data)
# add counts to response
r.data['counts'] = counts_data
return r
serializer = serializers.ListingSerializer(queryset,
context={'request': request}, many=True)
r = Response(serializer.data)
# add counts to response
counts = {'counts': counts_data}
r.data.append(counts)
return r
def list(self, request):
queryset = self.filter_queryset(self.get_queryset())
counts_data = model_access.put_counts_in_listings_endpoint(queryset)
# it appears that because we override the queryset here, we must
# manually invoke the pagination methods
page = self.paginate_queryset(queryset)
if page is not None:
serializer = serializers.ListingSerializer(
page, context={'request': request}, many=True)
r = self.get_paginated_response(serializer.data)
# add counts to response
r.data['counts'] = counts_data
return r
serializer = serializers.ListingSerializer(
queryset, context={'request': request}, many=True)
r = Response(serializer.data)
# add counts to response
counts = {'counts': counts_data}
r.data.append(counts)
return r
