Support URLs without scheme

Author: browniebroke

django_hosts/test/client.py

@@ -7,13 +7,14 @@ class HostClientMixin:
     """Test client to help work with django-hosts in tests."""
 
     def generic(self, method, path, *args, **extra):
-        if path.startswith('http'):
+        scheme, netloc, *_others = urlparse(path)
+        if scheme:
+            extra["wsgi.url_scheme"] = scheme
+        if netloc:
             # Populate the host header from the URL host
-            _scheme, host, *_others = urlparse(path)
-            if extra.get('headers') is None:
-                extra['headers'] = {}
-            if extra['headers'].get('host') is None:
-                extra['headers']['host'] = host
+            extra["headers"] = extra["headers"] or {}
+            if extra["headers"].get("host") is None:
+                extra["headers"]["host"] = netloc
         return super().generic(method, path, *args, **extra)
 
 

tests/test_client.py

@@ -6,10 +6,14 @@
 class TestHostsClient(TestCase):
     client_class = HostsClient
 
-    def test_host_header_set_from_url(self):
+    def test_host_header_set_from_full_url(self):
         response = self.client.get('https://testserver.local/api/v1/users/')
         assert response.request['HTTP_HOST'] == 'testserver.local'
 
+    def test_host_header_set_from_url_without_scheme(self):
+        response = self.client.get('//testserver.local/api/v1/users/')
+        assert response.request['HTTP_HOST'] == 'testserver.local'
+
     def test_host_header_from_user_is_not_overridden(self):
         response = self.client.get(
             'https://testserver.local/api/v1/users/', headers={'host': 'api.example.com'}