def test_parses_double_value_enum():
# type: () -> None
body = "enum Hello { WO, RLD }"
loc = create_loc_fn(body)
doc = parse(body)
expected = ast.Document(
definitions=[
ast.EnumTypeDefinition(
name=ast.Name(value="Hello", loc=loc(5, 10)),
directives=[],
values=[
ast.EnumValueDefinition(
name=ast.Name(value="WO", loc=loc(13, 15)),
directives=[],
loc=loc(13, 15),
),
ast.EnumValueDefinition(
name=ast.Name(value="RLD", loc=loc(17, 20)),
directives=[],
loc=loc(17, 20),
),
],
loc=loc(0, 22),
)
],
loc=loc(0, 22),
)
assert doc == expected
def test_parses_single_value_enum():
body = 'enum Hello { WORLD }'
loc = create_loc_fn(body)
doc = parse(body)
expected = ast.Document(definitions=[
ast.EnumTypeDefinition(name=ast.Name(value='Hello', loc=loc(5, 10)),
values=[
ast.EnumValueDefinition(name=ast.Name(
value='WORLD', loc=loc(13, 18)),
loc=loc(13, 18))
],
loc=loc(0, 20))
],
loc=loc(0, 20))
assert doc == expected
def test_parses_double_value_enum():
body = 'enum Hello { WO, RLD }'
loc = create_loc_fn(body)
doc = parse(body)
expected = ast.Document(definitions=[
ast.EnumTypeDefinition(
name=ast.Name(value='Hello', loc=loc(5, 10)),
directives=[],
values=[
ast.EnumValueDefinition(name=ast.Name(value='WO',
loc=loc(13, 15)),
directives=[],
loc=loc(13, 15)),
ast.EnumValueDefinition(name=ast.Name(value='RLD',
loc=loc(17, 20)),
directives=[],
loc=loc(17, 20))
],
loc=loc(0, 22))
],
loc=loc(0, 22))
assert doc == expected
