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