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