def test_register_enum():
reg = Registry()
sa_enum = SQLAlchemyEnum("cat", "dog")
graphene_enum = GrapheneEnum("PetKind", [("CAT", 1), ("DOG", 2)])
reg.register_enum(sa_enum, graphene_enum)
assert reg.get_graphene_enum_for_sa_enum(sa_enum) is graphene_enum
def test_register_enum_incorrect_types():
reg = Registry()
sa_enum = SQLAlchemyEnum("cat", "dog")
graphene_enum = GrapheneEnum("PetKind", [("CAT", 1), ("DOG", 2)])
re_err = r"Expected Graphene Enum, but got: Enum\('cat', 'dog'\)"
with pytest.raises(TypeError, match=re_err):
reg.register_enum(sa_enum, sa_enum)
re_err = r"Expected SQLAlchemyEnumType, but got: .*PetKind.*"
with pytest.raises(TypeError, match=re_err):
reg.register_enum(graphene_enum, graphene_enum)
def test_register_sort_enum():
reg = Registry()
class PetType(SQLAlchemyObjectType):
class Meta:
model = Pet
registry = reg
sort_enum = GrapheneEnum(
"PetSort",
[("ID", EnumValue("id", Pet.id)), ("NAME", EnumValue("name", Pet.name))],
)
reg.register_sort_enum(PetType, sort_enum)
assert reg.get_sort_enum_for_object_type(PetType) is sort_enum
def test_register_sort_enum_incorrect_types():
reg = Registry()
class PetType(SQLAlchemyObjectType):
class Meta:
model = Pet
registry = reg
sort_enum = GrapheneEnum(
"PetSort",
[("ID", EnumValue("id", Pet.id)), ("NAME", EnumValue("name", Pet.name))],
)
re_err = r"Expected SQLAlchemyObjectType, but got: .*PetSort.*"
with pytest.raises(TypeError, match=re_err):
reg.register_sort_enum(sort_enum, sort_enum)
re_err = r"Expected Graphene Enum, but got: .*PetType.*"
with pytest.raises(TypeError, match=re_err):
reg.register_sort_enum(PetType, PetType)