From 7c57d71e84cf43f47bda9cdeab14033e2adaed50 Mon Sep 17 00:00:00 2001 From: Syrus Akbary Date: Tue, 22 Nov 2016 22:35:24 -0800 Subject: [PATCH] Added a test case for covering InputObjectType as Argument --- graphene/types/tests/test_inputobjecttype.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/graphene/types/tests/test_inputobjecttype.py b/graphene/types/tests/test_inputobjecttype.py index ca73156c..7f8eaa7a 100644 --- a/graphene/types/tests/test_inputobjecttype.py +++ b/graphene/types/tests/test_inputobjecttype.py @@ -1,7 +1,9 @@ from ..abstracttype import AbstractType from ..field import Field +from ..argument import Argument from ..inputfield import InputField +from ..objecttype import ObjectType from ..inputobjecttype import InputObjectType from ..unmountedtype import UnmountedType @@ -61,6 +63,22 @@ def test_generate_inputobjecttype_unmountedtype(): assert isinstance(MyInputObjectType._meta.fields['field'], InputField) +def test_generate_inputobjecttype_as_argument(): + class MyInputObjectType(InputObjectType): + field = MyScalar() + + class MyObjectType(ObjectType): + field = Field(MyType, input=MyInputObjectType()) + + assert 'field' in MyObjectType._meta.fields + field = MyObjectType._meta.fields['field'] + assert isinstance(field, Field) + assert field.type == MyType + assert 'input' in field.args + assert isinstance(field.args['input'], Argument) + assert field.args['input'].type == MyInputObjectType + + def test_generate_inputobjecttype_inherit_abstracttype(): class MyAbstractType(AbstractType): field1 = MyScalar(MyType)