def resolve_all_option_collections(self, info, **kwargs):
field_map = {
"option": ("option", "select"),
}
return optimize(OptionCollection.objects.all(),
ast_to_dict(info.field_asts),
field_map)
def test_optimize(query_node, push_stored):
qs = Push.objects.filter(revision=push_stored[0]["revision"])
field_map = {
"jobType": ("job_type", "select"),
"jobGroup": ("job_group", "select"),
"failureClassification": ("failure_classification", "prefetch"),
}
qs = helpers.optimize(qs, query_node, field_map)
assert ('failure_classification',) == qs._prefetch_related_lookups
assert {'job_type': {}, 'job_group': {}} == qs.query.select_related
def test_optimize(query_node, result_set_stored):
qs = Push.objects.filter(revision=result_set_stored[0]["revision"])
field_map = {
"jobType": ("job_type", "select"),
"jobGroup": ("job_type__job_group", "select"),
"failureClassification": ("failure_classification", "prefetch"),
}
qs = helpers.optimize(qs, query_node, field_map)
assert ('failure_classification', ) == qs._prefetch_related_lookups
assert {'job_type': {'job_group': {}}} == qs.query.select_related
def resolve_jobs(self, info, **kwargs):
field_map = {
"buildPlatform": ("build_platform", "select"),
"jobLog": ("job_log", "prefetch"),
"jobType": ("job_type", "select"),
"jobGroup": ("job_group", "select"),
"failureClassification": ("failure_classification", "prefetch"),
"failureLine": ("job_log__failure_line", "prefetch"),
"group": ("job_log__failure_line__group", "prefetch"),
"textLogStep": ("text_log_step", "prefetch"),
"errors": ("text_log_step__errors", "prefetch"),
}
return optimize(Job.objects.filter(push=self, **kwargs),
ast_to_dict(info.field_asts), field_map)
def resolve_jobs(self, info, **kwargs):
field_map = {
"buildPlatform": ("build_platform", "select"),
"jobLog": ("job_log", "prefetch"),
"jobType": ("job_type", "select"),
"jobGroup": ("job_group", "select"),
"failureClassification": ("failure_classification", "prefetch"),
"failureLine": ("job_log__failure_line", "prefetch"),
"group": ("job_log__failure_line__group", "prefetch"),
"textLogStep": ("text_log_step", "prefetch"),
"errors": ("text_log_step__errors", "prefetch"),
}
return optimize(Job.objects.filter(push=self, **kwargs),
ast_to_dict(info.field_asts),
field_map)
def resolve_all_option_collections(self, info, **kwargs):
field_map = {
"option": ("option", "select"),
}
return optimize(OptionCollection.objects.all(),
ast_to_dict(info.field_asts), field_map)
