+ print "========== run rubbos ==========="
+
+ nova = _get_nova_client()
+ instances = get_instances(nova)
+ if instances == None:
+ print "Found *None* instances, exit rubbos_run()!"
+ return False
+
+ control_public_ip = ""
+ control_server = ""
+ client_servers = ""
+ web_servers = ""
+ app_servers = ""
+ cjdbc_controller = ""
+ database_servers = ""
+ for instance in instances:
+ name = getattr(instance, 'name')
+ private_ip = [ x['addr'] for x in getattr(instance, 'addresses').itervalues().next() if x['OS-EXT-IPS:type'] == 'fixed']
+ public_ip = [ x['addr'] for x in getattr(instance, 'addresses').itervalues().next() if x['OS-EXT-IPS:type'] == 'floating']
+
+ if name.find("rubbos-control") >= 0:
+ control_public_ip = public_ip[0]
+ control_server = str(name) + ':' + public_ip[0] + ':' + private_ip[0]
+ if name.find("rubbos-client") >= 0:
+ client_servers = client_servers + str(name)+':'+private_ip[0] + ","
+ if name.find("rubbos-httpd") >= 0:
+ web_servers = web_servers + str(name)+':'+private_ip[0] + ","
+ if name.find("rubbos-tomcat") >= 0:
+ app_servers = app_servers + str(name) + ':' + private_ip[0] + ","
+ if name.find("rubbos-cjdbc") >= 0:
+ cjdbc_controller = str(name) + ':' + private_ip[0]
+ if name.find("rubbos-mysql") >= 0:
+ database_servers = database_servers + str(name) + ':' + private_ip[0] + ","
+
+ client_servers = client_servers[0:len(client_servers)-1]
+ web_servers = web_servers[0:len(web_servers)-1]
+ app_servers = app_servers[0:len(app_servers)-1]
+ database_servers = database_servers[0:len(database_servers)-1]
+ print "control_server: %s" % control_server
+ print "client_servers: %s" % client_servers
+ print "web_servers: %s" % web_servers
+ print "app_servers: %s" % app_servers
+ print "cjdbc_controller: %s" % cjdbc_controller
+ print "database_servers: %s" % database_servers
+ with open(Bottlenecks_repo_dir+"/testsuites/rubbos/puppet_manifests/internal/rubbos.conf") as temp_f, open('rubbos.conf', 'w') as new_f:
+ for line in temp_f.readlines():
+ if line.find("REPLACED_CONTROLLER") >= 0 :
+ new_f.write( line.replace("REPLACED_CONTROLLER", control_server) )
+ elif line.find("REPLACED_CLIENT_SERVERS") >= 0:
+ new_f.write( line.replace("REPLACED_CLIENT_SERVERS", client_servers) )
+ elif line.find("REPLACED_WEB_SERVERS") >= 0:
+ new_f.write( line.replace("REPLACED_WEB_SERVERS", web_servers) )
+ elif line.find("REPLACED_APP_SERVERS") >= 0:
+ new_f.write( line.replace("REPLACED_APP_SERVERS", app_servers) )
+ elif line.find("REPLACED_CJDBC_CONTROLLER") >= 0:
+ new_f.write( line.replace("REPLACED_CJDBC_CONTROLLER", cjdbc_controller) )
+ elif line.find("REPLACED_DB_SERVERS") >= 0:
+ new_f.write( line.replace("REPLACED_DB_SERVERS", database_servers) )
+ elif line.find("REPLACED_CLIENTS_PER_NODE") >= 0:
+ new_f.write( line.replace("REPLACED_CLIENTS_PER_NODE", "200 400 800 1600 3200") )
+ else:
+ new_f.write(line)
+ if os.path.exists("rubbos.conf") == False:
+ return False
+
+ cmd = "sudo chmod 0600 " + Bottlenecks_repo_dir + "/utils/infra_setup/bottlenecks_key/bottlenecks_key"
+ subprocess.call(cmd, shell=True)
+ ssh_args = "-o StrictHostKeyChecking=no -o BatchMode=yes -i " + Bottlenecks_repo_dir + "/utils/infra_setup/bottlenecks_key/bottlenecks_key "
+
+ print "############### Test #################"
+ cmd = 'ssh-keygen -f "/root/.ssh/known_hosts" -R ' + control_public_ip
+ subprocess.call(cmd, shell=True)
+ print "## Ping test:"
+ cmd = "ping -c 5 " + control_public_ip
+ print cmd
+ subprocess.call(cmd, shell=True)
+ print "## ssh date test:"
+ cmd = "ssh " + ssh_args + " ubuntu@" + control_public_ip + ' "date"'
+ print cmd
+ subprocess.call(cmd, shell=True)
+ print "## neutruon net-list:"
+ subprocess.call("neutron net-list", shell=True)
+ print "## nova list:"
+ subprocess.call("nova list", shell=True)
+ print "############### Test #################"
+
+ cmd = "scp " + ssh_args + "rubbos.conf ubuntu@" + control_public_ip + ":/home/ubuntu/"
+ print "Exec shell: " + cmd
+ subprocess.call(cmd, shell=True)
+
+ cmd = "scp " + ssh_args + Bottlenecks_repo_dir + "/testsuites/rubbos/puppet_manifests/internal/run_rubbos_internal.sh ubuntu@" + control_public_ip + ":/home/ubuntu/"
+ print "Exec shell: " + cmd
+ subprocess.call(cmd, shell=True)
+
+ # call remote run_rubbos_internal.sh
+ cmd = "ssh " + ssh_args + " ubuntu@" + control_public_ip + ' "sudo /home/ubuntu/run_rubbos_internal.sh /home/ubuntu/rubbos.conf /home/ubuntu/btnks-results" '
+ print "Exec shell: " + cmd
+ subprocess.call(cmd, shell=True)
+ cmd = "scp " + ssh_args + " ubuntu@" + control_public_ip + ":/home/ubuntu/btnks-results/rubbos.out ./rubbos.out"
+ print "Exec shell: " + cmd
+ subprocess.call(cmd, shell=True)
+ if os.path.exists("rubbos.out") == False:
+ print "Failed to fetch results from the rubbos_control node!"
+ return False
+
+ with open("rubbos.out") as f:
+ lines = f.readlines()
+ print "Rubbos results:"
+ for line in lines:
+ print line
+ return True