def verify_file(self, filename: str, parser: QueryParser,
renderer: DataclassesRenderer) -> None:
root, _s = os.path.splitext(filename)
target_filename = "".join([root, ".py"])
dir_name = os.path.dirname(target_filename)
with open(filename, "r") as fin:
query = fin.read()
parsed = parser.parse(query)
rendered = renderer.render(parsed)
with open(target_filename, "r") as f:
file_content = f.read()
self.assert_rendered_file(target_filename, file_content,
rendered)
enums = renderer.render_enums(parsed)
for enum_name, code in enums.items():
target_enum_filename = os.path.join(
dir_name, "".join([get_enum_filename(enum_name), ".py"]))
with open(target_enum_filename, "r") as f:
file_content = f.read()
self.assert_rendered_file(target_enum_filename,
file_content, code)
input_objects = renderer.render_input_objects(parsed)
for input_object_name, code in input_objects.items():
target_input_object_filename = os.path.join(
dir_name,
"".join([get_input_filename(input_object_name), ".py"]))
with open(target_input_object_filename, "r") as f:
file_content = f.read()
self.assert_rendered_file(target_input_object_filename,
file_content, code)
def test_non_nullable_list(self):
PersonType = GraphQLObjectType(
"Person", lambda: {"name": GraphQLField(GraphQLString)}
)
schema = GraphQLSchema(
query=GraphQLObjectType(
name="RootQueryType",
fields={
"people": GraphQLField(
GraphQLList(GraphQLNonNull(PersonType)),
resolve=lambda obj, info: {"name": "eran"},
)
},
)
)
query = """
query GetPeople {
people {
name
}
}
"""
parser = QueryParser(schema)
dataclass_renderer = DataclassesRenderer(schema)
parsed = parser.parse(query)
rendered = dataclass_renderer.render(parsed)
m = self.load_module(rendered)
mock_client = MagicMock()
mock_client.call = MagicMock(
return_value="""
{
"data": {
"people": [
{
"name": "eran"
},
{
"name": "eran1"
}
]
}
}
"""
)
result = m.GetPeople.execute(mock_client)
assert result
assert isinstance(result, m.GetPeople.GetPeopleData)
assert len(result.people) == 2
assert result.people[0].name == "eran"
assert result.people[1].name == "eran1"
def verify_file(self, filename: str, parser: QueryParser,
renderer: DataclassesRenderer):
root, _s = os.path.splitext(filename)
target_filename = root + '.py'
with open(filename, 'r') as fin:
query = fin.read()
parsed = parser.parse(query)
rendered = renderer.render(parsed)
assert rendered == open(target_filename, 'r').read(), \
f"Generated file name {target_filename} does " \
"not match compilation result"
def test_simple_query_with_enums_default_value(self):
"""
enum LengthUnit {
METER
KM
}
type Starship {
id: ID!
name: String!
length(unit: LengthUnit = METER): Float
}
type Query {
ship(id: String!): Starship
}
"""
length_unit_enum = GraphQLEnumType(
"LengthUnit",
{"METER": GraphQLEnumValue("METER"), "KM": GraphQLEnumValue("KM")},
description="One of the films in the Star Wars Trilogy",
)
starship_type = GraphQLObjectType(
"Starship",
lambda: {
"id": GraphQLField(
GraphQLNonNull(GraphQLString), description="The id of the ship."
),
"name": GraphQLField(
GraphQLString, description="The name of the ship."
),
"length": GraphQLField(
GraphQLInt,
args={
"unit": GraphQLArgument(
GraphQLNonNull(length_unit_enum),
default_value="METER",
description="id of the droid",
)
},
),
},
)
query_type = GraphQLObjectType(
"Query",
lambda: {
"ship": GraphQLField(
starship_type,
args={
"id": GraphQLArgument(
GraphQLNonNull(GraphQLString), description="id of the ship"
)
},
)
},
)
schema = GraphQLSchema(query_type, types=[length_unit_enum, starship_type])
query = """
query GetStarship {
ship(id: "Enterprise") {
id
name
length(unit: METER)
}
}
"""
query_parser = QueryParser(schema)
query_renderer = DataclassesRenderer(schema)
parsed = query_parser.parse(query)
rendered_enums = query_renderer.render_enums(parsed)
for rendered_enum in rendered_enums:
self.load_module(rendered_enum)
rendered = query_renderer.render(parsed)
m = self.load_module(rendered)
response = m.GetStarship.from_json(
"""
{
"data": {
"ship": {
"id": "Enterprise",
"name": "Enterprise",
"length": 100
}
}
}
"""
)
assert response
ship = response.data.ship
assert ship
assert ship.id == "Enterprise"
assert ship.name == "Enterprise"
assert ship.length == 100
def test_simple_query_with_enums_default_value(self):
"""
enum LengthUnit {
METER
KM
}
type Starship {
id: ID!
name: String!
length(unit: LengthUnit = METER): Float
}
type Query {
ship(id: String!): Starship
}
"""
length_unit_enum = GraphQLEnumType(
'LengthUnit',
{
'METER': GraphQLEnumValue('METER'),
'KM': GraphQLEnumValue('KM'),
},
description='One of the films in the Star Wars Trilogy',
)
starship_type = GraphQLObjectType(
'Starship', lambda: {
'id':
GraphQLField(GraphQLNonNull(GraphQLString),
description='The id of the ship.'),
'name':
GraphQLField(GraphQLString,
description='The name of the ship.'),
'length':
GraphQLField(GraphQLInt,
args={
'unit':
GraphQLArgument(GraphQLNonNull(
length_unit_enum),
default_value='METER',
description='id of the droid')
})
})
query_type = GraphQLObjectType(
'Query', lambda: {
'ship':
GraphQLField(
starship_type,
args={
'id':
GraphQLArgument(GraphQLNonNull(GraphQLString),
description='id of the ship')
},
)
})
schema = GraphQLSchema(query_type,
types=[length_unit_enum, starship_type])
query = """
query GetStarship {
ship(id: "Enterprise") {
id
name
length(unit: METER)
}
}
"""
query_parser = QueryParser(schema)
query_renderer = DataclassesRenderer(schema)
parsed = query_parser.parse(query)
rendered = query_renderer.render(parsed)
m = self.load_module(rendered)
response = m.GetStarship.from_json("""
{
"data": {
"ship": {
"id": "Enterprise",
"name": "Enterprise",
"length": 100
}
}
}
""")
assert response
ship = response.data.ship
assert ship
assert ship.id == 'Enterprise'
assert ship.name == 'Enterprise'
assert ship.length == 100