From c8adab675a84057da9c85efc040e2a85c78b6a96 Mon Sep 17 00:00:00 2001 From: Christoph Zwerschke Date: Sun, 30 Jun 2019 22:02:50 +0200 Subject: [PATCH] Enum values are returned as dict in core-next --- graphene/types/tests/test_enum.py | 35 +++++++------------------------ 1 file changed, 7 insertions(+), 28 deletions(-) diff --git a/graphene/types/tests/test_enum.py b/graphene/types/tests/test_enum.py index e0937992..cf7227e4 100644 --- a/graphene/types/tests/test_enum.py +++ b/graphene/types/tests/test_enum.py @@ -80,36 +80,15 @@ def test_enum_from_builtin_enum_accepts_lambda_description(): class Query(ObjectType): foo = Episode() - schema = Schema(query=Query) + schema = Schema(query=Query).graphql_schema - GraphQLPyEpisode = schema._type_map["PyEpisode"].values + episode = schema.get_type("PyEpisode") - assert schema._type_map["PyEpisode"].description == "StarWars Episodes" - assert ( - GraphQLPyEpisode[0].name == "NEWHOPE" - and GraphQLPyEpisode[0].description == "New Hope Episode" - ) - assert ( - GraphQLPyEpisode[1].name == "EMPIRE" - and GraphQLPyEpisode[1].description == "Other" - ) - assert ( - GraphQLPyEpisode[2].name == "JEDI" - and GraphQLPyEpisode[2].description == "Other" - ) - - assert ( - GraphQLPyEpisode[0].name == "NEWHOPE" - and GraphQLPyEpisode[0].deprecation_reason == "meh" - ) - assert ( - GraphQLPyEpisode[1].name == "EMPIRE" - and GraphQLPyEpisode[1].deprecation_reason is None - ) - assert ( - GraphQLPyEpisode[2].name == "JEDI" - and GraphQLPyEpisode[2].deprecation_reason is None - ) + assert episode.description == "StarWars Episodes" + assert [(name, value.description, value.deprecation_reason) + for name, value in episode.values.items()] == [ + ('NEWHOPE', 'New Hope Episode', 'meh'), + ('EMPIRE', 'Other', None), ('JEDI', 'Other', None)] def test_enum_from_python3_enum_uses_enum_doc():