def test_delete_dir(self):
hosts = ['localhost', '127.0.0.1', 'st2build001']
client = ParallelSSHClient(hosts=hosts,
user='******',
pkey_file='~/.ssh/id_rsa',
connect=True)
client.delete_dir('/remote/stuff/', force=True)
expected_kwargs = {'force': True, 'timeout': None}
for host in hosts:
hostname, _ = client._get_host_port_info(host)
client._hosts_client[hostname].delete_dir.assert_called_with(
'/remote/stuff/', **expected_kwargs)
def test_delete_dir(self):
hosts = ["localhost", "127.0.0.1", "st2build001"]
client = ParallelSSHClient(hosts=hosts,
user="******",
pkey_file="~/.ssh/id_rsa",
connect=True)
client.delete_dir("/remote/stuff/", force=True)
expected_kwargs = {"force": True, "timeout": None}
for host in hosts:
hostname, _ = client._get_host_port_info(host)
client._hosts_client[hostname].delete_dir.assert_called_with(
"/remote/stuff/", **expected_kwargs)
def test_delete_dir(self):
hosts = ['localhost', '127.0.0.1', 'st2build001']
client = ParallelSSHClient(hosts=hosts,
user='******',
pkey_file='~/.ssh/id_rsa',
connect=True)
client.delete_dir('/remote/stuff/', force=True)
expected_kwargs = {
'force': True,
'timeout': None
}
for host in hosts:
hostname, _ = client._get_host_port_info(host)
client._hosts_client[hostname].delete_dir.assert_called_with('/remote/stuff/',
**expected_kwargs)
def main(user, pkey, password, hosts_str, cmd, file_path, dir_path,
delete_dir):
hosts = hosts_str.split(",")
client = ParallelSSHClient(user=user,
pkey_file=pkey,
password=password,
hosts=hosts)
pp = pprint.PrettyPrinter(indent=4)
if file_path:
if not os.path.exists(file_path):
raise Exception("File not found.")
results = client.put(file_path, "/home/lakshmi/test_file", mode="0660")
pp.pprint("Copy results: \n%s" % results)
results = client.run("ls -rlth")
pp.pprint("ls results: \n%s" % results)
if dir_path:
if not os.path.exists(dir_path):
raise Exception("File not found.")
results = client.put(dir_path, "/home/lakshmi/", mode="0660")
pp.pprint("Copy results: \n%s" % results)
results = client.run("ls -rlth")
pp.pprint("ls results: \n%s" % results)
if cmd:
results = client.run(cmd)
pp.pprint("cmd results: \n%s" % results)
if delete_dir:
results = client.delete_dir(delete_dir, force=True)
pp.pprint("Delete results: \n%s" % results)