fix return value of decorator

aodag · Nov 18, 2014 · 4376350 · 4376350
1 parent 7a90566
commit 4376350
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/webdispatch/base.py b/webdispatch/base.py
@@ -22,7 +22,8 @@ def register_app(self, name, app=None):
             def dec(app):
                 """ inner decorator for register app """
                 assert app is not None
-                return self.register_app(name, app)
+                self.register_app(name, app)
+                return app
             return dec
         self.applications[name] = app
 

diff --git a/webdispatch/tests/test_methoddispatcher.py b/webdispatch/tests/test_methoddispatcher.py
@@ -47,7 +47,9 @@ def test_register_app_decorator(self):
         """ test registering app"""
         app = self._make_one()
         dec = app.register_app("get")
-        dec(lambda environ, start_response: ["get"])
+        controller = lambda environ, start_response: ["get"]
+        ret = dec(controller)
+        compare(ret, controller)
         environ = setup_environ()
         start_response = mock.Mock()
         result = app(environ, start_response)