def test_parser_query_inline_fragment(self):
query = """
query GetFilm {
returnOfTheJedi: film(id: "1") {
... on Film {
title
director
}
}
}
"""
parser = QueryParser(self.swapi_schema)
parsed = parser.parse(query)
expected = asdict(
ParsedQuery(
query=query,
objects=[
ParsedOperation(
name="GetFilm",
type="query",
children=[
ParsedObject(
name="GetFilmData",
fields=[
ParsedField(
name="returnOfTheJedi",
type="Film",
nullable=True,
)
],
children=[
ParsedObject(
name="Film",
fields=[
ParsedField(name="title",
type="str",
nullable=False),
ParsedField(
name="director",
type="str",
nullable=False,
),
],
)
],
)
],
)
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))
def test_parser_mutation(self):
query = """
mutation CreateHero {
createHero {
hero {
name
}
ok
}
}
"""
parser = QueryParser(self.swapi_schema)
parsed = parser.parse(query)
expected = asdict(
ParsedQuery(
query=query,
objects=[
ParsedOperation(
name="CreateHero",
type="mutation",
children=[
ParsedObject(
name="CreateHeroData",
fields=[
ParsedField(
name="createHero",
type="CreateHeroPayload",
nullable=True,
)
],
children=[
ParsedObject(
name="CreateHeroPayload",
fields=[
ParsedField(name="hero",
type="Hero",
nullable=True),
ParsedField(name="ok",
type="bool",
nullable=True),
],
children=[
ParsedObject(
name="Hero",
fields=[
ParsedField(
name="name",
type="str",
nullable=False,
)
],
)
],
)
],
)
],
)
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))
def test_parser_query_with_enums(self):
query = """
query MyIssues {
viewer {
issues(first: 5) {
edges {
node {
author { login }
authorAssociation
}
}
}
}
}
"""
parsed = self.github_parser.parse(query)
issue_child = ParsedObject(
name="Issue",
fields=[
ParsedField(name="author", type="Actor", nullable=True),
ParsedField(
name="authorAssociation",
type="CommentAuthorAssociation",
nullable=False,
),
],
children=[
ParsedObject(
name="Actor",
fields=[
ParsedField(name="login", type="str", nullable=False)
],
)
],
)
child = ParsedObject(
name="MyIssuesData",
fields=[ParsedField(name="viewer", type="User", nullable=False)],
children=[
ParsedObject(
name="User",
fields=[
ParsedField(name="issues",
type="IssueConnection",
nullable=False)
],
children=[
ParsedObject(
name="IssueConnection",
fields=[
ParsedField(name="edges",
type="List[IssueEdge]",
nullable=True)
],
children=[
ParsedObject(
name="IssueEdge",
fields=[
ParsedField(name="node",
type="Issue",
nullable=True)
],
children=[issue_child],
)
],
)
],
)
],
)
expected = asdict(
ParsedQuery(
query=query,
enums=[
ParsedEnum(
name="CommentAuthorAssociation",
values={
"MEMBER": "MEMBER",
"OWNER": "OWNER",
"COLLABORATOR": "COLLABORATOR",
"CONTRIBUTOR": "CONTRIBUTOR",
"FIRST_TIME_CONTRIBUTOR": "FIRST_TIME_CONTRIBUTOR",
"FIRST_TIMER": "FIRST_TIMER",
"NONE": "NONE",
},
)
],
objects=[
ParsedOperation(name="MyIssues",
type="query",
children=[child])
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))
def test_parser_connection_query(self):
query = """
query GetAllFilms {
allFilms {
count: totalCount
edges {
node {
id
title
director
}
}
}
allHeroes {
edges {
node {
...HeroFields
}
}
}
}
fragment HeroFields on Hero {
id
name
}
"""
parser = QueryParser(self.swapi_schema)
parsed = parser.parse(query)
expected = asdict(
ParsedQuery(
query=query,
objects=[
ParsedOperation(
name="GetAllFilms",
type="query",
children=[
ParsedObject(
name="GetAllFilmsData",
fields=[
ParsedField(
name="allFilms",
type="FilmConnection",
nullable=True,
),
ParsedField(
name="allHeroes",
type="HeroConnection",
nullable=True,
),
],
children=[
ParsedObject(
name="FilmConnection",
fields=[
ParsedField(name="count",
type="int",
nullable=True),
ParsedField(
name="edges",
type="List[FilmEdge]",
nullable=True,
),
],
children=[
ParsedObject(
name="FilmEdge",
fields=[
ParsedField(
name="node",
type="Film",
nullable=True,
)
],
children=[
ParsedObject(
name="Film",
fields=[
ParsedField(
name="id",
type="str",
nullable=False,
),
ParsedField(
name="title",
type="str",
nullable=False,
),
ParsedField(
name="director",
type="str",
nullable=False,
),
],
)
],
)
],
),
ParsedObject(
name="HeroConnection",
fields=[
ParsedField(
name="edges",
type="List[HeroEdge]",
nullable=True,
)
],
children=[
ParsedObject(
name="HeroEdge",
fields=[
ParsedField(
name="node",
type="Hero",
nullable=True,
)
],
children=[
ParsedObject(
name="Hero",
parents=["HeroFields"],
fields=[],
)
],
)
],
),
],
)
],
),
ParsedObject(
name="HeroFields",
fields=[
ParsedField(name="id", type="str", nullable=False),
ParsedField(name="name",
type="str",
nullable=False),
],
),
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))
def test_parser_query_with_variables(self):
query = """
query GetFilm($theFilmID: ID!) {
returnOfTheJedi: film(id: $theFilmID) {
title
director
}
}
"""
parser = QueryParser(self.swapi_schema)
parsed = parser.parse(query)
expected = asdict(
ParsedQuery(
query=query,
objects=[
ParsedOperation(
name="GetFilm",
type="query",
variables=[
ParsedVariableDefinition(
name="theFilmID",
type="str",
nullable=False,
is_list=False,
)
],
children=[
ParsedObject(
name="GetFilmData",
fields=[
ParsedField(
name="returnOfTheJedi",
type="Film",
nullable=True,
)
],
children=[
ParsedObject(
name="Film",
fields=[
ParsedField(name="title",
type="str",
nullable=False),
ParsedField(
name="director",
type="str",
nullable=False,
),
],
)
],
)
],
)
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))
def test_parser_query_complex_fragment(self):
query = """
query GetPerson {
luke: character(id: "luke") {
...CharacterFields
}
}
fragment CharacterFields on Person {
name
home: homeworld {
name
}
}
"""
parser = QueryParser(self.swapi_schema)
parsed = parser.parse(query)
expected = asdict(
ParsedQuery(
query=query,
objects=[
ParsedOperation(
name="GetPerson",
type="query",
children=[
ParsedObject(
name="GetPersonData",
fields=[
ParsedField(name="luke",
type="Person",
nullable=True)
],
children=[
ParsedObject(
name="Person",
parents=["CharacterFields"],
fields=[],
)
],
)
],
),
ParsedObject(
name="CharacterFields",
fields=[
ParsedField(name="name",
type="str",
nullable=False),
ParsedField(name="home",
type="Planet",
nullable=False),
],
children=[
ParsedObject(
name="Planet",
fields=[
ParsedField(name="name",
type="str",
nullable=False)
],
)
],
),
],
))
parsed_dict = asdict(parsed)
assert bool(parsed)
assert parsed_dict == expected, str(DeepDiff(parsed_dict, expected))