def setUp(self):
self.request = Mock()
self.view = Mock()
self.backend = InequalityFilterBackend()
self.queryset = Mock()
self.queryset.filter.return_value = self.queryset
class TestInequalityFilterBackend(TestCase):
def setUp(self):
self.request = Mock()
self.view = Mock()
self.backend = InequalityFilterBackend()
self.queryset = Mock()
self.queryset.filter.return_value = self.queryset
def test_gt_whitelist(self):
"""gt works, and that non-whitelisted variables don't get filtered."""
self.view.filter_fields = ['x']
# `x` should be filtered, but `y` should not, since it is not in
# `filter_fields`
self.request.QUERY_PARAMS = {'x__gt': 10, 'y': 5}
self.backend.filter_queryset(self.request, self.queryset, self.view)
self.queryset.filter.assert_called_with(x__gt=10)
def test_lt_gte_multiple(self):
"""multiple fields, gte, and lt."""
self.view.filter_fields = ['x', 'y']
self.request.QUERY_PARAMS = {'x__gte': 10, 'y__lt': 5}
self.backend.filter_queryset(self.request, self.queryset, self.view)
calls = sorted(self.queryset.method_calls)
# Since both variables are in `filter_fields`, they both get processed.
expected = [('filter', (), {
'x__gte': 10
}), ('filter', (), {
'y__lt': 5
})]
eq_(calls, expected)
class TestInequalityFilterBackend(TestCase):
def setUp(self):
self.request = Mock()
self.view = Mock()
self.backend = InequalityFilterBackend()
self.queryset = Mock()
self.queryset.filter.return_value = self.queryset
def test_gt_whitelist(self):
"""gt works, and that non-whitelisted variables don't get filtered."""
self.view.filter_fields = ['x']
# `x` should be filtered, but `y` should not, since it is not in
# `filter_fields`
self.request.QUERY_PARAMS = {'x__gt': 10, 'y': 5}
self.backend.filter_queryset(self.request, self.queryset, self.view)
self.queryset.filter.assert_called_with(x__gt=10)
def test_lt_gte_multiple(self):
"""multiple fields, gte, and lt."""
self.view.filter_fields = ['x', 'y']
self.request.QUERY_PARAMS = {'x__gte': 10, 'y__lt': 5}
self.backend.filter_queryset(self.request, self.queryset, self.view)
calls = sorted(self.queryset.method_calls)
# Since both variables are in `filter_fields`, they both get processed.
expected = [('filter', (), {'x__gte': 10}),
('filter', (), {'y__lt': 5})]
eq_(calls, expected)
def setUp(self):
self.request = Mock()
self.view = Mock()
self.backend = InequalityFilterBackend()
self.queryset = Mock()
self.queryset.filter.return_value = self.queryset
