Fixed tests

This commit is contained in:
Syrus Akbary 2016-11-23 02:28:21 -08:00
parent 2e1f154441
commit b3fbac5818

View File

@ -11,6 +11,7 @@ from graphene_django.tests.models import Article, Pet, Reporter
from graphene_django.utils import DJANGO_FILTER_INSTALLED
pytestmark = []
if DJANGO_FILTER_INSTALLED:
import django_filters
from graphene_django.filter import (GlobalIDFilter, DjangoFilterConnectionField,
@ -22,7 +23,8 @@ else:
pytestmark.append(pytest.mark.django_db)
class ArticleNode(DjangoObjectType):
if DJANGO_FILTER_INSTALLED:
class ArticleNode(DjangoObjectType):
class Meta:
model = Article
@ -30,20 +32,20 @@ class ArticleNode(DjangoObjectType):
filter_fields = ('headline', )
class ReporterNode(DjangoObjectType):
class ReporterNode(DjangoObjectType):
class Meta:
model = Reporter
interfaces = (Node, )
class PetNode(DjangoObjectType):
class PetNode(DjangoObjectType):
class Meta:
model = Pet
interfaces = (Node, )
# schema = Schema()
# schema = Schema()
def get_args(field):