forked from mirrors/gecko-dev
02a7b4ebdf
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
102 lines
4.3 KiB
Python
102 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# 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 unittest
|
|
from taskgraph.util.attributes import (
|
|
attrmatch,
|
|
match_run_on_projects,
|
|
)
|
|
from mozunit import main
|
|
|
|
|
|
class Attrmatch(unittest.TestCase):
|
|
def test_trivial_match(self):
|
|
"""Given no conditions, anything matches"""
|
|
self.assertTrue(attrmatch({}))
|
|
|
|
def test_missing_attribute(self):
|
|
"""If a filtering attribute is not present, no match"""
|
|
self.assertFalse(attrmatch({}, someattr=10))
|
|
|
|
def test_literal_attribute(self):
|
|
"""Literal attributes must match exactly"""
|
|
self.assertTrue(attrmatch({"att": 10}, att=10))
|
|
self.assertFalse(attrmatch({"att": 10}, att=20))
|
|
|
|
def test_set_attribute(self):
|
|
"""Set attributes require set membership"""
|
|
self.assertTrue(attrmatch({"att": 10}, att=set([9, 10])))
|
|
self.assertFalse(attrmatch({"att": 10}, att=set([19, 20])))
|
|
|
|
def test_callable_attribute(self):
|
|
"""Callable attributes are called and any False causes the match to fail"""
|
|
self.assertTrue(attrmatch({"att": 10}, att=lambda val: True))
|
|
self.assertFalse(attrmatch({"att": 10}, att=lambda val: False))
|
|
|
|
def even(val):
|
|
return val % 2 == 0
|
|
|
|
self.assertTrue(attrmatch({"att": 10}, att=even))
|
|
self.assertFalse(attrmatch({"att": 11}, att=even))
|
|
|
|
def test_all_matches_required(self):
|
|
"""If only one attribute does not match, the result is False"""
|
|
self.assertFalse(attrmatch({"a": 1}, a=1, b=2, c=3))
|
|
self.assertFalse(attrmatch({"a": 1, "b": 2}, a=1, b=2, c=3))
|
|
self.assertTrue(attrmatch({"a": 1, "b": 2, "c": 3}, a=1, b=2, c=3))
|
|
|
|
|
|
class MatchRunOnProjects(unittest.TestCase):
|
|
def test_empty(self):
|
|
self.assertFalse(match_run_on_projects("birch", []))
|
|
|
|
def test_all(self):
|
|
self.assertTrue(match_run_on_projects("birch", ["all"]))
|
|
self.assertTrue(match_run_on_projects("larch", ["all"]))
|
|
self.assertTrue(match_run_on_projects("autoland", ["all"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-central", ["all"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-beta", ["all"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-release", ["all"]))
|
|
|
|
def test_release(self):
|
|
self.assertFalse(match_run_on_projects("birch", ["release"]))
|
|
self.assertFalse(match_run_on_projects("larch", ["release"]))
|
|
self.assertFalse(match_run_on_projects("autoland", ["release"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-central", ["release"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-beta", ["release"]))
|
|
self.assertTrue(match_run_on_projects("mozilla-release", ["release"]))
|
|
|
|
def test_integration(self):
|
|
self.assertFalse(match_run_on_projects("birch", ["integration"]))
|
|
self.assertFalse(match_run_on_projects("larch", ["integration"]))
|
|
self.assertTrue(match_run_on_projects("autoland", ["integration"]))
|
|
self.assertFalse(match_run_on_projects("mozilla-central", ["integration"]))
|
|
self.assertFalse(match_run_on_projects("mozilla-beta", ["integration"]))
|
|
self.assertFalse(match_run_on_projects("mozilla-integration", ["integration"]))
|
|
|
|
def test_combo(self):
|
|
self.assertTrue(match_run_on_projects("birch", ["release", "birch", "maple"]))
|
|
self.assertFalse(match_run_on_projects("larch", ["release", "birch", "maple"]))
|
|
self.assertTrue(match_run_on_projects("maple", ["release", "birch", "maple"]))
|
|
self.assertFalse(
|
|
match_run_on_projects("autoland", ["release", "birch", "maple"])
|
|
)
|
|
self.assertTrue(
|
|
match_run_on_projects("mozilla-central", ["release", "birch", "maple"])
|
|
)
|
|
self.assertTrue(
|
|
match_run_on_projects("mozilla-beta", ["release", "birch", "maple"])
|
|
)
|
|
self.assertTrue(
|
|
match_run_on_projects("mozilla-release", ["release", "birch", "maple"])
|
|
)
|
|
self.assertTrue(match_run_on_projects("birch", ["birch", "trunk"]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|