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 unittest
from mozunit import main
from taskgraph.util.attributes import attrmatch
from gecko_taskgraph.util.attributes import match_run_on_projects
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={9, 10}))
self.assertFalse(attrmatch({"att": 10}, att={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.assertTrue(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.assertTrue(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()