Added a test case for covering InputObjectType as Argument

Author: syrusakbary

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 @@ class MyInputObjectType(InputObjectType):
     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)