Improved mutation with thenable check

This commit is contained in:
Syrus Akbary 2017-07-23 17:36:20 -07:00
parent b892eee0ae
commit b185f4cae7

View File

@ -1,7 +1,7 @@
import re
from collections import OrderedDict
from promise import Promise
from promise import Promise, is_thenable
from ..types import Field, InputObjectType, String
from ..types.mutation import Mutation
@ -61,5 +61,8 @@ class ClientIDMutation(Mutation):
).format(repr(payload)))
return payload
return Promise.resolve(
cls.mutate_and_get_payload(input, context, info)).then(on_resolve)
result = cls.mutate_and_get_payload(input, context, info)
if is_thenable(result):
return Promise.resolve(result).then(on_resolve)
return on_resolve(result)