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()