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