mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-22 17:46:57 +03:00
Increase test coverage
This commit is contained in:
parent
e1f8480b32
commit
90272e5297
|
@ -76,8 +76,11 @@ def test_enum_from_builtin_enum_accepts_lambda_description():
|
|||
|
||||
return 'New Hope Episode' if value == Episode.NEWHOPE else 'Other'
|
||||
|
||||
def custom_deprecation_reason(value):
|
||||
return 'meh' if value == Episode.NEWHOPE else None
|
||||
|
||||
PyEpisode = PyEnum('PyEpisode', 'NEWHOPE,EMPIRE,JEDI')
|
||||
Episode = Enum.from_enum(PyEpisode, description=custom_description)
|
||||
Episode = Enum.from_enum(PyEpisode, description=custom_description, deprecation_reason=custom_deprecation_reason)
|
||||
|
||||
class Query(ObjectType):
|
||||
foo = Episode()
|
||||
|
@ -91,6 +94,10 @@ def test_enum_from_builtin_enum_accepts_lambda_description():
|
|||
assert GraphQLPyEpisode[1].name == 'EMPIRE' and GraphQLPyEpisode[1].description == 'Other'
|
||||
assert GraphQLPyEpisode[2].name == 'JEDI' and GraphQLPyEpisode[2].description == 'Other'
|
||||
|
||||
assert GraphQLPyEpisode[0].name == 'NEWHOPE' and GraphQLPyEpisode[0].deprecation_reason == 'meh'
|
||||
assert GraphQLPyEpisode[1].name == 'EMPIRE' and GraphQLPyEpisode[1].deprecation_reason == None
|
||||
assert GraphQLPyEpisode[2].name == 'JEDI' and GraphQLPyEpisode[2].deprecation_reason == None
|
||||
|
||||
|
||||
def test_enum_from_python3_enum_uses_enum_doc():
|
||||
if not six.PY3:
|
||||
|
|
Loading…
Reference in New Issue
Block a user