class HostnameRequestTest(BaseTestCase):
def setUp(self):
"""
Create a couple of hostname tests.
"""
super(HostnameRequestTest, self).setUp()
self.hostname_test1 = HostnameTestModel.objects.create(hostname="www.example.com")
self.hostname_test2 = HostnameTestModel.objects.create(hostname="example.com")
self.hostname_test3 = HostnameTestModel.objects.create(hostname="127.0.0.1:8000")
# And link to replacement data.
self.hostname_test1.replacements.add(self.data_host)
self.hostname_test2.replacements.add(self.data_host)
self.hostname_test3.replacements.add(self.data_another)
# Set up middleware test.
self.test = HostnameTest()
def test_creation(self):
assert HostnameTestModel.objects.all().count() == 3
assert HostnameTestModel.objects.filter(hostname__icontains="example").count() == 2
def test_simple_lookup(self):
"""
With the default request environ this should return
a match with our www.example.com hostnametest
"""
request = self.req_factory.request()
match = self.test.test(request)
assert match == self.hostname_test1
def test_deliberate_fail(self):
"""
Create a request with a HTTP_HOST we shouldn't
have a match for, test we return None.
"""
request = self.req_factory.request(HTTP_HOST="www.cantfindme.com")
match = self.test.test(request)
assert match is None
def test_localhost_lookup(self):
"""
Test the match returns for our port based hostname.
"""
request = self.req_factory.request(HTTP_HOST="127.0.0.1:8000")
match = self.test.test(request)
assert match == self.hostname_test3
def setUp(self):
"""
Create a couple of hostname tests.
"""
super(HostnameRequestTest, self).setUp()
self.hostname_test1 = HostnameTestModel.objects.create(hostname="www.example.com")
self.hostname_test2 = HostnameTestModel.objects.create(hostname="example.com")
self.hostname_test3 = HostnameTestModel.objects.create(hostname="127.0.0.1:8000")
# And link to replacement data.
self.hostname_test1.replacements.add(self.data_host)
self.hostname_test2.replacements.add(self.data_host)
self.hostname_test3.replacements.add(self.data_another)
# Set up middleware test.
self.test = HostnameTest()
