summaryrefslogtreecommitdiffstats
path: root/taskcluster/test/test_new_config.py
blob: 2410bc01a4e8123f1e8ddd8cc51121839dd9f6c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# Any copyright is dedicated to the public domain.
# http://creativecommons.org/publicdomain/zero/1.0/

import pytest
from mozunit import main
from tryselect.selectors.auto import TRY_AUTO_PARAMETERS

pytestmark = pytest.mark.slow

PARAMS = TRY_AUTO_PARAMETERS.copy()
PARAMS.update(
    {
        "head_repository": "https://hg.mozilla.org/try",
        "project": "try",
        "target_kind": "test",
        # These ensure this isn't considered a backstop. The pushdate must
        # be slightly higher than the one in data/pushes.json, and
        # pushlog_id must not be a multiple of 10.
        "pushdate": 1593029536,
        "pushlog_id": "2",
    }
)

PARAMS_NEW_CONFIG = TRY_AUTO_PARAMETERS.copy()
PARAMS_NEW_CONFIG.update(
    {
        "head_repository": "https://hg.mozilla.org/try",
        "project": "try",
        "target_kind": "test",
        # These ensure this isn't considered a backstop. The pushdate must
        # be slightly higher than the one in data/pushes.json, and
        # pushlog_id must not be a multiple of 10.
        "pushdate": 1593029536,
        "pushlog_id": "2",
        "try_task_config": {"new-test-config": True},
        "try_mode": "try_task_config",
        "target_tasks_method": "try_tasks",
        "test_manifest_loader": "default",
    }
)


@pytest.mark.parametrize(
    "func,min_expected",
    (
        pytest.param(
            lambda t: (
                t.kind == "test"
                and t.attributes["unittest_suite"] == "mochitest-browser-chrome"
                and t.attributes["test_platform"] == "linux1804-64-qr/opt"
                and ("spi-nw" not in t.label and "a11y-checks" not in t.label)
            ),
            32,
            id="mochitest-browser-chrome",
        ),
    ),
)
def test_tasks_new_config_false(full_task_graph, filter_tasks, func, min_expected):
    """Ensure when using new-test-config that we have -cf tasks and they are half the total tasks."""
    tasks = [t.label for t in filter_tasks(full_task_graph, func)]
    assert len(tasks) == min_expected

    cf_tasks = [
        t.label for t in filter_tasks(full_task_graph, func) if t.label.endswith("-cf")
    ]
    assert len(cf_tasks) == min_expected / 2


@pytest.mark.parametrize(
    "func,min_expected",
    (
        pytest.param(
            lambda t: (
                t.kind == "test"
                and t.attributes["unittest_suite"] == "mochitest-browser-chrome"
                and t.attributes["test_platform"] == "linux1804-64-qr/opt"
            ),
            64,
            id="mochitest-browser-chrome",
        ),
    ),
)
def test_tasks_new_config_true(
    full_task_graph_new_config, filter_tasks, func, min_expected
):
    """Ensure when using new-test-config that no -cf tasks are scheduled and we have 2x the default and NO -cf."""
    tasks = [t.label for t in filter_tasks(full_task_graph_new_config, func)]
    assert len(tasks) == min_expected

    cf_tasks = [
        t.label
        for t in filter_tasks(full_task_graph_new_config, func)
        if t.label.endswith("-cf")
    ]
    assert len(cf_tasks) == 0


if __name__ == "__main__":
    main()