def run(config, status):
manager = RemoteJobManager("localhost", 5000)
pip_cmd = config.get(KEYS.PIP_CMD, None) if config.get(KEYS.PIP_CMD, None) is not None else config.get(KEYS.LOCAL_PATH, None)
if pip_cmd is None:
status.update(
{"succeeded": False, "errors": ["No local path specified"]})
return
ret, stdout = install_package_with_pip(pip_cmd)
if ret:
status.update(
{"succeeded": False, "stdout": stdout, "errors": ["Pip returned a non-zero error code"]})
return
manager.reload_site_packages()
status.update({"succeeded": True, "stdout": stdout})
def setUp(self):
self.logger = logging.getLogger('hoplite.remote_enabler')
self.logger_path = tempfile.mktemp()
self.logger_handler = logging.FileHandler(self.logger_path)
self.logger.addHandler(self.logger_handler)
self.logger.setLevel(logging.DEBUG)
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
def remote_install(host, package, port=5000, options=''):
'''
Provides a convient interface to install a python package
through the hoplite job mechanism.
:param string host: The host machine to install pip package
:param string package: The package to install
:param int port: The hoplite port to use (5000 by default)
:param string options: The pip options to use ('' by default)
'''
job_manager = RemoteJobManager(host, port)
job = job_manager.create_job(
plugin_name=constants.INSTALL_PYTHON_PACKAGE_JOB_NAME,
config={'PIP_CMD': 'install {0} {1}'.format(options, package)})
job.start()
job.join()
status = job.status()
if not status['succeeded']:
raise Error(status['stdout'])
def setUp(self):
self.logger = logging.getLogger('hoplite.remote_enabler')
self.logger_path = tempfile.mktemp()
self.logger_handler = logging.FileHandler(self.logger_path)
self.logger.addHandler(self.logger_handler)
self.logger.setLevel(logging.DEBUG)
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
self.class_instance = remotable_class.TestClass('this_is@a#string', 12349)
class TestRemoteDaemonManager(unittest2.TestCase):
def setUp(self):
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
def tearDown(self):
tear_down_hoplite(self.proc)
def test_create_job(self):
job = None
job = self.manager.create_job(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, {})
self.assertIsNotNone(job)
self.assertEquals(job.name, DOWNLOAD_NETWORK_FOLDER_JOB_NAME)
self.assertEquals(job.running(), False)
def test_get_job(self):
job = self.manager.create_job(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, {})
r_job = self.manager.get_job(job.uuid)
self.assertEquals(r_job.uuid, job.uuid)
self.assertEquals(r_job.name, job.name)
def test_get_job_raises(self):
self.assertRaises(JobDoesNotExistError, self.manager.get_job, 393939)
def test_get_job_plugins(self):
job_plugins = self.manager.get_job_plugins()
self.assertEquals(len(job_plugins), 7)
actual = job_plugins
self.assertIn(DOWNLOAD_FOLDER_FROM_FTP_JOB_NAME, actual)
self.assertIn(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, actual)
def test_get_job_plugins_raises(self):
with HTTMock(response_500):
self.assertRaises(InternalServerError,
self.manager.get_job_plugins)
def test_get_running_jobs(self):
jobs = self.manager.get_running_jobs()
self.assertEqual(len(jobs), 0)
with HTTMock(get_running_jobs):
jobs = self.manager.get_running_jobs()
self.assertEqual(len(jobs), 1)
def test_reload_site_packages(self):
done = self.manager.reload_site_packages()
self.assertTrue(done)
def test_is_manager_available(self):
time.sleep(1) # Give some time for job manager to connect to start
self.assertTrue(self.manager.is_available())
invalid_manager = RemoteJobManager("localhost", 48423)
self.assertFalse(invalid_manager.is_available())
class TestRemoteDaemonManager(unittest2.TestCase):
def setUp(self):
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
def tearDown(self):
tear_down_hoplite(self.proc)
def test_create_job(self):
job = None
job = self.manager.create_job(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, {})
self.assertIsNotNone(job)
self.assertEquals(job.name, DOWNLOAD_NETWORK_FOLDER_JOB_NAME)
self.assertEquals(job.running(), False)
def test_get_job(self):
job = self.manager.create_job(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, {})
r_job = self.manager.get_job(job.uuid)
self.assertEquals(r_job.uuid, job.uuid)
self.assertEquals(r_job.name, job.name)
def test_get_job_raises(self):
self.assertRaises(JobDoesNotExistError, self.manager.get_job, 393939)
def test_get_job_plugins(self):
job_plugins = self.manager.get_job_plugins()
self.assertEquals(len(job_plugins), 7)
actual = job_plugins
self.assertIn(DOWNLOAD_FOLDER_FROM_FTP_JOB_NAME, actual)
self.assertIn(DOWNLOAD_NETWORK_FOLDER_JOB_NAME, actual)
def test_get_job_plugins_raises(self):
with HTTMock(response_500):
self.assertRaises(InternalServerError, self.manager.get_job_plugins)
def test_get_running_jobs(self):
jobs = self.manager.get_running_jobs()
self.assertEqual(len(jobs), 0)
with HTTMock(get_running_jobs):
jobs = self.manager.get_running_jobs()
self.assertEqual(len(jobs), 1)
def test_reload_site_packages(self):
done = self.manager.reload_site_packages()
self.assertTrue(done)
def test_is_manager_available(self):
time.sleep(1) # Give some time for job manager to connect to start
self.assertTrue(self.manager.is_available())
invalid_manager = RemoteJobManager("localhost", 48423)
self.assertFalse(invalid_manager.is_available())
def remote_install(host, package, port=5000, options=''):
'''
Provides a convient interface to install a python package
through the hoplite job mechanism.
:param string host: The host machine to install pip package
:param string package: The package to install
:param int port: The hoplite port to use (5000 by default)
:param string options: The pip options to use ('' by default)
'''
job_manager = RemoteJobManager(host, port)
job = job_manager.create_job(
plugin_name=constants.INSTALL_PYTHON_PACKAGE_JOB_NAME,
config={
'PIP_CMD': 'install {0} {1}'.format(options, package)
})
job.start()
job.join()
status = job.status()
if not status['succeeded']:
raise Error(status['stdout'])
def test_is_manager_available(self):
time.sleep(1) # Give some time for job manager to connect to start
self.assertTrue(self.manager.is_available())
invalid_manager = RemoteJobManager("localhost", 48423)
self.assertFalse(invalid_manager.is_available())
def setUp(self):
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
def client_main():
parser = get_client_options_parser()
args = parser.parse_args()
manager = RemoteJobManager(args.address, args.port)
args.func(manager, args)
def setUp(self):
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
def setUp(self):
self.proc = start_hoplite_server(5001)
self.manager = RemoteJobManager("localhost", 5001)
self.class_instance = remotable_class.TestClass('this_is@a#string', 12349)
def test_is_manager_available(self):
time.sleep(1) # Give some time for job manager to connect to start
self.assertTrue(self.manager.is_available())
invalid_manager = RemoteJobManager("localhost", 48423)
self.assertFalse(invalid_manager.is_available())
