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()
