mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-11 20:27:03 +03:00
71 lines
1.3 KiB
Python
71 lines
1.3 KiB
Python
from py.test import raises
|
|
from collections import namedtuple
|
|
from pytest import raises
|
|
from graphql.core import graphql
|
|
from graphene.core.fields import (
|
|
Field,
|
|
StringField,
|
|
ListField,
|
|
)
|
|
from graphql.core.type import (
|
|
GraphQLObjectType,
|
|
GraphQLSchema,
|
|
GraphQLInterfaceType
|
|
)
|
|
|
|
from graphene.core.types import (
|
|
Interface,
|
|
ObjectType
|
|
)
|
|
|
|
|
|
class Character(Interface):
|
|
name = StringField()
|
|
|
|
|
|
class Pet(ObjectType):
|
|
type = StringField(resolve=lambda *_: 'Dog')
|
|
|
|
|
|
class Human(Character):
|
|
friends = ListField(Character)
|
|
pet = Field(Pet)
|
|
|
|
def resolve_name(self, *args):
|
|
return 'Peter'
|
|
|
|
def resolve_friend(self, *args):
|
|
return Human(object())
|
|
|
|
def resolve_pet(self, *args):
|
|
return Pet(object())
|
|
# def resolve_friends(self, *args, **kwargs):
|
|
# return 'HEY YOU!'
|
|
|
|
schema = object()
|
|
|
|
Human_type = Human.internal_type(schema)
|
|
|
|
def test_type():
|
|
assert Human._meta.fields_map['name'].resolve(Human(object()), 1, 2) == 'Peter'
|
|
|
|
def test_query():
|
|
schema = GraphQLSchema(query=Human_type)
|
|
query = '''
|
|
{
|
|
name
|
|
pet {
|
|
type
|
|
}
|
|
}
|
|
'''
|
|
expected = {
|
|
'name': 'Peter',
|
|
'pet': {
|
|
'type': 'Dog'
|
|
}
|
|
}
|
|
result = graphql(schema, query, root=Human(object()))
|
|
assert not result.errors
|
|
assert result.data == expected
|