def test_with_server_name(self):
request = HttpRequest()
request.META = {
'SERVER_NAME': 'example.com',
'SERVER_PORT': 80
}
self.assertEqual(Site.find_for_request(request), self.site)
def process_request(self, request):
"""
Set request.site to contain the Site object responsible for handling this request,
according to hostname matching rules
"""
try:
request.site = Site.find_for_request(request)
except Site.DoesNotExist:
request.site = None
def setUp(self):
self.home_page = Page.objects.get(id=2)
self.routable_page = self.home_page.add_child(
instance=RoutablePageTest(
title="Routable Page",
live=True,
))
self.rf = RequestFactory()
self.request = self.rf.get(self.routable_page.url)
self.request.site = Site.find_for_request(self.request)
self.context = {'request': self.request}
def test_with_x_forwarded_host(self):
with self.settings(USE_X_FORWARDED_HOST=True):
request = HttpRequest()
request.META = {'HTTP_X_FORWARDED_HOST': 'example.com'}
self.assertEqual(Site.find_for_request(request), self.site)
def test_with_unknown_host(self):
request = HttpRequest()
request.META = {'HTTP_HOST': 'unknown.com'}
self.assertEqual(Site.find_for_request(request), self.default_site)
def test_with_host(self):
request = HttpRequest()
request.META = {'HTTP_HOST': 'example.com'}
self.assertEqual(Site.find_for_request(request), self.site)
def test_default(self):
request = HttpRequest()
self.assertEqual(Site.find_for_request(request), self.default_site)