Skip to content

Commit 725a728

Browse files
committed
Prefer to register enums like this.
1 parent 924dfb0 commit 725a728

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

epoxy/registry.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ class TypeRegistry(object):
5050
# Types
5151
'ObjectType', 'InputType', 'Union' 'Interface', 'Implements', 'Scalar'
5252
# Functions
53-
'Schema', 'Register', 'Mixin',
53+
'Schema', 'Register', 'Mixin',
5454
# Mutations
5555
'Mutation', 'Mutations',
5656
# Fields

tests/test_register_enum.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,17 @@
33
from enum import Enum
44

55

6-
def test_register_single_type():
6+
def test_register_builtin_enum():
77
R = TypeRegistry()
88

9+
@R
910
class MyEnum(Enum):
1011
FOO = 1
1112
BAR = 2
1213
BAZ = 3
1314

14-
R(MyEnum)
15-
1615
enum = R.type('MyEnum')
1716
assert isinstance(enum, GraphQLEnumType)
1817
values = enum.get_values()
1918
assert [v.name for v in values] == ['FOO', 'BAR', 'BAZ']
20-
assert [v.value for v in values] == [1, 2, 3]
19+
assert [v.value for v in values] == [MyEnum.FOO.value, MyEnum.BAR.value, MyEnum.BAZ.value]

0 commit comments

Comments
 (0)