graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-02-09 08:00:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restrict the usage of TypeProxy instances

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
graphene/types/tests/test_argument.py
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

12
graphene/types/tests/test_structures.py
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

4
graphene/utils/is_graphene_type.py
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))