mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-11 20:27:03 +03:00
109 lines
2.0 KiB
Python
109 lines
2.0 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 import (
|
|
Interface,
|
|
ObjectType,
|
|
Schema
|
|
)
|
|
|
|
|
|
schema = Schema(name='My own schema')
|
|
|
|
|
|
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())
|
|
|
|
schema.query = Human
|
|
|
|
|
|
def test_get_registered_type():
|
|
assert schema.get_type('Character') == Character
|
|
|
|
|
|
def test_get_unregistered_type():
|
|
with raises(Exception) as excinfo:
|
|
schema.get_type('NON_EXISTENT_MODEL')
|
|
assert 'not found' in str(excinfo.value)
|
|
|
|
|
|
def test_schema_query():
|
|
assert schema.query == Human
|
|
|
|
|
|
def test_query_schema_graphql():
|
|
a = object()
|
|
query = '''
|
|
{
|
|
name
|
|
pet {
|
|
type
|
|
}
|
|
}
|
|
'''
|
|
expected = {
|
|
'name': 'Peter',
|
|
'pet': {
|
|
'type': 'Dog'
|
|
}
|
|
}
|
|
result = graphql(schema.schema, query, root=Human(object()))
|
|
assert not result.errors
|
|
assert result.data == expected
|
|
|
|
|
|
def test_query_schema_execute():
|
|
a = object()
|
|
query = '''
|
|
{
|
|
name
|
|
pet {
|
|
type
|
|
}
|
|
}
|
|
'''
|
|
expected = {
|
|
'name': 'Peter',
|
|
'pet': {
|
|
'type': 'Dog'
|
|
}
|
|
}
|
|
result = schema.execute(query, root=object())
|
|
assert not result.errors
|
|
assert result.data == expected
|
|
|
|
|
|
def test_schema_get_type_map():
|
|
assert schema.schema.get_type_map().keys() == [
|
|
'__Field', 'String', 'Pet', 'Character', '__InputValue', '__Directive', '__TypeKind', '__Schema', '__Type', 'Human', '__EnumValue', 'Boolean']
|