def testGetHostName(self): """Tests that a correct host name is returned. """ test_data = RequestData() test_site = site.Site(link_id='test', hostname='test_host') test_data.site = test_site try: expected_host = os.environ['HTTP_HOST'] = 'some.testing.host.tld' self.assertEqual(system.getHostname(), expected_host) finally: if self.default_host is None: del os.environ['HTTP_HOST'] else: os.environ['HTTP_HOST'] = self.default_host #test a data object expected_host = 'test_host' self.assertEqual(system.getHostname(data=test_data), expected_host) test_data.site.hostname = '' try: expected_host = os.environ['HTTP_HOST'] = 'some.testing.host.tld' self.assertEqual(system.getHostname(data=test_data), expected_host) finally: if self.default_host is None: del os.environ['HTTP_HOST'] else: os.environ['HTTP_HOST'] = self.default_host
def testIsSecondaryHostName(self): """Tests if a request is from a secondary hostname. """ test_data = RequestData() test_site = site.Site(link_id='test', hostname='test_host') test_data.site = test_site try: os.environ['HTTP_HOST'] = 'some.testing.host.tld' self.assertFalse(system.isSecondaryHostname()) self.assertFalse(system.isSecondaryHostname(data=test_data)) test_data.site.hostname = "testing" self.assertTrue(system.isSecondaryHostname(data=test_data)) finally: if self.default_host is None: del os.environ['HTTP_HOST'] else: os.environ['HTTP_HOST'] = self.default_host