summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBaptiste Mispelon <bmispelon@gmail.com>2013-04-20 05:20:01 +0200
committerTim Graham <timograham@gmail.com>2013-07-30 13:40:18 -0400
commitbadca4716fee99372ae545eaf6d5521db11348c1 (patch)
treed1072a8b9295018523ca8ada4c275ff1e61b4e04
parentbf132bcb8d741a161cd9ff61073ac40e7e873b59 (diff)
[1.6.x] Fixed #10491 -- Allowed passing lazy objects to HttpResponseRedirect.
Thanks liangent for the report. Backport of 3c45fb8589 from master
-rw-r--r--django/http/response.py4
-rw-r--r--tests/httpwrappers/tests.py14
2 files changed, 15 insertions, 3 deletions
diff --git a/django/http/response.py b/django/http/response.py
index 784f21174e..8d8db2c8b4 100644
--- a/django/http/response.py
+++ b/django/http/response.py
@@ -15,7 +15,7 @@ from django.core import signing
from django.core.exceptions import DisallowedRedirect
from django.http.cookie import SimpleCookie
from django.utils import six, timezone
-from django.utils.encoding import force_bytes, iri_to_uri
+from django.utils.encoding import force_bytes, force_text, iri_to_uri
from django.utils.http import cookie_date
from django.utils.six.moves import map
@@ -454,7 +454,7 @@ class HttpResponseRedirectBase(HttpResponse):
allowed_schemes = ['http', 'https', 'ftp']
def __init__(self, redirect_to, *args, **kwargs):
- parsed = urlparse(redirect_to)
+ parsed = urlparse(force_text(redirect_to))
if parsed.scheme and parsed.scheme not in self.allowed_schemes:
raise DisallowedRedirect("Unsafe redirect to URL with protocol '%s'" % parsed.scheme)
super(HttpResponseRedirectBase, self).__init__(*args, **kwargs)
diff --git a/tests/httpwrappers/tests.py b/tests/httpwrappers/tests.py
index 194232e92f..c4c7996aa5 100644
--- a/tests/httpwrappers/tests.py
+++ b/tests/httpwrappers/tests.py
@@ -15,11 +15,14 @@ from django.http import (QueryDict, HttpResponse, HttpResponseRedirect,
SimpleCookie, BadHeaderError,
parse_cookie)
from django.test import TestCase
-from django.utils.encoding import smart_str
+from django.utils.encoding import smart_str, force_text
+from django.utils.functional import lazy
from django.utils._os import upath
from django.utils import six
from django.utils import unittest
+lazystr = lazy(force_text, six.text_type)
+
class QueryDictTests(unittest.TestCase):
def test_missing_key(self):
@@ -379,6 +382,10 @@ class HttpResponseTests(unittest.TestCase):
self.assertEqual(list(i), [b'abc'])
self.assertEqual(list(i), [])
+ def test_lazy_content(self):
+ r = HttpResponse(lazystr('helloworld'))
+ self.assertEqual(r.content, b'helloworld')
+
def test_file_interface(self):
r = HttpResponse()
r.write(b"hello")
@@ -415,6 +422,11 @@ class HttpResponseSubclassesTests(TestCase):
# Test that url attribute is right
self.assertEqual(response.url, response['Location'])
+ def test_redirect_lazy(self):
+ """Make sure HttpResponseRedirect works with lazy strings."""
+ r = HttpResponseRedirect(lazystr('/redirected/'))
+ self.assertEqual(r.url, '/redirected/')
+
def test_not_modified(self):
response = HttpResponseNotModified()
self.assertEqual(response.status_code, 304)