mirror of
https://github.com/graphql-python/graphene.git
synced 2025-05-05 00:43:41 +03:00
55 lines
1.1 KiB
Python
55 lines
1.1 KiB
Python
# https://github.com/graphql-python/graphene/issues/313
|
|
|
|
import graphene
|
|
from graphene import resolve_only_args
|
|
|
|
class Query(graphene.ObjectType):
|
|
rand = graphene.String()
|
|
|
|
class Success(graphene.ObjectType):
|
|
yeah = graphene.String()
|
|
|
|
|
|
class Error(graphene.ObjectType):
|
|
message = graphene.String()
|
|
|
|
|
|
class CreatePostResult(graphene.Union):
|
|
class Meta:
|
|
types = [Success, Error]
|
|
|
|
|
|
class CreatePost(graphene.Mutation):
|
|
class Input:
|
|
text = graphene.String(required=True)
|
|
|
|
result = graphene.Field(CreatePostResult)
|
|
|
|
@resolve_only_args
|
|
def mutate(self, text):
|
|
result = Success(yeah='yeah')
|
|
|
|
return CreatePost(result=result)
|
|
|
|
|
|
class Mutations(graphene.ObjectType):
|
|
create_post = CreatePost.Field()
|
|
|
|
# tests.py
|
|
|
|
def test_create_post():
|
|
query_string = '''
|
|
mutation {
|
|
createPost(text: "Try this out") {
|
|
result {
|
|
__typename
|
|
}
|
|
}
|
|
}
|
|
'''
|
|
|
|
schema = graphene.Schema(query=Query, mutation=Mutations)
|
|
result = schema.execute(query_string)
|
|
|
|
assert not result.errors
|
|
assert result.data['createPost']['result']['__typename'] == 'Success' |