diff options
Diffstat (limited to '')
-rw-r--r-- | taskcluster/android_taskgraph/target_tasks.py | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/taskcluster/android_taskgraph/target_tasks.py b/taskcluster/android_taskgraph/target_tasks.py new file mode 100644 index 0000000000..38b1f7b772 --- /dev/null +++ b/taskcluster/android_taskgraph/target_tasks.py @@ -0,0 +1,91 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. + +from taskgraph.target_tasks import _target_task + +from android_taskgraph.release_type import does_task_match_release_type + + +@_target_task("promote_android") +def target_tasks_promote(full_task_graph, parameters, graph_config): + return _filter_release_promotion( + full_task_graph, + parameters, + filtered_for_candidates=[], + shipping_phase="promote", + ) + + +@_target_task("push_android") +def target_tasks_push(full_task_graph, parameters, graph_config): + filtered_for_candidates = target_tasks_promote( + full_task_graph, + parameters, + graph_config, + ) + return _filter_release_promotion( + full_task_graph, parameters, filtered_for_candidates, shipping_phase="push" + ) + + +@_target_task("ship_android") +def target_tasks_ship(full_task_graph, parameters, graph_config): + filtered_for_candidates = target_tasks_push( + full_task_graph, + parameters, + graph_config, + ) + return _filter_release_promotion( + full_task_graph, parameters, filtered_for_candidates, shipping_phase="ship" + ) + + +def _filter_release_promotion( + full_task_graph, parameters, filtered_for_candidates, shipping_phase +): + def filter(task, parameters): + # Include promotion tasks; these will be optimized out + if task.label in filtered_for_candidates: + return True + + # Ship geckoview in firefox-android ship graph + if ( + shipping_phase == "ship" + and task.attributes.get("shipping_product") == "fennec" + and task.kind in ("beetmover-geckoview", "upload-symbols") + and parameters["release_product"] == "firefox-android" + ): + return True + + # TODO: get rid of this release_type match + if ( + task.attributes.get("shipping_phase") == shipping_phase + and task.attributes.get("shipping_product") == parameters["release_product"] + and does_task_match_release_type(task, parameters["release_type"]) + ): + return True + + return False + + return [l for l, t in full_task_graph.tasks.items() if filter(t, parameters)] + + +@_target_task("screenshots") +def target_tasks_screnshots(full_task_graph, parameters, graph_config): + """Select the set of tasks required to generate screenshots on a real device.""" + + def filter(task, parameters): + return task.attributes.get("screenshots", False) + + return [l for l, t in full_task_graph.tasks.items() if filter(t, parameters)] + + +@_target_task("legacy_api_ui_tests") +def target_tasks_legacy_api_ui_tests(full_task_graph, parameters, graph_config): + """Select the set of tasks required to run select UI tests on other API.""" + + def filter(task, parameters): + return task.attributes.get("legacy", False) + + return [l for l, t in full_task_graph.tasks.items() if filter(t, parameters)] |