Restrict the usage of TypeProxy instances

This commit is contained in:
Syrus Akbary 2016-06-06 02:39:47 -07:00
parent 3a566241f3
commit ec67303594
3 changed files with 15 additions and 17 deletions

View File

@ -28,7 +28,7 @@ def test_argument_type():
def test_argument_graphene_type(): def test_argument_graphene_type():
argument = Argument(String()) argument = Argument(String)
assert argument.type == GraphQLString assert argument.type == GraphQLString
@ -44,7 +44,7 @@ def test_copy_argument_works():
def test_to_arguments(): def test_to_arguments():
arguments = to_arguments(a=String(), b=Argument(GraphQLString), c=Argument(String())) arguments = to_arguments(a=String(), b=Argument(GraphQLString), c=Argument(String))
assert list(arguments.keys()) == ['a', 'b', 'c'] assert list(arguments.keys()) == ['a', 'b', 'c']
assert [a.type for a in arguments.values()] == [GraphQLString] * 3 assert [a.type for a in arguments.values()] == [GraphQLString] * 3

View File

@ -7,38 +7,38 @@ from ..scalars import String
def test_list(): def test_list():
list_instance = List(String()) list_instance = List(String)
assert isinstance(list_instance, GraphQLList) assert isinstance(list_instance, GraphQLList)
assert list_instance.of_type == GraphQLString assert list_instance.of_type == GraphQLString
def test_list_lambda(): def test_list_lambda():
list_instance = List(lambda: String()) list_instance = List(lambda: String)
assert isinstance(list_instance, GraphQLList) assert isinstance(list_instance, GraphQLList)
assert list_instance.of_type == GraphQLString assert list_instance.of_type == GraphQLString
def test_list_list(): def test_list_list():
list_instance = List(List(String())) list_instance = List(List(String))
assert isinstance(list_instance, GraphQLList) assert isinstance(list_instance, GraphQLList)
assert isinstance(list_instance.of_type, GraphQLList) assert isinstance(list_instance.of_type, GraphQLList)
assert list_instance.of_type.of_type == GraphQLString assert list_instance.of_type.of_type == GraphQLString
def test_nonnull(): def test_nonnull():
list_instance = NonNull(String()) list_instance = NonNull(String)
assert isinstance(list_instance, GraphQLNonNull) assert isinstance(list_instance, GraphQLNonNull)
assert list_instance.of_type == GraphQLString assert list_instance.of_type == GraphQLString
def test_nonnull_lambda(): def test_nonnull_lambda():
list_instance = NonNull(lambda: String()) list_instance = NonNull(lambda: String)
assert isinstance(list_instance, GraphQLNonNull) assert isinstance(list_instance, GraphQLNonNull)
assert list_instance.of_type == GraphQLString assert list_instance.of_type == GraphQLString
def test_nonnull_list(): def test_nonnull_list():
list_instance = NonNull(List(String())) list_instance = NonNull(List(String))
assert isinstance(list_instance, GraphQLNonNull) assert isinstance(list_instance, GraphQLNonNull)
assert isinstance(list_instance.of_type, GraphQLList) assert isinstance(list_instance.of_type, GraphQLList)
assert list_instance.of_type.of_type == GraphQLString assert list_instance.of_type.of_type == GraphQLString

View File

@ -7,12 +7,10 @@ from ..types.enum import Enum
def is_graphene_type(_type): def is_graphene_type(_type):
if inspect.isclass(_type): return inspect.isclass(_type) and issubclass(_type, (
return issubclass(_type, ( Interface,
Interface, ObjectType,
ObjectType, InputObjectType,
InputObjectType, Scalar,
Scalar, Enum
Enum ))
))
return is_graphene_type(type(_type))