mirror of
https://github.com/graphql-python/graphene-django.git
synced 2025-06-14 10:33:11 +03:00
Fix filtering with a resolver and DjangoFilter filter.
This commit is contained in:
parent
1139507a14
commit
7085437a12
|
@ -54,7 +54,7 @@ class DjangoConnectionField(ConnectionField):
|
||||||
iterable = maybe_queryset(iterable)
|
iterable = maybe_queryset(iterable)
|
||||||
if isinstance(iterable, QuerySet):
|
if isinstance(iterable, QuerySet):
|
||||||
if iterable is not default_manager:
|
if iterable is not default_manager:
|
||||||
iterable &= maybe_queryset(default_manager)
|
iterable = maybe_queryset(default_manager)
|
||||||
_len = iterable.count()
|
_len = iterable.count()
|
||||||
else:
|
else:
|
||||||
_len = len(iterable)
|
_len = len(iterable)
|
||||||
|
|
|
@ -5,12 +5,15 @@ from django.db import models
|
||||||
from django.utils.functional import SimpleLazyObject
|
from django.utils.functional import SimpleLazyObject
|
||||||
from py.test import raises
|
from py.test import raises
|
||||||
|
|
||||||
|
from django_filters import FilterSet, NumberFilter
|
||||||
|
|
||||||
import graphene
|
import graphene
|
||||||
from graphene.relay import Node
|
from graphene.relay import Node
|
||||||
|
|
||||||
from ..utils import DJANGO_FILTER_INSTALLED
|
from ..utils import DJANGO_FILTER_INSTALLED
|
||||||
from ..compat import MissingType, JSONField
|
from ..compat import MissingType, JSONField
|
||||||
from ..fields import DjangoConnectionField
|
from ..fields import DjangoConnectionField
|
||||||
|
from ..filter.fields import DjangoFilterConnectionField
|
||||||
from ..types import DjangoObjectType
|
from ..types import DjangoObjectType
|
||||||
from .models import Article, Reporter
|
from .models import Article, Reporter
|
||||||
|
|
||||||
|
@ -42,7 +45,6 @@ def test_should_query_simplelazy_objects():
|
||||||
model = Reporter
|
model = Reporter
|
||||||
only_fields = ('id', )
|
only_fields = ('id', )
|
||||||
|
|
||||||
|
|
||||||
class Query(graphene.ObjectType):
|
class Query(graphene.ObjectType):
|
||||||
reporter = graphene.Field(ReporterType)
|
reporter = graphene.Field(ReporterType)
|
||||||
|
|
||||||
|
@ -364,3 +366,106 @@ def test_should_query_node_filtering():
|
||||||
result = schema.execute(query)
|
result = schema.execute(query)
|
||||||
assert not result.errors
|
assert not result.errors
|
||||||
assert result.data == expected
|
assert result.data == expected
|
||||||
|
|
||||||
|
|
||||||
|
def test_should_query_filter_node_limit():
|
||||||
|
class ReporterFilter(FilterSet):
|
||||||
|
limit = NumberFilter(method='filter_limit')
|
||||||
|
|
||||||
|
def filter_limit(self, queryset, name, value):
|
||||||
|
return queryset[:value]
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = Reporter
|
||||||
|
fields = ['first_name', ]
|
||||||
|
|
||||||
|
class ReporterType(DjangoObjectType):
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = Reporter
|
||||||
|
interfaces = (Node, )
|
||||||
|
|
||||||
|
class ArticleType(DjangoObjectType):
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = Article
|
||||||
|
interfaces = (Node, )
|
||||||
|
filter_fields = ('lang', )
|
||||||
|
|
||||||
|
class Query(graphene.ObjectType):
|
||||||
|
all_reporters = DjangoFilterConnectionField(
|
||||||
|
ReporterType,
|
||||||
|
filterset_class=ReporterFilter
|
||||||
|
)
|
||||||
|
|
||||||
|
def resolve_all_reporters(self, args, context, info):
|
||||||
|
return Reporter.objects.all()
|
||||||
|
|
||||||
|
r = Reporter.objects.create(
|
||||||
|
first_name='John',
|
||||||
|
last_name='Doe',
|
||||||
|
email='johndoe@example.com',
|
||||||
|
a_choice=1
|
||||||
|
)
|
||||||
|
Reporter.objects.create(
|
||||||
|
first_name='Bob',
|
||||||
|
last_name='Doe',
|
||||||
|
email='bobdoe@example.com',
|
||||||
|
a_choice=1
|
||||||
|
)
|
||||||
|
|
||||||
|
Article.objects.create(
|
||||||
|
headline='Article Node 1',
|
||||||
|
pub_date=datetime.date.today(),
|
||||||
|
reporter=r,
|
||||||
|
editor=r,
|
||||||
|
lang='es'
|
||||||
|
)
|
||||||
|
Article.objects.create(
|
||||||
|
headline='Article Node 2',
|
||||||
|
pub_date=datetime.date.today(),
|
||||||
|
reporter=r,
|
||||||
|
editor=r,
|
||||||
|
lang='en'
|
||||||
|
)
|
||||||
|
|
||||||
|
schema = graphene.Schema(query=Query)
|
||||||
|
query = '''
|
||||||
|
query NodeFilteringQuery {
|
||||||
|
allReporters(limit: 1) {
|
||||||
|
edges {
|
||||||
|
node {
|
||||||
|
id
|
||||||
|
articles(lang: "es") {
|
||||||
|
edges {
|
||||||
|
node {
|
||||||
|
id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
'''
|
||||||
|
|
||||||
|
expected = {
|
||||||
|
'allReporters': {
|
||||||
|
'edges': [{
|
||||||
|
'node': {
|
||||||
|
'id': 'UmVwb3J0ZXJUeXBlOjE=',
|
||||||
|
'articles': {
|
||||||
|
'edges': [{
|
||||||
|
'node': {
|
||||||
|
'id': 'QXJ0aWNsZVR5cGU6MQ=='
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result = schema.execute(query)
|
||||||
|
assert not result.errors
|
||||||
|
assert result.data == expected
|
||||||
|
|
Loading…
Reference in New Issue
Block a user