mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-26 11:33:55 +03:00
42 lines
806 B
Python
42 lines
806 B
Python
import graphene
|
|
|
|
|
|
class User(graphene.ObjectType):
|
|
id = graphene.ID()
|
|
name = graphene.String()
|
|
|
|
|
|
class Query(graphene.ObjectType):
|
|
me = graphene.Field(User)
|
|
|
|
@graphene.annotate(context=graphene.Context)
|
|
def resolve_me(self, context):
|
|
return context['user']
|
|
|
|
|
|
schema = graphene.Schema(query=Query)
|
|
query = '''
|
|
query something{
|
|
me {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
'''
|
|
|
|
|
|
def test_query():
|
|
result = schema.execute(query, context_value={'user': User(id='1', name='Syrus')})
|
|
assert not result.errors
|
|
assert result.data == {
|
|
'me': {
|
|
'id': '1',
|
|
'name': 'Syrus',
|
|
}
|
|
}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
result = schema.execute(query, context_value={'user': User(id='X', name='Console')})
|
|
print(result.data['me'])
|