def test_redirect_to_sudo_simple(self): response = redirect_to_sudo('/foo') self.assertEqual(response.status_code, 302) self.assertEqual(response['Location'], '/sudo/?next=/foo')
def test_redirect_to_sudo_with_querystring(self): response = redirect_to_sudo('/foo?foo=bar') self.assertEqual(response.status_code, 302) self.assertEqual(response['Location'], '/sudo/?next=/foo%3Ffoo%3Dbar')
def inner(request, *args, **kwargs): if not request.is_sudo(): return redirect_to_sudo(request.get_full_path()) return func(request, *args, **kwargs)