46 lines
1.8 KiB
Python
46 lines
1.8 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/.
|
|
|
|
import unittest
|
|
from taskgraph.util.attributes import attrmatch
|
|
|
|
|
|
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))
|