def test_filter_fields(self, mock_super_get_filter_class):
"""
Should return super call if view has filter_fields attr
"""
mock_view = Mock(filter_fields='abides')
filter_backend = filters.TimestampedFilterBackend()
eq_(filter_backend.get_filter_class(mock_view), 'The Dude')
def test_no_queryset(self, mock_super_get_filter_class):
"""
Should return None if queryset is None (the default)
"""
view = 'nice'
filter_backend = filters.TimestampedFilterBackend()
eq_(filter_backend.get_filter_class(view), None)
ok_(not mock_super_get_filter_class.called)
def test_non_timestampedmodel(self, mock_super_get_filter_class):
"""
Should return None if queryset.model is not a subclass of
models.TimeStampedModel
"""
view = 'nice'
queryset = Mock(model=models.models.Model) # model
filter_backend = filters.TimestampedFilterBackend()
eq_(filter_backend.get_filter_class(view, queryset=queryset), None)
ok_(not mock_super_get_filter_class.called)
def test_exclude_fields(self, mock_super_get_filter_class):
"""
Should not include fields named in the view.
"""
mock_view = Mock(filter_class=None,
filter_fields=None,
filter_fields_exclude=('created_before', 'id'))
queryset = Mock(model=TimeStampedModelSubclass)
filter_backend = filters.TimestampedFilterBackend()
filter_class = filter_backend.get_filter_class(mock_view,
queryset=queryset)
eq_(filter_class.Meta.model, TimeStampedModelSubclass)
eq_(filter_class.Meta.fields,
['created_after', 'modified_before', 'modified_after', 'test'])
eq_(mock_super_get_filter_class.called, 0)
def test_default(self, mock_super_get_filter_class):
"""
Should return a subclass of the default_filter_set instance
attr with the inner Meta class model attr equal to the queryset
model and fields equal to all of the model fields except
created and modified, and in addition the created_before,
created_after, modified_before, and modified_after fields
"""
view = 'nice'
queryset = Mock(model=TimeStampedModelSubclass)
filter_backend = filters.TimestampedFilterBackend()
filter_class = filter_backend.get_filter_class(view, queryset=queryset)
eq_(filter_class.Meta.model, TimeStampedModelSubclass)
eq_(filter_class.Meta.fields, [
'created_before', 'created_after', 'modified_before',
'modified_after', 'id', 'test'
])
ok_(not mock_super_get_filter_class.called)