diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/requests/tests.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/requests/tests.py b/tests/requests/tests.py index 8c56e48f58..5bd3e5141f 100644 --- a/tests/requests/tests.py +++ b/tests/requests/tests.py @@ -620,12 +620,20 @@ class HostValidationTests(SimpleTestCase): } self.assertEqual(request.get_host(), 'example.com') + # Invalid hostnames would normally raise a SuspiciousOperation, + # but we have DEBUG=True, so this check is disabled. + request = HttpRequest() + request.META = { + 'HTTP_HOST': "invalid_hostname.com", + } + self.assertEqual(request.get_host(), "invalid_hostname.com") @override_settings(ALLOWED_HOSTS=[]) def test_get_host_suggestion_of_allowed_host(self): """get_host() makes helpful suggestions if a valid-looking host is not in ALLOWED_HOSTS.""" msg_invalid_host = "Invalid HTTP_HOST header: %r." msg_suggestion = msg_invalid_host + "You may need to add %r to ALLOWED_HOSTS." + msg_suggestion2 = msg_invalid_host + "The domain name provided is not valid according to RFC 1034/1035" for host in [ # Valid-looking hosts 'example.com', @@ -664,6 +672,14 @@ class HostValidationTests(SimpleTestCase): request.get_host ) + request = HttpRequest() + request.META = {'HTTP_HOST': "invalid_hostname.com"} + self.assertRaisesMessage( + SuspiciousOperation, + msg_suggestion2 % "invalid_hostname.com", + request.get_host + ) + @skipIf(connection.vendor == 'sqlite' and connection.settings_dict['TEST_NAME'] in (None, '', ':memory:'), |