Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
[1.6.x] Fixed #10491 -- Allowed passing lazy objects to HttpResponseR…
…edirect.

Thanks liangent for the report.

Backport of 3c45fb8 from master
  • Loading branch information
bmispelon authored and timgraham committed Jul 30, 2013
1 parent bf132bc commit badca47
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 3 deletions.
4 changes: 2 additions & 2 deletions django/http/response.py
Expand Up @@ -15,7 +15,7 @@
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

Expand Down Expand Up @@ -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)
Expand Down
14 changes: 13 additions & 1 deletion tests/httpwrappers/tests.py
Expand Up @@ -15,11 +15,14 @@
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):
Expand Down Expand Up @@ -379,6 +382,10 @@ def test_iterator_isnt_rewound(self):
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")
Expand Down Expand Up @@ -415,6 +422,11 @@ def test_redirect(self):
# 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)
Expand Down

0 comments on commit badca47

Please sign in to comment.