def converts_name_node_to_dict():
node = NameNode(value="test")
res = ast_to_dict(node)
assert res == {"kind": "name", "value": "test"}
assert list(res)[0] == "kind"
assert ast_to_dict(node, locations=True) == res
assert node.to_dict() == res
assert node.to_dict(locations=True) == res
def converts_two_name_nodes_to_list():
nodes = [NameNode(value="foo"), NameNode(value="bar")]
res = ast_to_dict(nodes)
assert ast_to_dict(nodes, locations=True) == res
assert res == [
{
"kind": "name",
"value": "foo"
},
{
"kind": "name",
"value": "bar"
},
]
def converts_recursive_ast_to_recursive_dict():
field = FieldNode(name="foo", arguments=(), selection_set=())
ast = SelectionSetNode(selections=(field, ))
field.selection_set = ast
res = ast_to_dict(ast)
assert res == {
"kind":
"selection_set",
"selections": [{
"kind": "field",
"name": "foo",
"alias": None,
"arguments": [],
"directives": None,
"selection_set": res,
}],
}
def converts_simple_schema_to_dict():
ast = parse("""
type Query {
me: User
}
type User {
id: ID
name: String
}
""")
res = ast_to_dict(ast)
assert ast.to_dict() == res
assert res == {
"definitions": [
{
"description":
None,
"directives": [],
"fields": [{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"name": {
"kind": "name",
"value": "me"
},
"type": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "User"
},
},
}],
"interfaces": [],
"kind":
"object_type_definition",
"name": {
"kind": "name",
"value": "Query"
},
},
{
"description":
None,
"directives": [],
"fields": [
{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"name": {
"kind": "name",
"value": "id"
},
"type": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "ID"
},
},
},
{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"name": {
"kind": "name",
"value": "name"
},
"type": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "String"
},
},
},
],
"interfaces": [],
"kind":
"object_type_definition",
"name": {
"kind": "name",
"value": "User"
},
},
],
"kind":
"document",
}
assert list(res)[0] == "kind"
def converts_simple_query_to_dict_with_locations():
ast = parse("""
query HeroForEpisode($ep: Episode!) {
hero(episode: $ep) {
name
... on Droid {
primaryFunction
}
... on Human {
height
}
}
}
""")
res = ast_to_dict(ast, locations=True)
assert ast.to_dict(locations=True) == res
assert res == {
"definitions": [{
"directives": [],
"kind":
"operation_definition",
"loc": {
"end": 293,
"start": 13
},
"name": {
"kind": "name",
"loc": {
"end": 33,
"start": 19
},
"value": "HeroForEpisode",
},
"operation":
"query",
"selection_set": {
"kind":
"selection_set",
"loc": {
"end": 293,
"start": 49
},
"selections": [{
"alias":
None,
"arguments": [{
"kind": "argument",
"loc": {
"end": 82,
"start": 70
},
"name": {
"kind": "name",
"loc": {
"end": 77,
"start": 70
},
"value": "episode",
},
"value": {
"kind": "variable",
"loc": {
"end": 82,
"start": 79
},
"name": {
"kind": "name",
"loc": {
"end": 82,
"start": 80
},
"value": "ep",
},
},
}],
"directives": [],
"kind":
"field",
"loc": {
"end": 279,
"start": 65
},
"name": {
"kind": "name",
"loc": {
"end": 69,
"start": 65
},
"value": "hero",
},
"selection_set": {
"kind":
"selection_set",
"loc": {
"end": 279,
"start": 84
},
"selections": [
{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"loc": {
"end": 106,
"start": 102
},
"name": {
"kind": "name",
"loc": {
"end": 106,
"start": 102
},
"value": "name",
},
"selection_set": None,
},
{
"directives": [],
"kind": "inline_fragment",
"loc": {
"end": 189,
"start": 123
},
"selection_set": {
"kind":
"selection_set",
"loc": {
"end": 189,
"start": 136
},
"selections": [{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"loc": {
"end": 171,
"start": 156,
},
"name": {
"kind": "name",
"loc": {
"end": 171,
"start": 156,
},
"value": "primaryFunction",
},
"selection_set": None,
}],
},
"type_condition": {
"kind": "named_type",
"loc": {
"end": 135,
"start": 130
},
"name": {
"kind": "name",
"loc": {
"end": 135,
"start": 130
},
"value": "Droid",
},
},
},
{
"directives": [],
"kind": "inline_fragment",
"loc": {
"end": 263,
"start": 206
},
"selection_set": {
"kind":
"selection_set",
"loc": {
"end": 263,
"start": 219
},
"selections": [{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"loc": {
"end": 245,
"start": 239,
},
"name": {
"kind": "name",
"loc": {
"end": 245,
"start": 239,
},
"value": "height",
},
"selection_set": None,
}],
},
"type_condition": {
"kind": "named_type",
"loc": {
"end": 218,
"start": 213
},
"name": {
"kind": "name",
"loc": {
"end": 218,
"start": 213
},
"value": "Human",
},
},
},
],
},
}],
},
"variable_definitions": [{
"default_value": None,
"directives": [],
"kind": "variable_definition",
"loc": {
"end": 47,
"start": 34
},
"type": {
"kind": "non_null_type",
"loc": {
"end": 47,
"start": 39
},
"type": {
"kind": "named_type",
"loc": {
"end": 46,
"start": 39
},
"name": {
"kind": "name",
"loc": {
"end": 46,
"start": 39
},
"value": "Episode",
},
},
},
"variable": {
"kind": "variable",
"loc": {
"end": 37,
"start": 34
},
"name": {
"kind": "name",
"loc": {
"end": 37,
"start": 35
},
"value": "ep",
},
},
}],
}],
"kind":
"document",
"loc": {
"end": 306,
"start": 0
},
}
keys = list(res)
assert keys[0] == "kind"
assert keys[-1] == "loc"
assert list(res["loc"]) == ["start", "end"]
def converts_simple_query_to_dict():
ast = parse("""
query HeroForEpisode($ep: Episode!) {
hero(episode: $ep) {
name
... on Droid {
primaryFunction
}
... on Human {
height
}
}
}
""")
res = ast_to_dict(ast)
assert ast.to_dict() == res
assert res == {
"definitions": [{
"directives": [],
"kind":
"operation_definition",
"name": {
"kind": "name",
"value": "HeroForEpisode"
},
"operation":
"query",
"selection_set": {
"kind":
"selection_set",
"selections": [{
"alias":
None,
"arguments": [{
"kind": "argument",
"name": {
"kind": "name",
"value": "episode"
},
"value": {
"kind": "variable",
"name": {
"kind": "name",
"value": "ep"
},
},
}],
"directives": [],
"kind":
"field",
"name": {
"kind": "name",
"value": "hero"
},
"selection_set": {
"kind":
"selection_set",
"selections": [
{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"name": {
"kind": "name",
"value": "name"
},
"selection_set": None,
},
{
"directives": [],
"kind": "inline_fragment",
"selection_set": {
"kind":
"selection_set",
"selections": [{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"name": {
"kind": "name",
"value": "primaryFunction",
},
"selection_set": None,
}],
},
"type_condition": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "Droid",
},
},
},
{
"directives": [],
"kind": "inline_fragment",
"selection_set": {
"kind":
"selection_set",
"selections": [{
"alias": None,
"arguments": [],
"directives": [],
"kind": "field",
"name": {
"kind": "name",
"value": "height",
},
"selection_set": None,
}],
},
"type_condition": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "Human",
},
},
},
],
},
}],
},
"variable_definitions": [{
"default_value": None,
"directives": [],
"kind": "variable_definition",
"type": {
"kind": "non_null_type",
"type": {
"kind": "named_type",
"name": {
"kind": "name",
"value": "Episode"
},
},
},
"variable": {
"kind": "variable",
"name": {
"kind": "name",
"value": "ep"
},
},
}],
}],
"kind":
"document",
}
assert list(res)[0] == "kind"
def keeps_all_other_leaf_nodes():
assert ast_to_dict(None) is None # type: ignore
assert ast_to_dict(42) == 42 # type: ignore
assert ast_to_dict("foo") == "foo" # type: ignore
ast = {"foo": "bar"}
assert ast_to_dict(ast) is ast # type: ignore
def converts_operation_type_to_its_value():
assert ast_to_dict(OperationType.QUERY) == "query"
def converts_simple_schema_to_dict_with_locations():
ast = parse("""
type Query {
me: User
}
type User {
id: ID
name: String
}
""")
res = ast_to_dict(ast, locations=True)
assert ast.to_dict(locations=True) == res
assert res == {
"definitions": [
{
"description":
None,
"directives": [],
"fields": [{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"loc": {
"end": 48,
"start": 40
},
"name": {
"kind": "name",
"loc": {
"end": 42,
"start": 40
},
"value": "me",
},
"type": {
"kind": "named_type",
"loc": {
"end": 48,
"start": 44
},
"name": {
"kind": "name",
"loc": {
"end": 48,
"start": 44
},
"value": "User",
},
},
}],
"interfaces": [],
"kind":
"object_type_definition",
"loc": {
"end": 62,
"start": 13
},
"name": {
"kind": "name",
"loc": {
"end": 23,
"start": 18
},
"value": "Query",
},
},
{
"description":
None,
"directives": [],
"fields": [
{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"loc": {
"end": 108,
"start": 102
},
"name": {
"kind": "name",
"loc": {
"end": 104,
"start": 102
},
"value": "id",
},
"type": {
"kind": "named_type",
"loc": {
"end": 108,
"start": 106
},
"name": {
"kind": "name",
"loc": {
"end": 108,
"start": 106
},
"value": "ID",
},
},
},
{
"arguments": [],
"description": None,
"directives": [],
"kind": "field_definition",
"loc": {
"end": 135,
"start": 123
},
"name": {
"kind": "name",
"loc": {
"end": 127,
"start": 123
},
"value": "name",
},
"type": {
"kind": "named_type",
"loc": {
"end": 135,
"start": 129
},
"name": {
"kind": "name",
"loc": {
"end": 135,
"start": 129
},
"value": "String",
},
},
},
],
"interfaces": [],
"kind":
"object_type_definition",
"loc": {
"end": 149,
"start": 76
},
"name": {
"kind": "name",
"loc": {
"end": 85,
"start": 81
},
"value": "User",
},
},
],
"kind":
"document",
"loc": {
"end": 162,
"start": 0
},
}
keys = list(res)
assert keys[0] == "kind"
assert keys[-1] == "loc"
assert list(res["loc"]) == ["start", "end"]