firefox-main/taskcluster/gecko_taskgraph/transforms/ui_test_apk.py (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

# 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/.
"""
UI-test-apk-specific transforms. Build the Slack notification payload from
the task's shipping-product, and build the test-lab.py invocation from the
task's flank-config / artifact-type / no-test-apk fields.
"""
import json
from taskgraph.transforms.base import TransformSequence
transforms = TransformSequence()
SLACK_PRODUCT_HEADERS = {
"fenix": "Firefox for Android :firefox:",
"focus": "Focus for Android :focusandroid:",
}
SLACK_TEXT = (
"|${task.metadata.name}>"
)
APK_APP = "/builds/worker/fetches/target.arm64-v8a.apk"
APK_TEST = "/builds/worker/fetches/target.noarch.apk"
TEST_LAB_SCRIPT = "taskcluster/scripts/tests/test-lab.py"
def _slack_attachments(product_header):
return json.dumps([
{
"color": "#FF0000",
"blocks": [
{
"type": "header",
"text": {"type": "plain_text", "text": product_header},
},
{
"type": "header",
"text": {
"type": "plain_text",
"text": "${task.metadata.name}",
},
},
{
"type": "section",
"fields": [
{
"type": "mrkdwn",
},
{
"type": "mrkdwn",
"text": "*Owner:*\n${task.metadata.owner}",
},
{
"type": "mrkdwn",
"text": "*Project:*\n${task.tags.project}",
},
{
"type": "mrkdwn",
},
{
"type": "mrkdwn",
"text": "*Source:*\n<${task.payload.env.GECKO_BASE_REPOSITORY}/rev/${task.payload.env.GECKO_HEAD_REV}|Commit> :hg:",
},
{
"type": "mrkdwn",
},
],
},
{"type": "divider"},
{
"type": "context",
"elements": [
{
"type": "mrkdwn",
"text": ":testops-notify: Mobile Test Engineering",
}
],
},
],
}
])
@transforms.add
def build_slack_notification(config, tasks):
for task in tasks:
extra = task.setdefault("extra", {})
notify = extra.get("notify") or {}
if not notify.get("enabled"):
extra["notify"] = {}
yield task
continue
product = task.get("attributes", {}).get("shipping-product")
header = SLACK_PRODUCT_HEADERS.get(product)
if header is None:
extra["notify"] = {}
yield task
continue
extra["notify"] = {
"slackText": SLACK_TEXT,
"slackAttachments": _slack_attachments(header),
}
yield task
@transforms.add
def build_test_lab_command(config, tasks):
for task in tasks:
flank_config = task.pop("flank-config")
artifact_type = task.pop("artifact-type", None)
no_test_apk = task.pop("no-test-apk", False)
command = ["python3", TEST_LAB_SCRIPT, flank_config, APK_APP]
if not no_test_apk:
command.extend(["--apk_test", APK_TEST])
if artifact_type:
command.extend(["--artifact_type", artifact_type])
run = task.setdefault("run", {})
run.setdefault("commands", []).append(command)
yield task