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