File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -76,8 +76,11 @@ def custom_description(value):
76
76
77
77
return 'New Hope Episode' if value == Episode .NEWHOPE else 'Other'
78
78
79
+ def custom_deprecation_reason (value ):
80
+ return 'meh' if value == Episode .NEWHOPE else None
81
+
79
82
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 )
81
84
82
85
class Query (ObjectType ):
83
86
foo = Episode ()
@@ -91,6 +94,10 @@ class Query(ObjectType):
91
94
assert GraphQLPyEpisode [1 ].name == 'EMPIRE' and GraphQLPyEpisode [1 ].description == 'Other'
92
95
assert GraphQLPyEpisode [2 ].name == 'JEDI' and GraphQLPyEpisode [2 ].description == 'Other'
93
96
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
+
94
101
95
102
def test_enum_from_python3_enum_uses_enum_doc ():
96
103
if not six .PY3 :
You can’t perform that action at this time.
0 commit comments