def test_run_action_runs_action_for_rule_class(self): rule_class = mock.Mock() rule_registry.register(rule_class) args = [mock.Mock()] kwargs = {'my_mock': mock.Mock()} validation_object = mock.Mock() try: rule = models.Rule(key=rule_class.key) rule.run_action(validation_object, *args, **kwargs) rule_class.assert_called_once_with(rule, validation_object) rule_class.return_value.run.assert_called_once_with(*args, **kwargs) finally: rule_registry.unregister(rule_class)
def _register_rules(self, *rules): for r in rules: rule_registry.register(r)
def _register_rules(self, *rules): _ = [rule_registry.register(r) for r in rules]