Ticket #18314: TICKET-18314.diff

django/http/__init__.py

@@ -205 +205 @@
 
     def build_absolute_uri(self, location=None):
         """
-        Builds an absolute URI from the location and the variables available in
-        this request. If no location is specified, the absolute URI is built on
-        ``request.get_full_path()``.
+        Builds an absolute URI, using location and/or variables available in
+        the request, and returns it. If ``location`` is provided and is
+        absolute, it is simply converted to an RFC 3987 compliant URI and
+        returned. If ``location`` is provided but is relative or is
+        scheme-relative (i.e., //example.com/), it is urljoined to a base URL
+        constructed from the request variables.
         """
-        if not location:
-            location = self.get_full_path()
-        if not absolute_http_url_re.match(location):
-            current_uri = '%s://%s%s' % (self.is_secure() and 'https' or 'http',
-                                         self.get_host(), self.path)
-            location = urljoin(current_uri, location)
+        if location is None:
+            # Build the URI using ``self.get_full_path()``
+            location = '%(scheme)s//%(host)s%(path)s' % {
+                'scheme': 'https:' if self.is_secure() else 'http:',
+                'host': self.get_host(),
+                'path': self.get_full_path()
+            }
+        elif not absolute_http_url_re.match(location):
+            # The location was explicitly provided, but isn't absolute
+            base_url = '%(scheme)s//%(host)s%(path)s' % {
+                'scheme': 'https:' if self.is_secure() else 'http:',
+                'host': self.get_host(),
+                'path': self.path
+            }
+            # Join the constructed URL with the provided location, which will
+            # allow the provided ``location`` to apply query strings to the
+            # base path as well as override the host, if it begins with //
+            location = urljoin(base_url, location)
+
         return iri_to_uri(location)
 
     def _is_secure(self):
@@ -750 +766 @@
         return unicode(s, encoding, 'replace')
     else:
         return s
-

new file tests/regressiontests/http/__init__.py

@@ +1 @@
+

new file tests/regressiontests/http/models.py

@@ +1 @@
+

new file tests/regressiontests/http/tests.py

@@ +1 @@
+from django.utils import unittest
+from django.test.client import RequestFactory
+
+
+class HttpRequestTestCase(unittest.TestCase):
+    """
+    Regression tests for ticket # 18314.
+    """
+    def setUp(self):
+        """
+        Attaches a request factory.
+        """
+        self.factory = RequestFactory()
+
+    def test_build_absolute_uri_no_location(self):
+        """
+        Ensures that ``request.build_absolute_uri()`` returns the proper value
+        when the ``location`` argument is not provided, and ``request.path``
+        begins with //.
+        """
+        # //// is needed to create a request with a path beginning with //
+        request = self.factory.get('////django-ate-my-baby')
+        self.assertEqual(
+            request.build_absolute_uri(),
+            'http://testserver//django-ate-my-baby'
+        )
+
+    def test_build_absolute_uri_absolute_location(self):
+        """
+        Ensures that ``request.build_absolute_uri()`` returns the proper value
+        when an absolute URL ``location`` argument is provided, and
+        ``request.path`` begins with //.
+        """
+        # //// is needed to create a request with a path beginning with //
+        request = self.factory.get('////django-ate-my-baby')
+        self.assertEqual(
+            request.build_absolute_uri(location='http://example.com/?foo=bar'),
+            'http://example.com/?foo=bar'
+        )
+
+    def test_build_absolute_uri_schema_relateive_location(self):
+        """
+        Ensures that ``request.build_absolute_uri()`` returns the proper value
+        when a schema-relative URL ``location`` argument is provided, and
+        ``request.path`` begins with //.
+        """
+        # //// is needed to create a request with a path beginning with //
+        request = self.factory.get('////django-ate-my-baby')
+        self.assertEqual(
+            request.build_absolute_uri(location='//example.com/?foo=bar'),
+            'http://example.com/?foo=bar'
+        )
+
+    def test_build_absolute_uri_relative_location(self):
+        """
+        Ensures that ``request.build_absolute_uri()`` returns the proper value
+        when a relative URL ``location`` argument is provided, and
+        ``request.path`` begins with //.
+        """
+        # //// is needed to create a request with a path beginning with //
+        request = self.factory.get('////django-ate-my-baby')
+        self.assertEqual(
+            request.build_absolute_uri(location='/foo/bar/'),
+            'http://testserver/foo/bar/'
+        )