mirror of
https://github.com/graphql-python/graphene.git
synced 2025-07-18 12:02:19 +03:00
Make Graphene enums iterable like Python enums
This commit is contained in:
parent
f891a3683d
commit
88fe5c2ea1
13
graphene/tests/issues/test_1264.py
Normal file
13
graphene/tests/issues/test_1264.py
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
from graphene.types.enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
def test_enum_iteration():
|
||||||
|
class TestEnum(Enum):
|
||||||
|
FIRST = 1
|
||||||
|
SECOND = 2
|
||||||
|
|
||||||
|
result = []
|
||||||
|
expected_values = ["FIRST", "SECOND"]
|
||||||
|
for c in TestEnum:
|
||||||
|
result.append(c.name)
|
||||||
|
assert result == expected_values
|
|
@ -56,6 +56,9 @@ class EnumMeta(SubclassWithMeta_Meta):
|
||||||
return super(EnumMeta, cls).__call__(*args, **kwargs)
|
return super(EnumMeta, cls).__call__(*args, **kwargs)
|
||||||
# return cls._meta.enum(*args, **kwargs)
|
# return cls._meta.enum(*args, **kwargs)
|
||||||
|
|
||||||
|
def __iter__(cls):
|
||||||
|
return cls._meta.enum.__iter__()
|
||||||
|
|
||||||
def from_enum(
|
def from_enum(
|
||||||
cls, enum, name=None, description=None, deprecation_reason=None
|
cls, enum, name=None, description=None, deprecation_reason=None
|
||||||
): # noqa: N805
|
): # noqa: N805
|
||||||
|
|
Loading…
Reference in New Issue
Block a user