Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
# 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
import pytest
from mozunit import main
from gecko_taskgraph.test.conftest import FakeParameters
from gecko_taskgraph.transforms.build import collapse_unified_builds
def unified_build():
return {
"name": "macosx64-shippable/opt",
"attributes": {"shippable": True},
"index": {"job-name": "macosx64-opt"},
"worker": {"env": {}, "max-run-time": 1800},
"run": {"job-script": "taskcluster/scripts/misc/unify.sh"},
"dependencies": {
"x64": "build-macosx64-x64-shippable/opt",
"aarch64": "build-macosx64-aarch64-shippable/opt",
},
"fetches": {
"x64": [{"artifact": "target.dmg", "dest": "x64"}],
"aarch64": [{"artifact": "target.dmg", "dest": "aarch64"}],
},
}
def per_arch_build(arch, package_tests=True):
env = {"MOZ_AUTOMATION_PACKAGE_TESTS": "1"} if package_tests else {}
return {
"name": f"macosx64-{arch}-shippable/opt",
"attributes": {},
"index": {"job-name": f"macosx64-{arch}-opt"},
"worker": {"env": env, "max-run-time": 7200},
"run": {"config": ["builds/releng_base_mac_64_cross_builds.py"]},
"fetches": {"toolchain": ["linux64-clang"]},
}
def run(run_transform, use_artifact, package_tests=True):
jobs = [
per_arch_build("x64", package_tests),
per_arch_build("aarch64", package_tests),
unified_build(),
]
params = FakeParameters({
"try_task_config": {"use-artifact-builds": use_artifact},
})
return list(
run_transform(collapse_unified_builds, jobs, kind="build", params=params)
)[-1]
def test_collapse_unified_build(run_transform):
job = run(run_transform, use_artifact=True)
assert "dependencies" not in job
assert "job-script" not in job["run"]
assert job["run"]["config"] == ["builds/releng_base_mac_64_cross_builds.py"]
assert job["fetches"] == {"toolchain": ["linux64-clang"]}
assert job["worker"]["env"]["MOZ_AUTOMATION_PACKAGE_TESTS"] == "1"
assert job["worker"]["max-run-time"] == 7200
@pytest.mark.parametrize(
"use_artifact,package_tests",
(
pytest.param(False, True, id="artifact-builds-disabled"),
pytest.param(True, False, id="tests-not-packaged"),
),
)
def test_unified_build_left_alone(run_transform, use_artifact, package_tests):
job = run(run_transform, use_artifact, package_tests)
assert job == unified_build()
if __name__ == "__main__":
main()