mirror of
https://github.com/graphql-python/graphene.git
synced 2025-07-27 08:19:45 +03:00
Re-add default value for Field.
This commit is contained in:
parent
94d46f7960
commit
fce0b85b8a
|
@ -2,6 +2,8 @@ import inspect
|
||||||
from collections import Mapping, OrderedDict
|
from collections import Mapping, OrderedDict
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
|
from promise import Promise
|
||||||
|
|
||||||
from ..utils.orderedtype import OrderedType
|
from ..utils.orderedtype import OrderedType
|
||||||
from .argument import to_arguments
|
from .argument import to_arguments
|
||||||
from .structures import NonNull
|
from .structures import NonNull
|
||||||
|
@ -18,7 +20,8 @@ class Field(OrderedType):
|
||||||
|
|
||||||
def __init__(self, type, args=None, resolver=None, source=None,
|
def __init__(self, type, args=None, resolver=None, source=None,
|
||||||
deprecation_reason=None, name=None, description=None,
|
deprecation_reason=None, name=None, description=None,
|
||||||
required=False, _creation_counter=None, **extra_args):
|
required=False, _creation_counter=None, default_value=None,
|
||||||
|
**extra_args):
|
||||||
super(Field, self).__init__(_creation_counter=_creation_counter)
|
super(Field, self).__init__(_creation_counter=_creation_counter)
|
||||||
assert not args or isinstance(args, Mapping), (
|
assert not args or isinstance(args, Mapping), (
|
||||||
'Arguments in a field have to be a mapping, received "{}".'
|
'Arguments in a field have to be a mapping, received "{}".'
|
||||||
|
@ -27,7 +30,7 @@ class Field(OrderedType):
|
||||||
'A Field cannot have a source and a resolver in at the same time.'
|
'A Field cannot have a source and a resolver in at the same time.'
|
||||||
)
|
)
|
||||||
|
|
||||||
if required:
|
if required or default_value is not None:
|
||||||
type = NonNull(type)
|
type = NonNull(type)
|
||||||
|
|
||||||
self.name = name
|
self.name = name
|
||||||
|
@ -38,6 +41,7 @@ class Field(OrderedType):
|
||||||
self.resolver = resolver
|
self.resolver = resolver
|
||||||
self.deprecation_reason = deprecation_reason
|
self.deprecation_reason = deprecation_reason
|
||||||
self.description = description
|
self.description = description
|
||||||
|
self.default_value = default_value
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def type(self):
|
def type(self):
|
||||||
|
@ -46,4 +50,14 @@ class Field(OrderedType):
|
||||||
return self._type
|
return self._type
|
||||||
|
|
||||||
def get_resolver(self, parent_resolver):
|
def get_resolver(self, parent_resolver):
|
||||||
return self.resolver or parent_resolver
|
resolver = self.resolver or parent_resolver
|
||||||
|
|
||||||
|
def default_resolve(result):
|
||||||
|
return self.default_value if result is None else result
|
||||||
|
|
||||||
|
def defualt_resolver(self, *args, **kwargs):
|
||||||
|
return Promise.resolve(
|
||||||
|
resolver(*args, **kwargs)
|
||||||
|
).then(default_resolve)
|
||||||
|
|
||||||
|
return defualt_resolver if self.default_value is not None else resolver
|
||||||
|
|
|
@ -16,19 +16,22 @@ def test_field_basic():
|
||||||
resolver = lambda: None
|
resolver = lambda: None
|
||||||
deprecation_reason = 'Deprecated now'
|
deprecation_reason = 'Deprecated now'
|
||||||
description = 'My Field'
|
description = 'My Field'
|
||||||
|
my_default='something'
|
||||||
field = Field(
|
field = Field(
|
||||||
MyType,
|
MyType,
|
||||||
name='name',
|
name='name',
|
||||||
args=args,
|
args=args,
|
||||||
resolver=resolver,
|
resolver=resolver,
|
||||||
description=description,
|
description=description,
|
||||||
deprecation_reason=deprecation_reason
|
deprecation_reason=deprecation_reason,
|
||||||
|
default_value=my_default,
|
||||||
)
|
)
|
||||||
assert field.name == 'name'
|
assert field.name == 'name'
|
||||||
assert field.args == args
|
assert field.args == args
|
||||||
assert field.resolver == resolver
|
assert field.resolver == resolver
|
||||||
assert field.deprecation_reason == deprecation_reason
|
assert field.deprecation_reason == deprecation_reason
|
||||||
assert field.description == description
|
assert field.description == description
|
||||||
|
assert field.default_value == my_default
|
||||||
|
|
||||||
|
|
||||||
def test_field_required():
|
def test_field_required():
|
||||||
|
|
|
@ -1,9 +1,10 @@
|
||||||
import json
|
import json
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
from graphql import execute, Source, parse
|
from graphql import execute, Source, parse, GraphQLError
|
||||||
|
|
||||||
from ..objecttype import ObjectType
|
from ..objecttype import ObjectType
|
||||||
|
from ..field import Field
|
||||||
from ..inputfield import InputField
|
from ..inputfield import InputField
|
||||||
from ..inputobjecttype import InputObjectType
|
from ..inputobjecttype import InputObjectType
|
||||||
from ..scalars import String, Int
|
from ..scalars import String, Int
|
||||||
|
@ -22,6 +23,35 @@ def test_query():
|
||||||
assert executed.data == {'hello': 'World'}
|
assert executed.data == {'hello': 'World'}
|
||||||
|
|
||||||
|
|
||||||
|
def test_query_default_value():
|
||||||
|
class MyType(ObjectType):
|
||||||
|
field = String()
|
||||||
|
|
||||||
|
class Query(ObjectType):
|
||||||
|
hello = Field(MyType, default_value=MyType(field='something else!'))
|
||||||
|
|
||||||
|
hello_schema = Schema(Query)
|
||||||
|
|
||||||
|
executed = hello_schema.execute('{ hello { field } }')
|
||||||
|
assert not executed.errors
|
||||||
|
assert executed.data == {'hello': {'field': 'something else!'}}
|
||||||
|
|
||||||
|
|
||||||
|
def test_query_wrong_default_value():
|
||||||
|
class MyType(ObjectType):
|
||||||
|
field = String()
|
||||||
|
|
||||||
|
class Query(ObjectType):
|
||||||
|
hello = Field(MyType, default_value='hello')
|
||||||
|
|
||||||
|
hello_schema = Schema(Query)
|
||||||
|
|
||||||
|
executed = hello_schema.execute('{ hello { field } }')
|
||||||
|
assert len(executed.errors) == 1
|
||||||
|
assert executed.errors[0].message == GraphQLError('Expected value of type "MyType" but got: str.').message
|
||||||
|
assert executed.data is None
|
||||||
|
|
||||||
|
|
||||||
def test_query_resolve_function():
|
def test_query_resolve_function():
|
||||||
class Query(ObjectType):
|
class Query(ObjectType):
|
||||||
hello = String()
|
hello = String()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user