def test_call_second_true(self): rules = [mock.Mock(return_value=False), mock.Mock(return_value=True)] check = policy.OrCheck(rules) self.assertEqual(check('target', 'cred', None), True) rules[0].assert_called_once_with('target', 'cred', None) rules[1].assert_called_once_with('target', 'cred', None)
def test_call_all_false(self): rules = [mock.Mock(return_value=False), mock.Mock(return_value=False)] check = policy.OrCheck(rules) self.assertEqual(check('target', 'cred'), False) rules[0].assert_called_once_with('target', 'cred') rules[1].assert_called_once_with('target', 'cred')
def test_str(self): check = policy.OrCheck(['rule1', 'rule2']) self.assertEqual(str(check), '(rule1 or rule2)')
def test_add_check(self): check = policy.OrCheck(['rule1', 'rule2']) check.add_check('rule3') self.assertEqual(check.rules, ['rule1', 'rule2', 'rule3'])
def test_init(self): check = policy.OrCheck(['rule1', 'rule2']) self.assertEqual(check.rules, ['rule1', 'rule2'])