Skip to content

Commit b1bffc4

Browse files
authored
Added context example
1 parent 6c7cd4e commit b1bffc4

File tree

1 file changed

+39
-0
lines changed

1 file changed

+39
-0
lines changed

examples/context_example.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
import graphene
2+
3+
4+
class User(graphene.ObjectType):
5+
id = graphene.ID()
6+
name = graphene.String()
7+
8+
9+
class Query(graphene.ObjectType):
10+
me = graphene.Field(User)
11+
12+
def resolve_me(self, args, context, info):
13+
return context['user']
14+
15+
schema = graphene.Schema(query=Query)
16+
query = '''
17+
query something{
18+
me {
19+
id
20+
name
21+
}
22+
}
23+
'''
24+
25+
26+
def test_query():
27+
result = schema.execute(query, context_value={'user': User(id='1', name='Syrus')})
28+
assert not result.errors
29+
assert result.data == {
30+
'me': {
31+
'id': '1',
32+
'name': 'Syrus',
33+
}
34+
}
35+
36+
37+
if __name__ == '__main__':
38+
result = schema.execute(query, context_value={'user': User(id='X', name='Console')})
39+
print(result.data['me'])

0 commit comments

Comments
 (0)