graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2024-11-29 21:13:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

Improved Test Framework to support promises as returned GraphQL execution

Browse Source
This commit is contained in:
Syrus Akbary 2017-06-16 11:09:36 -07:00
parent fccc22b651
commit 645bfddfe9
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
graphene/test/__init__.py
View File

@ -1,3 +1,4 @@
from promise import Promise, is_thenable
import six import six
from graphql.error import format_error as format_graphql_error from graphql.error import format_error as format_graphql_error
from graphql.error import GraphQLError from graphql.error import GraphQLError
@ -31,9 +32,16 @@ class Client(object):
self.schema = schema self.schema = schema
self.execute_options = execute_options self.execute_options = execute_options
self.format_error = format_error or default_format_error self.format_error = format_error or default_format_error
def execute(self, *args, **kwargs): def format_result(self, result):
return format_execution_result( return format_execution_result(
self.schema.execute(*args, **dict(self.execute_options, **kwargs)), result,
self.format_error self.format_error
) )
def execute(self, *args, **kwargs):
executed = self.schema.execute(*args, **dict(self.execute_options, **kwargs))
if is_thenable(executed):
return Promise.resolve(executed).then(self.format_result)
return self.format_result(executed)