def test_get_queryset_sort_desc(self):
"""Test get_queryset with sort desc."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.request = self.request_factory.get('/?sort=desc')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['-pk'])
def test_get_queryset_sort_desc(self):
"""Test get_queryset with sort desc."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.request = self.request_factory.get('/?sort=desc')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['-pk'])
def test_get_queryset_order_by_valid(self):
"""Test get_queryset with a valid order_by value."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.valid_order_by = ['cookies']
mixin.request = self.request_factory.get('/?order_by=cookies')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['cookies'])
def test_get_queryset_sort_not_valid(self):
"""Test get_queryset with an invalid sort option."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.default_sort = 'asc'
mixin.request = self.request_factory.get('/?sort=blah')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['pk'])
def test_get_queryset_order_by_valid(self):
"""Test get_queryset with a valid order_by value."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.valid_order_by = ['cookies']
mixin.request = self.request_factory.get('/?order_by=cookies')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['cookies'])
def test_get_queryset_sort_not_valid(self):
"""Test get_queryset with an invalid sort option."""
mixin = SortableListMixin()
mixin.model = FakeModel
mixin.default_sort = 'asc'
mixin.request = self.request_factory.get('/?sort=blah')
response = mixin.get_queryset()
self.assertEqual(response.query.order_by, ['pk'])
