Ticket #12662: hostname_middleware.diff

tests/regressiontests/middleware/tests.py

@@ -3 +3 @@
 from django.test import TestCase
 from django.http import HttpRequest
 from django.middleware.common import CommonMiddleware
+from django.middleware.hostname import EnforceHostnameMiddleware
 from django.conf import settings
 
 class CommonMiddlewareTest(TestCase):
@@ -125 +126 @@
         self.assertEquals(r.status_code, 301)
         self.assertEquals(r['Location'],
                           'http://www.testserver/middleware/slash/')
+
+class EnforceHostnameMiddlewareTest(TestCase):
+    def setUp(self):
+        try:
+            self.enforce = settings.ENFORCE_HOSTNAME
+        except:
+            self.enforce = None
+        settings.ENFORCE_HOSTNAME = None
+
+    def tearDown(self):
+        if self.enforce:
+            settings.ENFORCE_HOSTNAME = self.enforce
+    
+    def _get_request(self, path):
+        request = HttpRequest()
+        request.META = {
+            'SERVER_NAME': 'testserver',
+            'SERVER_PORT': 80,
+        }
+        request.path = request.path_info = "/middleware/%s" % path
+        return request
+    
+    def test_no_enforce_setting(self):
+        """
+        No enforce setting
+        """
+        request = self._get_request('test/')
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r, None)
+    
+    def test_no_redirect_necessary(self):
+        """
+        Enforce is set, but the hostname matches
+        """
+        settings.ENFORCE_HOSTNAME = 'testserver'
+        request = self._get_request('test/')
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r, None)
+    
+    def test_single_hostname_with_redirect(self):
+        """
+        Enforce is set and the single hostname does not match
+        """
+        settings.ENFORCE_HOSTNAME = 'example.com'
+        request = self._get_request('test/')
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r.status_code, 301)
+        self.assertEqual(r['location'], 'http://example.com/middleware/test/')
+    
+    def test_single_hostname_with_redirect_and_querystring(self):
+        """
+        Enforce is set and the single hostname does not match (and there is a querystring)
+        """
+        settings.ENFORCE_HOSTNAME = 'example.com'
+        request = self._get_request('test/')
+        request.GET['test'] = 1
+        request.META['QUERY_STRING'] = 'test=1'
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r.status_code, 301)
+        self.assertEqual(r['location'], 'http://example.com/middleware/test/?test=1')
+    
+    def test_multiple_hostnames_with_redirect(self):
+        """
+        Enforce is set with multiple hostnames and the hostname does not match
+        """
+        settings.ENFORCE_HOSTNAME = ['example.com', 'www.example.com']
+        request = self._get_request('test/')
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r.status_code, 301)
+        self.assertEqual(r['location'], 'http://example.com/middleware/test/')
+    
+    def test_multiple_hostnames_without_redirect(self):
+        """
+        Enforce is set with multiple hostnames and the hostname does match
+        """
+        settings.ENFORCE_HOSTNAME = ['example.com', 'www.example.com', 'testserver']
+        request = self._get_request('test/')
+        r = EnforceHostnameMiddleware().process_request(request)
+        self.assertEqual(r, None)
+