DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (1aeaa33a64f9)

VCS Links

Line Code
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
# -*- 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('try', []))

    def test_all(self):
        self.assertTrue(match_run_on_projects('try', ['all']))
        self.assertTrue(match_run_on_projects('larch', ['all']))
        self.assertTrue(match_run_on_projects('autoland', ['all']))
        self.assertTrue(match_run_on_projects('mozilla-inbound', ['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('try', ['release']))
        self.assertFalse(match_run_on_projects('larch', ['release']))
        self.assertFalse(match_run_on_projects('autoland', ['release']))
        self.assertFalse(match_run_on_projects('mozilla-inbound', ['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('try', ['integration']))
        self.assertFalse(match_run_on_projects('larch', ['integration']))
        self.assertTrue(match_run_on_projects('autoland', ['integration']))
        self.assertTrue(match_run_on_projects('mozilla-inbound', ['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('try', ['release', 'try', 'maple']))
        self.assertFalse(match_run_on_projects('larch', ['release', 'try', 'maple']))
        self.assertTrue(match_run_on_projects('maple', ['release', 'try', 'maple']))
        self.assertFalse(match_run_on_projects('autoland', ['release', 'try', 'maple']))
        self.assertFalse(match_run_on_projects('mozilla-inbound', ['release', 'try', 'maple']))
        self.assertTrue(match_run_on_projects('mozilla-central', ['release', 'try', 'maple']))
        self.assertTrue(match_run_on_projects('mozilla-beta', ['release', 'try', 'maple']))
        self.assertTrue(match_run_on_projects('mozilla-release', ['release', 'try', 'maple']))


if __name__ == '__main__':
    main()