Turning setup_fresh_app into SetupFreshApp, a decorator with options!

This way people can pass in particular paste/mediagoblin configs that they want to use.

diff --git a/mediagoblin/tests/tools.py b/mediagoblin/tests/tools.py
index 43da16de2875450e3a57a9dea29b43790e9fc9fb..526a2a1d89a367ff2bce0b4a7d7b7b2d6efe829e 100644 (file)
--- a/mediagoblin/tests/tools.py
+++ b/mediagoblin/tests/tools.py
@@ -171,19 +171,29 @@ def get_app(paste_config=None, mgoblin_config=None, dump_old_app=True):
     return app
 
 
-def setup_fresh_app(func):
+class SetupFreshApp(object):
     """
     Decorator to setup a fresh test application for this function.
 
     Cleans out test buckets and passes in a new, fresh test_app.
     """
-    @wraps(func)
-    def wrapper(*args, **kwargs):
-        test_app = get_app()
-        testing.clear_test_buckets()
-        return func(test_app, *args, **kwargs)
-
-    return wrapper
+    def __init__(self, paste_config, mgoblin_config, dump_old_app=True):
+        self.paste_config = paste_config
+        self.mgoblin_config = mgoblin_config
+
+    def __call__(self, func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            test_app = get_app(
+                paste_config=self.paste_config,
+                mgoblin_config=self.mgoblin_config,
+                dump_old_app=self.dump_old_app)
+            testing.clear_test_buckets()
+            return func(test_app, *args, **kwargs)
+
+        return wrapper
+
+setup_fresh_app = SetupFreshApp(TEST_SERVER_CONFIG, TEST_APP_CONFIG)
 
 
 def install_fixtures_simple(db, fixtures):