e4a75ac2ce
For a long time two copies of the 'taskgraph' module have existed in parallel. We've attempted to keep them in sync, but over time they have diverged and the maintenance burden has increased. In order to reduce this burden, we'd like to re-join the two code bases. The canonical repo will be the one that lives outside of mozilla-central, and this module will depend on it. Since they both have the same module name (taskgraph) we need to rename the version in mozilla-central to avoid collisions. Other consumers of 'taskgraph' (like mobile repos) have standardized on '<project>_taskgraph' as their module names. So replicating that here as well. Differential Revision: https://phabricator.services.mozilla.com/D127118
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
# Any copyright is dedicated to the public domain.
|
|
# http://creativecommons.org/publicdomain/zero/1.0/
|
|
|
|
import pytest
|
|
from mozunit import main as mozunit_main
|
|
|
|
import gecko_taskgraph
|
|
from gecko_taskgraph.main import main as taskgraph_main
|
|
|
|
|
|
@pytest.fixture
|
|
def run_main(maketgg, monkeypatch):
|
|
def inner(args, **kwargs):
|
|
kwargs.setdefault("target_tasks", ["_fake-t-0", "_fake-t-1"])
|
|
tgg = maketgg(**kwargs)
|
|
|
|
def fake_get_taskgraph_generator(*args):
|
|
return tgg
|
|
|
|
monkeypatch.setattr(
|
|
gecko_taskgraph.main,
|
|
"get_taskgraph_generator",
|
|
fake_get_taskgraph_generator,
|
|
)
|
|
taskgraph_main(args)
|
|
return tgg
|
|
|
|
return inner
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"attr,expected",
|
|
(
|
|
("tasks", ["_fake-t-0", "_fake-t-1", "_fake-t-2"]),
|
|
("full", ["_fake-t-0", "_fake-t-1", "_fake-t-2"]),
|
|
("target", ["_fake-t-0", "_fake-t-1"]),
|
|
("target-graph", ["_fake-t-0", "_fake-t-1"]),
|
|
("optimized", ["_fake-t-0", "_fake-t-1"]),
|
|
("morphed", ["_fake-t-0", "_fake-t-1"]),
|
|
),
|
|
)
|
|
def test_show_taskgraph(run_main, capsys, attr, expected):
|
|
run_main([attr])
|
|
out, err = capsys.readouterr()
|
|
assert out.strip() == "\n".join(expected)
|
|
assert "Dumping result" in err
|
|
|
|
|
|
def test_tasks_regex(run_main, capsys):
|
|
run_main(["full", "--tasks=_.*-t-1"])
|
|
out, _ = capsys.readouterr()
|
|
assert out.strip() == "_fake-t-1"
|
|
|
|
|
|
def test_output_file(run_main, tmpdir):
|
|
output_file = tmpdir.join("out.txt")
|
|
assert not output_file.check()
|
|
|
|
run_main(["full", f"--output-file={output_file.strpath}"])
|
|
assert output_file.check()
|
|
assert output_file.read_text("utf-8").strip() == "\n".join(
|
|
["_fake-t-0", "_fake-t-1", "_fake-t-2"]
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mozunit_main()
|