Skip to content

Commit

Permalink
Merge pull request encode#1836 from jpadilla/filter-ordering
Browse files Browse the repository at this point in the history
Preserve default ordering when using filtering
  • Loading branch information
tomchristie committed Sep 5, 2014
2 parents 2b47c6b + ab213cb commit f4e0244
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 13 deletions.
1 change: 0 additions & 1 deletion rest_framework/filters.py
Expand Up @@ -56,7 +56,6 @@ class AutoFilterSet(self.default_filter_set):
class Meta:
model = queryset.model
fields = filter_fields
order_by = True
return AutoFilterSet

return None
Expand Down
69 changes: 57 additions & 12 deletions tests/test_filters.py
Expand Up @@ -408,16 +408,61 @@ class SearchListView(generics.ListAPIView):
)


class OrdringFilterModel(models.Model):
class OrderingFilterModel(models.Model):
title = models.CharField(max_length=20)
text = models.CharField(max_length=100)


class OrderingFilterRelatedModel(models.Model):
related_object = models.ForeignKey(OrdringFilterModel,
related_object = models.ForeignKey(OrderingFilterModel,
related_name="relateds")


class DjangoFilterOrderingModel(models.Model):
date = models.DateField()
text = models.CharField(max_length=10)

class Meta:
ordering = ['-date']


class DjangoFilterOrderingTests(TestCase):
def setUp(self):
data = [{
'date': datetime.date(2012, 10, 8),
'text': 'abc'
}, {
'date': datetime.date(2013, 10, 8),
'text': 'bcd'
}, {
'date': datetime.date(2014, 10, 8),
'text': 'cde'
}]

for d in data:
DjangoFilterOrderingModel.objects.create(**d)

def test_default_ordering(self):
class DjangoFilterOrderingView(generics.ListAPIView):
model = DjangoFilterOrderingModel
filter_backends = (filters.DjangoFilterBackend,)
filter_fields = ['text']
ordering = ('-date',)

view = DjangoFilterOrderingView.as_view()
request = factory.get('/')
response = view(request)

self.assertEqual(
response.data,
[
{'id': 3, 'date': datetime.date(2014, 10, 8), 'text': 'cde'},
{'id': 2, 'date': datetime.date(2013, 10, 8), 'text': 'bcd'},
{'id': 1, 'date': datetime.date(2012, 10, 8), 'text': 'abc'}
]
)


class OrderingFilterTests(TestCase):
def setUp(self):
# Sequence of title/text is:
Expand All @@ -436,11 +481,11 @@ def setUp(self):
chr(idx + ord('b')) +
chr(idx + ord('c'))
)
OrdringFilterModel(title=title, text=text).save()
OrderingFilterModel(title=title, text=text).save()

def test_ordering(self):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = ('title',)
ordering_fields = ('text',)
Expand All @@ -459,7 +504,7 @@ class OrderingListView(generics.ListAPIView):

def test_reverse_ordering(self):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = ('title',)
ordering_fields = ('text',)
Expand All @@ -478,7 +523,7 @@ class OrderingListView(generics.ListAPIView):

def test_incorrectfield_ordering(self):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = ('title',)
ordering_fields = ('text',)
Expand All @@ -497,7 +542,7 @@ class OrderingListView(generics.ListAPIView):

def test_default_ordering(self):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = ('title',)
oredering_fields = ('text',)
Expand All @@ -516,7 +561,7 @@ class OrderingListView(generics.ListAPIView):

def test_default_ordering_using_string(self):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = 'title'
ordering_fields = ('text',)
Expand All @@ -536,7 +581,7 @@ class OrderingListView(generics.ListAPIView):
def test_ordering_by_aggregate_field(self):
# create some related models to aggregate order by
num_objs = [2, 5, 3]
for obj, num_relateds in zip(OrdringFilterModel.objects.all(),
for obj, num_relateds in zip(OrderingFilterModel.objects.all(),
num_objs):
for _ in range(num_relateds):
new_related = OrderingFilterRelatedModel(
Expand All @@ -545,11 +590,11 @@ def test_ordering_by_aggregate_field(self):
new_related.save()

class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = 'title'
ordering_fields = '__all__'
queryset = OrdringFilterModel.objects.all().annotate(
queryset = OrderingFilterModel.objects.all().annotate(
models.Count("relateds"))

view = OrderingListView.as_view()
Expand All @@ -567,7 +612,7 @@ class OrderingListView(generics.ListAPIView):
def test_ordering_with_nonstandard_ordering_param(self):
with temporary_setting('ORDERING_PARAM', 'order', filters):
class OrderingListView(generics.ListAPIView):
model = OrdringFilterModel
model = OrderingFilterModel
filter_backends = (filters.OrderingFilter,)
ordering = ('title',)
ordering_fields = ('text',)
Expand Down

0 comments on commit f4e0244

Please sign in to comment.