def test_custom_check_disallowed_for_wrong_arg(self):
admin_settings.ACCESS_CALLBACK = custom_check_with_arg
handler = TestHandler()
checked_handler = check()
check_wrapper = checked_handler(handler.get)
result = check_wrapper(handler, 'first_arg', kwarg='first_kwarg_wrong')
self.assertEquals('fail', result)
def test_custom_check_allowed_for_specific_arg(self):
admin_settings.ACCESS_CALLBACK = custom_check_with_arg
handler = TestHandler()
checked_handler = check()
check_wrapper = checked_handler(handler.get)
result = check_wrapper(handler, 'first_arg', kwarg='first_kwarg')
self.assertEquals('success!', result)
def test_default_check_admin(self):
# Stubs
users.get_current_user = fake_get_current_user(True)
users.is_current_user_admin = lambda *a, **kw: True
handler = TestHandler()
checked_handler = check()
check_wrapper = checked_handler(handler.get)
result = check_wrapper(handler)
self.assertEquals('success!', result)
def test_default_check_not_logged_in(self):
# Stubs
users.get_current_user = fake_get_current_user(False)
users.is_current_user_admin = lambda *a, **kw: False
handler = TestHandler()
checked_handler = check()
check_wrapper = checked_handler(handler.get)
result = check_wrapper(handler)
self.assertIsNone(result)
