def without_extensions(): some_enum = GraphQLEnumType("SomeEnum", {"SOME_VALUE": None}) assert some_enum.extensions is None some_value = some_enum.values["SOME_VALUE"] assert some_value.extensions is None assert some_enum.to_kwargs()["extensions"] is None assert some_value.to_kwargs()["extensions"] is None
def with_extensions(): enum_extensions = {"SomeEnumExt": "enum"} value_extensions = {"SomeValueExt": "value"} some_enum = GraphQLEnumType( "SomeEnum", {"SOME_VALUE": GraphQLEnumValue(extensions=value_extensions)}, extensions=enum_extensions, ) assert some_enum.extensions is enum_extensions some_value = some_enum.values["SOME_VALUE"] assert some_value.extensions is value_extensions assert some_enum.to_kwargs()["extensions"] is enum_extensions assert some_value.to_kwargs()["extensions"] is value_extensions