def test_simple_non_null_type(): body = ''' type Hello { world: String! }''' doc = parse(body) loc = create_loc_fn(body) expected = ast.Document(definitions=[ ast.ObjectTypeDefinition( name=ast.Name(value='Hello', loc=loc(6, 11)), interfaces=[], directives=[], fields=[ ast.FieldDefinition(name=ast.Name(value='world', loc=loc(16, 21)), arguments=[], type=ast.NonNullType(type=ast.NamedType( name=ast.Name(value='String', loc=loc(23, 29)), loc=loc(23, 29)), loc=loc(23, 30)), directives=[], loc=loc(16, 30)) ], loc=loc(1, 32)) ], loc=loc(1, 32)) assert doc == expected
def get_variable_type(self, return_type: GraphQLType): if isinstance(return_type, GraphQLNonNull): return graphql_ast.NonNullType( type=self.get_variable_type(return_type.of_type)) elif isinstance(return_type, GraphQLList): return graphql_ast.ListType( type=self.get_variable_type(return_type.of_type)) else: return graphql_ast.NamedType(name=graphql_ast.Name( value=return_type.name))
def test_simple_non_null_type(): # type: () -> None body = """ type Hello { world: String! }""" doc = parse(body) loc = create_loc_fn(body) expected = ast.Document( definitions=[ ast.ObjectTypeDefinition( name=ast.Name(value="Hello", loc=loc(6, 11)), interfaces=[], directives=[], fields=[ ast.FieldDefinition( name=ast.Name(value="world", loc=loc(16, 21)), arguments=[], type=ast.NonNullType( type=ast.NamedType( name=ast.Name(value="String", loc=loc(23, 29)), loc=loc(23, 29), ), loc=loc(23, 30), ), directives=[], loc=loc(16, 30), ) ], loc=loc(1, 32), ) ], loc=loc(1, 32), ) assert doc == expected