def test_dedicated_controller_wrapper_after_milestone_reached(self): conf = AppConfig(minimal=True) def f(): pass milestones.environment_loaded.reach() conf.register_controller_wrapper(None, controller=f)
def test_global_controller_wrapper_after_milestone_reached(self): milestones.environment_loaded.reach() conf = AppConfig(minimal=True) conf.register_controller_wrapper(None)
# fix imports for appengine environments import fix_imports (fix_imports) from tg import AppConfig from tg import redirect from google.appengine.api import users from main import MainController def controller_wrapper(next_caller): def call(*args, **kw): user = users.get_current_user() if not user: login_url = users.create_login_url('/') redirect(login_url) return next_caller(*args, **kw) return call config = AppConfig(minimal=True, root_controller=MainController()) config.register_controller_wrapper(controller_wrapper) app = config.make_wsgi_app()