Minor improvements

Author: oryan-block

src/main/kotlin/graphql/kickstart/tools/SchemaParser.kt

@@ -392,12 +392,7 @@ class SchemaParser internal constructor(
                         throw SchemaError("Expected type '${typeDefinition.name}' to be a ${expectedType.simpleName}, but it wasn't!  " +
                             "Was a type only permitted for object types incorrectly used as an input type, or vice-versa?")
                     }
-                    val found = inputObjects.filter { it.name == typeDefinition.name }
-                    if (found.size == 1) {
-                        found[0]
-                    } else {
-                        GraphQLTypeReference(typeDefinition.name)
-                    }
+                    inputObjects.find { it.name == typeDefinition.name } ?: GraphQLTypeReference(typeDefinition.name)
                 }
             }
             else -> throw SchemaError("Unknown type: $typeDefinition")

src/test/groovy/graphql/kickstart/tools/SchemaParserSpec.groovy

@@ -4,7 +4,6 @@ import graphql.kickstart.tools.resolver.FieldResolverError
 import graphql.language.SourceLocation
 import graphql.schema.GraphQLArgument
 import graphql.schema.GraphQLInputObjectType
-import graphql.schema.GraphQLInputType
 import graphql.schema.GraphQLNonNull
 import graphql.schema.GraphQLSchema
 import graphql.schema.idl.SchemaDirectiveWiring
@@ -394,7 +393,7 @@ class SchemaParserSpec extends Specification {
                         GraphQLArgument onArgument(SchemaDirectiveWiringEnvironment<GraphQLArgument> environment) {
                             switch (environment.getElement().type.class) {
                                 case GraphQLNonNull:
-                                    assert (environment.getElement().type as graphql.schema.GraphQLNonNull).wrappedType.class == GraphQLInputObjectType
+                                    assert (environment.getElement().type as GraphQLNonNull).wrappedType.class == GraphQLInputObjectType
                             }
                             return environment.getElement()
                         }})
@@ -408,8 +407,8 @@ class SchemaParserSpec extends Specification {
         GraphQLArgument testNullableArgument = schema.getObjectType("Query")
                 .getFieldDefinition("testNullable")
                 .arguments.first()
-        testNonNullableArgument.type.class == graphql.schema.GraphQLNonNull
-        (testNonNullableArgument.type as graphql.schema.GraphQLNonNull).wrappedType.class == GraphQLInputObjectType
+        testNonNullableArgument.type.class == GraphQLNonNull
+        (testNonNullableArgument.type as GraphQLNonNull).wrappedType.class == GraphQLInputObjectType
         testNullableArgument.type.class == GraphQLInputObjectType
     }