Add a new decorator that can be used to declaratively set options on the checker instance for CheckerTestCase.

1 files changed, 14 insertions, 0 deletions

diff --git a/testutils.py b/testutils.py
index 818a5769f..668bd6d28 100644
--- a/testutils.py
+++ b/testutils.py
@@ -18,6 +18,7 @@ from __future__ import with_statement
 
 import collections
 import contextlib
+import functools
 import sys
 import re
 
@@ -149,6 +150,19 @@ class UnittestLinter(object):
             self.stats[name] = value
 
 
+def set_config(**kwargs):
+    """Decorator for setting config values on a checker."""
+    def _Wrapper(fun):
+        functools.wraps(fun)
+        def _Forward(self):
+            for key, value in kwargs.iteritems():
+                setattr(self.checker.config, key, value)
+            fun(self)
+
+        return _Forward
+    return _Wrapper
+
+
 class CheckerTestCase(testlib.TestCase):
     """A base testcase class for unittesting individual checker classes."""
     CHECKER_CLASS = None