def import_taskqueue_services(cls):
"""Imports and returns a taskqueue_services module from
core.platform.taskqueue.
Returns:
module. The core.platform.taskqueue services module.
"""
if (constants.DEV_MODE or utils.is_local_server_environment()):
from core.platform.taskqueue import dev_mode_taskqueue_services
return dev_mode_taskqueue_services
else:
from core.platform.taskqueue import cloud_taskqueue_services
return cloud_taskqueue_services
def test_get_current_appengine_environment(self):
saved_appengine_runtime = (os.environ['APPENGINE_RUNTIME'] if
'APPENGINE_RUNTIME' in os.environ else None)
saved_server_software = (os.environ['SERVER_SOFTWARE']
if 'SERVER_SOFTWARE' in os.environ else None)
os.environ['APPENGINE_RUNTIME'] = 'True'
self.assertTrue(utils.is_local_server_environment())
os.environ['SERVER_SOFTWARE'] = 'Google App Engine/'
self.assertTrue(utils.is_appengine_cloud_environment())
if saved_appengine_runtime is not None:
os.environ['SERVER_SOFTWARE'] = saved_appengine_runtime
else:
del os.environ['SERVER_SOFTWARE']
if saved_server_software is not None:
os.environ['SERVER_SOFTWARE'] = saved_server_software
else:
del os.environ['SERVER_SOFTWARE']