Commit 70be31bb authored by Tim Graham's avatar Tim Graham
Browse files

Fixed #24836 -- Made force_text() resolve lazy objects.

parent 10945ebe
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -77,12 +77,12 @@ def force_text(s, encoding='utf-8', strings_only=False, errors='strict'):
    If strings_only is True, don't convert (some) non-string-like objects.
    """
    # Handle the common case first for performance reasons.
    if isinstance(s, six.text_type):
    if issubclass(type(s), six.text_type):
        return s
    if strings_only and is_protected_type(s):
        return s
    try:
        if not isinstance(s, six.string_types):
        if not issubclass(type(s), six.string_types):
            if six.PY3:
                if isinstance(s, bytes):
                    s = six.text_type(s, encoding, errors)
+3 −0
Original line number Diff line number Diff line
@@ -19,3 +19,6 @@ Bugfixes
  ``Count()`` (:ticket:`24835`).

* Corrected ``HStoreField.has_changed()`` (:ticket:`24844`).

* Reverted an optimization to the CSRF template context processor which caused
  a regression (:ticket:`24836`).
+15 −0
Original line number Diff line number Diff line
import json

from django.http import HttpRequest
from django.template.context_processors import csrf
from django.test import SimpleTestCase
from django.utils.encoding import force_text


class TestContextProcessor(SimpleTestCase):

    def test_force_text_on_token(self):
        request = HttpRequest()
        request.META['CSRF_COOKIE'] = 'test-token'
        token = csrf(request).get('csrf_token')
        self.assertEqual(json.dumps(force_text(token)), '"test-token"')
+5 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ from django.utils.encoding import (
    escape_uri_path, filepath_to_uri, force_bytes, force_text, iri_to_uri,
    smart_text, uri_to_iri,
)
from django.utils.functional import SimpleLazyObject
from django.utils.http import urlquote_plus


@@ -28,6 +29,10 @@ class TestEncodingUtils(unittest.TestCase):
        exception = TypeError if six.PY3 else UnicodeError
        self.assertRaises(exception, force_text, MyString())

    def test_force_text_lazy(self):
        s = SimpleLazyObject(lambda: 'x')
        self.assertTrue(issubclass(type(force_text(s)), six.text_type))

    def test_force_bytes_exception(self):
        """
        Test that force_bytes knows how to convert to bytes an exception