|
5 | 5 | from django.utils.functional import SimpleLazyObject
|
6 | 6 | from py.test import raises
|
7 | 7 |
|
| 8 | +from django_filters import FilterSet, NumberFilter |
| 9 | + |
8 | 10 | import graphene
|
9 | 11 | from graphene.relay import Node
|
10 | 12 |
|
11 | 13 | from ..utils import DJANGO_FILTER_INSTALLED
|
12 | 14 | from ..compat import MissingType, JSONField
|
13 | 15 | from ..fields import DjangoConnectionField
|
| 16 | +from ..filter.fields import DjangoFilterConnectionField |
14 | 17 | from ..types import DjangoObjectType
|
15 | 18 | from .models import Article, Reporter
|
16 | 19 |
|
@@ -42,7 +45,6 @@ class Meta:
|
42 | 45 | model = Reporter
|
43 | 46 | only_fields = ('id', )
|
44 | 47 |
|
45 |
| - |
46 | 48 | class Query(graphene.ObjectType):
|
47 | 49 | reporter = graphene.Field(ReporterType)
|
48 | 50 |
|
@@ -360,7 +362,110 @@ class Query(graphene.ObjectType):
|
360 | 362 | }]
|
361 | 363 | }
|
362 | 364 | }
|
363 |
| - |
| 365 | + |
| 366 | + result = schema.execute(query) |
| 367 | + assert not result.errors |
| 368 | + assert result.data == expected |
| 369 | + |
| 370 | + |
| 371 | +def test_should_query_filter_node_limit(): |
| 372 | + class ReporterFilter(FilterSet): |
| 373 | + limit = NumberFilter(method='filter_limit') |
| 374 | + |
| 375 | + def filter_limit(self, queryset, name, value): |
| 376 | + return queryset[:value] |
| 377 | + |
| 378 | + class Meta: |
| 379 | + model = Reporter |
| 380 | + fields = ['first_name', ] |
| 381 | + |
| 382 | + class ReporterType(DjangoObjectType): |
| 383 | + |
| 384 | + class Meta: |
| 385 | + model = Reporter |
| 386 | + interfaces = (Node, ) |
| 387 | + |
| 388 | + class ArticleType(DjangoObjectType): |
| 389 | + |
| 390 | + class Meta: |
| 391 | + model = Article |
| 392 | + interfaces = (Node, ) |
| 393 | + filter_fields = ('lang', ) |
| 394 | + |
| 395 | + class Query(graphene.ObjectType): |
| 396 | + all_reporters = DjangoFilterConnectionField( |
| 397 | + ReporterType, |
| 398 | + filterset_class=ReporterFilter |
| 399 | + ) |
| 400 | + |
| 401 | + def resolve_all_reporters(self, args, context, info): |
| 402 | + return Reporter.objects.all() |
| 403 | + |
| 404 | + r = Reporter.objects.create( |
| 405 | + first_name='John', |
| 406 | + last_name='Doe', |
| 407 | + |
| 408 | + a_choice=1 |
| 409 | + ) |
| 410 | + Reporter.objects.create( |
| 411 | + first_name='Bob', |
| 412 | + last_name='Doe', |
| 413 | + |
| 414 | + a_choice=1 |
| 415 | + ) |
| 416 | + |
| 417 | + Article.objects.create( |
| 418 | + headline='Article Node 1', |
| 419 | + pub_date=datetime.date.today(), |
| 420 | + reporter=r, |
| 421 | + editor=r, |
| 422 | + lang='es' |
| 423 | + ) |
| 424 | + Article.objects.create( |
| 425 | + headline='Article Node 2', |
| 426 | + pub_date=datetime.date.today(), |
| 427 | + reporter=r, |
| 428 | + editor=r, |
| 429 | + lang='en' |
| 430 | + ) |
| 431 | + |
| 432 | + schema = graphene.Schema(query=Query) |
| 433 | + query = ''' |
| 434 | + query NodeFilteringQuery { |
| 435 | + allReporters(limit: 1) { |
| 436 | + edges { |
| 437 | + node { |
| 438 | + id |
| 439 | + articles(lang: "es") { |
| 440 | + edges { |
| 441 | + node { |
| 442 | + id |
| 443 | + } |
| 444 | + } |
| 445 | + } |
| 446 | + } |
| 447 | + } |
| 448 | + } |
| 449 | + } |
| 450 | + ''' |
| 451 | + |
| 452 | + expected = { |
| 453 | + 'allReporters': { |
| 454 | + 'edges': [{ |
| 455 | + 'node': { |
| 456 | + 'id': 'UmVwb3J0ZXJUeXBlOjE=', |
| 457 | + 'articles': { |
| 458 | + 'edges': [{ |
| 459 | + 'node': { |
| 460 | + 'id': 'QXJ0aWNsZVR5cGU6MQ==' |
| 461 | + } |
| 462 | + }] |
| 463 | + } |
| 464 | + } |
| 465 | + }] |
| 466 | + } |
| 467 | + } |
| 468 | + |
364 | 469 | result = schema.execute(query)
|
365 | 470 | assert not result.errors
|
366 | 471 | assert result.data == expected
|
0 commit comments