Fix model property

This commit is contained in:
Jonathan Kim 2019-08-04 08:27:16 +01:00
parent 59f4f134b5
commit 609baec3a0
2 changed files with 22 additions and 1 deletions

View File

@ -20,7 +20,10 @@ class DjangoListField(Field):
@property
def model(self):
return self.type.of_type._meta.node._meta.model
_type = self.type.of_type
if isinstance(_type, NonNull):
_type = _type.of_type
return _type._meta.model
@staticmethod
def list_resolver(resolver, root, info, **args):

View File

@ -0,0 +1,18 @@
import pytest
from graphene import ObjectType, Schema
from ..fields import DjangoListField
from ..types import DjangoObjectType
from .models import Reporter as ReporterModel
@pytest.mark.django_db
class TestDjangoListField:
def test_get_django_model(self):
class Reporter(DjangoObjectType):
class Meta:
model = ReporterModel
fields = ("first_name",)
list_field = DjangoListField(Reporter)
assert list_field.model is ReporterModel