fune/taskcluster/taskgraph/test/test_util_bugbug.py
Ricky Stewart 02a7b4ebdf Bug 1654103: Standardize on Black for Python code in mozilla-central.
Allow-list all Python code in tree for use with the black linter, and re-format all code in-tree accordingly.

To produce this patch I did all of the following:

1. Make changes to tools/lint/black.yml to remove include: stanza and update list of source extensions.

2. Run ./mach lint --linter black --fix

3. Make some ad-hoc manual updates to python/mozbuild/mozbuild/test/configure/test_configure.py -- it has some hard-coded line numbers that the reformat breaks.

4. Make some ad-hoc manual updates to `testing/marionette/client/setup.py`, `testing/marionette/harness/setup.py`, and `testing/firefox-ui/harness/setup.py`, which have hard-coded regexes that break after the reformat.

5. Add a set of exclusions to black.yml. These will be deleted in a follow-up bug (1672023).

# ignore-this-changeset

Differential Revision: https://phabricator.services.mozilla.com/D94045
2020-10-26 18:34:53 +00:00

61 lines
1.7 KiB
Python

# 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/.
from __future__ import absolute_import, print_function, unicode_literals
import mozunit
from taskgraph.util.bugbug import (
BUGBUG_BASE_URL,
push_schedules,
)
def test_group_translation(responses):
branch = ("integration/autoland",)
rev = "abcdef"
query = "/push/{}/{}/schedules".format(branch, rev)
url = BUGBUG_BASE_URL + query
responses.add(
responses.GET,
url,
json={
"groups": {
"dom/indexedDB": 1,
"testing/web-platform/tests/IndexedDB": 1,
"testing/web-platform/mozilla/tests/IndexedDB": 1,
},
"config_groups": {
"dom/indexedDB": ["label1", "label2"],
"testing/web-platform/tests/IndexedDB": ["label3"],
"testing/web-platform/mozilla/tests/IndexedDB": ["label4"],
},
},
status=200,
)
assert len(push_schedules) == 0
data = push_schedules(branch, rev)
print(data)
assert sorted(data["groups"]) == [
"/IndexedDB",
"/_mozilla/IndexedDB",
"dom/indexedDB",
]
assert data["config_groups"] == {
"dom/indexedDB": ["label1", "label2"],
"/IndexedDB": ["label3"],
"/_mozilla/IndexedDB": ["label4"],
}
assert len(push_schedules) == 1
# Value is memoized.
responses.reset()
push_schedules(branch, rev)
assert len(push_schedules) == 1
if __name__ == "__main__":
mozunit.main()