def test_run_action_runs_action_for_rule_class(self):
rule_class = mock.Mock()
site.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:
site.unregister(rule_class)
def setUp(self):
self._original_registry = site._registry.copy()
site._registry = {}
site.register(RuleOne)
site.register(RuleTwo)