diff --git a/examples/context_example.py b/examples/context_example.py new file mode 100644 index 00000000..058e578b --- /dev/null +++ b/examples/context_example.py @@ -0,0 +1,39 @@ +import graphene + + +class User(graphene.ObjectType): + id = graphene.ID() + name = graphene.String() + + +class Query(graphene.ObjectType): + me = graphene.Field(User) + + def resolve_me(self, args, context, info): + 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'])