def test_get_environment(self): import os os.environ['JI_ENVIRONMENT'] = 'debug.host.com' self.assertEqual('debug', Common.get_environment(False)) os.environ['JI_ENVIRONMENT'] = 'sandbox.host.com' self.assertEqual('sandbox', Common.get_environment(False)) os.environ.pop('JI_ENVIRONMENT') self.assertEqual('production', Common.get_environment(False)) hostname = Common.get_hostname() if hostname.lower().find('debug') != -1: self.assertEqual('debug', Common.get_environment(True)) elif hostname.lower().find('sandbox') != -1: self.assertEqual('sandbox', Common.get_environment(True)) else: self.assertEqual('production', Common.get_environment(True))
def test_get_hostname(self): import socket self.assertEqual(socket.gethostname(), Common.get_hostname())