Ejemplo n.º 1
0
 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 _unregister_rules(self, *rules):
     for r in rules:
         rule_registry.unregister(r)
 def _unregister_rules(self, *rules):
     _ = [rule_registry.unregister(r) for r in rules]