def test_generate_target_with_options(self):
add_command = AddCommand("image-processing",
None, "L",
"localhost", "8888", "8888",
"my.aws.com", None, None, None)
result = add_command.generate_target()
self.assertEqual(result["type"], "L")
self.assertEqual(result["name"], "image-processing")
self.assertEqual(result["local_port"], "8888")
def test_generate_target_with_argument(self):
add_command = AddCommand("image-processing",
"8888:localhost:8888 [email protected]",
None, None, None, None, None, None, None, None)
result = add_command.generate_target()
self.assertEqual(result["type"], "L")
self.assertEqual(result["name"], "image-processing")
self.assertEqual(result["local_port"], "8888")
self.assertEqual(result["ssh_server"], "workbench.aws.com")
self.assertEqual(result["login_user"], "root")
def test_fail_to_add_same_remote_port_in_same_host2(self):
targets = {'food-nonfood': {
'name': 'text-classification',
'local_port': '8888', 'login_user': None, 'remote_port': '9999', 'server_port': None,
'type': 'L', 'remote_host': 'localhost', 'ssh_server': 'my-ml-instance.ml.aws.com'
}
}
add_command = AddCommand("image-processing", None, 'L', 'localhost', '9999', '7777',
'my-ml-instance.ml.aws.com', None, None,
None)
self.assertEqual("9999", add_command.generate_consistent_target(targets)["remote_port"])
def test_add_target_without_remote_port(self):
targets = {'food-nonfood': {
'name': 'text-classification',
'local_port': '8888', 'login_user': None,
'remote_port': '6000', 'server_port': None,
'type': 'L', 'remote_host': 'localhost',
'ssh_server': 'my-ml-instance.ml.aws.com'
}
}
add_command = AddCommand("image-processing", None, "L",
"localhost", None, None,
"my-ml-instance.ml.aws.com", None, None, None)
self.assertEqual("49152", add_command.generate_consistent_target(targets)["remote_port"])
def add(ctx, name, forward_type, local_port, remote_port, ssh_server,
server_port, remote_host, login_user, ssh_argument):
try:
AddCommand(name, ssh_argument, forward_type, remote_host, remote_port,
local_port, ssh_server, server_port, login_user,
ctx.obj["config"]).run()
except RuntimeError as error:
logger.warn("Failed to register...")
logger.warn(error)
def test_raise_exception_with_inadiquate_parameters(self):
add_command = AddCommand("image-processing", None, "L",
"localhost", None, "8888",
None, None, None, None)
self.assertRaises(RuntimeError, lambda: add_command.generate_consistent_target({}))