Skip to content

Commit 90272e5

Browse files
committed
Increase test coverage
1 parent e1f8480 commit 90272e5

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

graphene/types/tests/test_enum.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,11 @@ def custom_description(value):
7676

7777
return 'New Hope Episode' if value == Episode.NEWHOPE else 'Other'
7878

79+
def custom_deprecation_reason(value):
80+
return 'meh' if value == Episode.NEWHOPE else None
81+
7982
PyEpisode = PyEnum('PyEpisode', 'NEWHOPE,EMPIRE,JEDI')
80-
Episode = Enum.from_enum(PyEpisode, description=custom_description)
83+
Episode = Enum.from_enum(PyEpisode, description=custom_description, deprecation_reason=custom_deprecation_reason)
8184

8285
class Query(ObjectType):
8386
foo = Episode()
@@ -91,6 +94,10 @@ class Query(ObjectType):
9194
assert GraphQLPyEpisode[1].name == 'EMPIRE' and GraphQLPyEpisode[1].description == 'Other'
9295
assert GraphQLPyEpisode[2].name == 'JEDI' and GraphQLPyEpisode[2].description == 'Other'
9396

97+
assert GraphQLPyEpisode[0].name == 'NEWHOPE' and GraphQLPyEpisode[0].deprecation_reason == 'meh'
98+
assert GraphQLPyEpisode[1].name == 'EMPIRE' and GraphQLPyEpisode[1].deprecation_reason == None
99+
assert GraphQLPyEpisode[2].name == 'JEDI' and GraphQLPyEpisode[2].deprecation_reason == None
100+
94101

95102
def test_enum_from_python3_enum_uses_enum_doc():
96103
if not six.PY3:

0 commit comments

Comments
 (0)