def test_vdbench_run(self, test_vars): # noqa: F811 log = logging.getLogger("test_vdbench_run") node_ip = test_vars["deploy_vd_outputs"]["node_0_ip_address"]["value"] with SSHTunnelForwarder( test_vars["public_ip"], ssh_username=test_vars["controller_user"], ssh_pkey=test_vars["ssh_priv_key"], remote_bind_address=(node_ip, 22), ) as ssh_tunnel: sleep(1) try: ssh_client = create_ssh_client( test_vars["controller_user"], "127.0.0.1", ssh_tunnel.local_bind_port, key_filename=test_vars["ssh_priv_key"]) scp_client = SCPClient(ssh_client.get_transport()) try: scp_client.put(test_vars["ssh_priv_key"], r"~/.ssh/id_rsa") finally: scp_client.close() commands = """ ~/copy_idrsa.sh cd ./run_vdbench.sh inmem32node3.conf uniquestring1 """.split("\n") run_ssh_commands(ssh_client, commands) finally: ssh_client.close()
def test_edasim_run(self, test_vars): # noqa: F811 log = logging.getLogger("test_edasim_run") node_ip = test_vars["deploy_edasim_outputs"][ "jobsubmitter_0_ip_address"]["value"] with SSHTunnelForwarder( test_vars["public_ip"], ssh_username=test_vars["controller_user"], ssh_pkey=test_vars["ssh_priv_key"], remote_bind_address=(node_ip, 22), ) as ssh_tunnel: sleep(1) try: ssh_client = create_ssh_client( test_vars["controller_user"], "127.0.0.1", ssh_tunnel.local_bind_port, key_filename=test_vars["ssh_priv_key"]) run_ssh_command(ssh_client, ". .profile && ./jobrun.sh testrun") finally: ssh_client.close()
def test_client_docker_run(self, test_vars): # noqa: F811 log = logging.getLogger("test_client_docker_run") node_ip = test_vars["deploy_client_docker_outputs"]["node_0_ip_address"]["value"] atd = test_vars["atd_obj"] cluster_mgmt_ip = test_vars["cluster_mgmt_ip"] with SSHTunnelForwarder( test_vars["public_ip"], ssh_username=test_vars["controller_user"], ssh_pkey=test_vars["ssh_priv_key"], remote_bind_address=(node_ip, 22), ) as ssh_tunnel: sleep(1) try: ssh_client = create_ssh_client( test_vars["controller_user"], "127.0.0.1", ssh_tunnel.local_bind_port, key_filename=test_vars["ssh_priv_key"] ) scp_client = SCPClient(ssh_client.get_transport()) try: scp_client.put(test_vars["ssh_priv_key"], r"~/.ssh/id_rsa") finally: scp_client.close() commands = """ cd curl -fsSL https://get.docker.com -o get-docker.sh sudo sh get-docker.sh sudo docker login https://{0} -u {1} -p {2} sudo docker pull {0}/test echo "export STORAGEACT='{3}'" >> ~/.bashrc echo "export MGMIP='{4}'" >> ~/.bashrc echo "export SA_KEY='{5}'" >> ~/.bashrc echo "export CLUSTER_MGMT_IP='{6}'" >> ~/.bashrc echo "export ADMIN_PW='{7}'" >> ~/.bashrc """.format(os.environ["dockerRegistry"], os.environ["dockerUsername"], os.environ["dockerPassword"], atd.deploy_id + "sa", test_vars["public_ip"], os.environ["SA_KEY"], cluster_mgmt_ip, os.environ["AVERE_ADMIN_PW"]).split("\n") run_ssh_commands(ssh_client, commands) finally: ssh_client.close()