mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-02-02 20:54:42 +03:00
commit
5358243ca5
|
@ -44,7 +44,7 @@ class DjangoFilterBackend(BaseFilterBackend):
|
||||||
if filter_class:
|
if filter_class:
|
||||||
filter_model = filter_class.Meta.model
|
filter_model = filter_class.Meta.model
|
||||||
|
|
||||||
assert issubclass(filter_model, queryset.model), \
|
assert issubclass(queryset.model, filter_model), \
|
||||||
'FilterSet model %s does not match queryset model %s' % \
|
'FilterSet model %s does not match queryset model %s' % \
|
||||||
(filter_model, queryset.model)
|
(filter_model, queryset.model)
|
||||||
|
|
||||||
|
|
|
@ -60,6 +60,18 @@ class ReadOnlyManyToManyModel(RESTFrameworkModel):
|
||||||
rel = models.ManyToManyField(Anchor)
|
rel = models.ManyToManyField(Anchor)
|
||||||
|
|
||||||
|
|
||||||
|
class BaseFilterableItem(RESTFrameworkModel):
|
||||||
|
text = models.CharField(max_length=100)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
abstract = True
|
||||||
|
|
||||||
|
|
||||||
|
class FilterableItem(BaseFilterableItem):
|
||||||
|
decimal = models.DecimalField(max_digits=4, decimal_places=2)
|
||||||
|
date = models.DateField()
|
||||||
|
|
||||||
|
|
||||||
# Model for regression test for #285
|
# Model for regression test for #285
|
||||||
|
|
||||||
class Comment(RESTFrameworkModel):
|
class Comment(RESTFrameworkModel):
|
||||||
|
@ -172,10 +184,3 @@ class NullableOneToOneSource(RESTFrameworkModel):
|
||||||
class BasicModelSerializer(serializers.ModelSerializer):
|
class BasicModelSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = BasicModel
|
model = BasicModel
|
||||||
|
|
||||||
|
|
||||||
# Models to test filters
|
|
||||||
class FilterableItem(models.Model):
|
|
||||||
text = models.CharField(max_length=100)
|
|
||||||
decimal = models.DecimalField(max_digits=4, decimal_places=2)
|
|
||||||
date = models.DateField()
|
|
||||||
|
|
|
@ -9,7 +9,7 @@ from django.conf.urls import patterns, url
|
||||||
from rest_framework import generics, serializers, status, filters
|
from rest_framework import generics, serializers, status, filters
|
||||||
from rest_framework.compat import django_filters
|
from rest_framework.compat import django_filters
|
||||||
from rest_framework.test import APIRequestFactory
|
from rest_framework.test import APIRequestFactory
|
||||||
from .models import FilterableItem, BasicModel
|
from .models import BaseFilterableItem, FilterableItem, BasicModel
|
||||||
from .utils import temporary_setting
|
from .utils import temporary_setting
|
||||||
|
|
||||||
factory = APIRequestFactory()
|
factory = APIRequestFactory()
|
||||||
|
@ -55,6 +55,18 @@ if django_filters:
|
||||||
filter_class = SeveralFieldsFilter
|
filter_class = SeveralFieldsFilter
|
||||||
filter_backends = (filters.DjangoFilterBackend,)
|
filter_backends = (filters.DjangoFilterBackend,)
|
||||||
|
|
||||||
|
# These classes are used to test base model filter support
|
||||||
|
class BaseFilterableItemFilter(django_filters.FilterSet):
|
||||||
|
text = django_filters.CharFilter()
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = BaseFilterableItem
|
||||||
|
|
||||||
|
class BaseFilterableItemFilterRootView(generics.ListCreateAPIView):
|
||||||
|
model = FilterableItem
|
||||||
|
filter_class = BaseFilterableItemFilter
|
||||||
|
filter_backends = (filters.DjangoFilterBackend,)
|
||||||
|
|
||||||
# Regression test for #814
|
# Regression test for #814
|
||||||
class FilterableItemSerializer(serializers.ModelSerializer):
|
class FilterableItemSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
|
@ -226,6 +238,18 @@ class IntegrationTestFiltering(CommonFilteringTestCase):
|
||||||
request = factory.get('/')
|
request = factory.get('/')
|
||||||
self.assertRaises(AssertionError, view, request)
|
self.assertRaises(AssertionError, view, request)
|
||||||
|
|
||||||
|
@unittest.skipUnless(django_filters, 'django-filter not installed')
|
||||||
|
def test_base_model_filter(self):
|
||||||
|
"""
|
||||||
|
The `get_filter_class` model checks should allow base model filters.
|
||||||
|
"""
|
||||||
|
view = BaseFilterableItemFilterRootView.as_view()
|
||||||
|
|
||||||
|
request = factory.get('/?text=aaa')
|
||||||
|
response = view(request).render()
|
||||||
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||||||
|
self.assertEqual(len(response.data), 1)
|
||||||
|
|
||||||
@unittest.skipUnless(django_filters, 'django-filter not installed')
|
@unittest.skipUnless(django_filters, 'django-filter not installed')
|
||||||
def test_unknown_filter(self):
|
def test_unknown_filter(self):
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue
Block a user