snap_test.py (1505B)
1 # This Source Code Form is subject to the terms of the Mozilla Public 2 # License, v. 2.0. If a copy of the MPL was not distributed with this 3 # file, You can obtain one at http://mozilla.org/MPL/2.0/. 4 """ """ 5 6 import logging 7 8 from taskgraph.transforms.base import TransformSequence 9 from taskgraph.util.dependencies import get_primary_dependency 10 from taskgraph.util.treeherder import inherit_treeherder_from_dep 11 12 logger = logging.getLogger(__name__) 13 14 transforms = TransformSequence() 15 16 17 @transforms.add 18 def fill_template(config, tasks): 19 for task in tasks: 20 assert "snap-upstream-test-" in task.get("label") 21 22 test_type = task.get("attributes")["snap_test_type"] 23 test_release = task.get("attributes")["snap_test_release"] 24 task["label"] = task.get("label").replace( 25 "-test-", "-test-" + test_type + "-" + test_release + "-" 26 ) 27 28 dep = get_primary_dependency(config, task) 29 assert dep 30 31 inherit_treeherder_from_dep(task, dep) 32 33 th_group = dep.task["extra"]["treeherder"]["groupSymbol"].replace("B", "Sel") 34 th_symbol = ( 35 f"{test_type}-{test_release}-{dep.task['extra']['treeherder']['symbol']}" 36 ) 37 task["treeherder"]["symbol"] = f"{th_group}({th_symbol})" 38 39 timeout = 10 40 if dep.attributes.get("build_type") != "opt": 41 timeout = 60 42 task["worker"]["env"]["BUILD_IS_DEBUG"] = "1" 43 44 task["worker"]["env"]["TEST_TIMEOUT"] = f"{timeout}" 45 46 yield task