graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-07-22 13:59:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: added async call to test client

Browse Source
This commit is contained in:
Cameron Hurst 2021-01-05 08:10:50 -05:00
parent 34b07e6209
commit 64a6afc92e
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
graphene/test/__init__.py
View File

@ -1,6 +1,5 @@
from graphql.error import format_error as format_graphql_error
from graphql.error import GraphQLError
from graphql.pyutils import is_awaitable
from graphene.types.schema import Schema
@ -32,14 +31,10 @@ class Client:
def execute(self, *args, **kwargs):
executed = self.schema.execute(*args, **dict(self.execute_options, **kwargs))
if is_awaitable(executed):
executed = asyncio.run(executed)
return self.format_result(executed)
async def execute_async(self, *args, **kwargs):
executed = self.schema.execute(*args, **dict(self.execute_options, **kwargs))
if is_awaitable(executed):
executed = await executed
executed = await self.schema.execute_async(*args, **dict(self.execute_options, **kwargs))
return self.format_result(executed)