aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorwu.zhihui <wu.zhihui1@zte.com.cn>2017-01-19 15:49:10 +0800
committerwu.zhihui <wu.zhihui1@zte.com.cn>2017-01-23 10:24:49 +0800
commitcf7acd2b1f054c220da1a1582a005035bfd92a56 (patch)
treecd7d5a742693d958ba976fd3e90e9c9c18b5237d
parent1d4610afd990101eead7ede56e6252577ab245d8 (diff)
refactor useful legacy code about preparing test enviroment
- move useful scripts to qtip/scripts - refactor some functions which belong to env_setup.py and add them to qtip/runner/env.py Change-Id: Ibf6c27103dfc45016920f4b602cd72d703c1f636 Signed-off-by: wu.zhihui <wu.zhihui1@zte.com.cn>
-rwxr-xr-xqtip/scripts/fetch_compute_ips.sh (renamed from legacy/scripts/fetch_compute_ips.sh)2
-rwxr-xr-xqtip/scripts/qtip_creds.sh (renamed from legacy/scripts/qtip_creds.sh)0
-rw-r--r--qtip/util/env.py137
3 files changed, 138 insertions, 1 deletions
diff --git a/legacy/scripts/fetch_compute_ips.sh b/qtip/scripts/fetch_compute_ips.sh
index 4bdc9a48..a2618ec9 100755
--- a/legacy/scripts/fetch_compute_ips.sh
+++ b/qtip/scripts/fetch_compute_ips.sh
@@ -2,7 +2,6 @@
##############################################################################
#Copyright (c) 2016 Ericsson AB, ZTE and others.
#jose.lausuch@ericsson.com
-#wu.zhihui1@zte.com.cn
#All rights reserved. This program and the accompanying materials
#are made available under the terms of the Apache License, Version 2.0
#which accompanies this distribution, and is available at
@@ -109,6 +108,7 @@ if [ -z "$IPS" ]; then
error "The compute node $IPS are not up. Please check that the POD is correctly deployed."
else
echo "-------- all compute node ips: --------"
+ rm $HOME/ips.log
touch $HOME/ips.log
echo "$IPS" > $HOME/qtip/ips.log
echo $IPS
diff --git a/legacy/scripts/qtip_creds.sh b/qtip/scripts/qtip_creds.sh
index af051ac5..af051ac5 100755
--- a/legacy/scripts/qtip_creds.sh
+++ b/qtip/scripts/qtip_creds.sh
diff --git a/qtip/util/env.py b/qtip/util/env.py
new file mode 100644
index 00000000..4e7a31c7
--- /dev/null
+++ b/qtip/util/env.py
@@ -0,0 +1,137 @@
+##############################################################################
+# Copyright (c) 2016 Dell Inc, ZTE and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+import os
+import paramiko
+import socket
+import time
+from os import path
+from os.path import expanduser
+
+SCRIPT_DIR = path.join(path.dirname(__file__), path.pardir, 'scripts')
+CONFIG_DIR = path.join(path.dirname(__file__), path.pardir, path.pardir,
+ 'config')
+PRIVATE_KEY = CONFIG_DIR + '/QtipKey'
+PUBLIC_KEY = CONFIG_DIR + '/QtipKey.pub'
+IPS_FILE = expanduser('~') + "/qtip/ips.log"
+HOST_FILE = CONFIG_DIR + "/host"
+
+
+def fetch_compute_ips_via_installer():
+ clean_file(IPS_FILE)
+
+ installer_type = str(os.environ['INSTALLER_TYPE'].lower())
+ installer_ip = str(os.environ['INSTALLER_IP'])
+ if installer_type not in ["fuel"]:
+ raise RuntimeError("%s is not supported" % installer_type)
+ if not installer_ip:
+ raise RuntimeError("undefine environment variable INSTALLER_IP")
+
+ cmd = "bash %s/fetch_compute_ips.sh -i %s -a %s" % \
+ (SCRIPT_DIR, installer_type, installer_ip)
+ os.system(cmd)
+ if path.isfile(IPS_FILE):
+ return True
+ else:
+ return False
+
+
+def parse_ips():
+ ip_list = []
+ with open(IPS_FILE, "r") as outfile:
+ data = outfile.read()
+ if data:
+ ip_list.extend(data.rstrip('\n').split('\n'))
+ return ip_list
+
+
+def ssh_test(ip):
+ os.system('ssh-keyscan %s >> /root/.ssh/known_hosts' % ip)
+ time.sleep(2)
+
+ ssh_cmd = '%s/qtip_creds.sh %s' % (SCRIPT_DIR, ip)
+ os.system(ssh_cmd)
+
+ ssh = paramiko.SSHClient()
+ ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+ ssh.connect(ip, key_filename='{0}/QtipKey'.format(CONFIG_DIR))
+
+ for attempts in range(100):
+ try:
+ stdin, stdout, stderr = ssh.exec_command('uname')
+ if not stderr.readlines():
+ print("{0}: SSH test successful")
+ return True
+ except socket.error:
+ if attempts == 99:
+ return False
+ time.sleep(2)
+
+
+def ping_test(ip, attempts=30):
+ ping_cmd = 'ping -D -c1 {0}'.format(ip)
+ for i in range(attempts):
+ if os.system(ping_cmd):
+ print('\nWaiting for machine\n')
+ time.sleep(10)
+ else:
+ print('\n\n %s is UP \n\n ' % ip)
+ return True
+ if i == 29:
+ return False
+
+
+def check_nodes_connectivity():
+ ip_list = parse_ips()
+ for ip in ip_list:
+ if not ping_test(ip):
+ raise RuntimeError("{0}: Ping test failed".format(ip))
+ if not ssh_test(ip):
+ raise RuntimeError("{0}: SSH test failed".format(ip))
+
+
+def generate_host_file():
+ ip_list = parse_ips()
+ with open(HOST_FILE, 'w') as host_file:
+ for index, item in enumerate(ip_list):
+ host_file.write("[host_{0}]\n".format(index))
+ host_file.write(item + '\n')
+
+
+def generate_keypair():
+ """Generating ssh keypair"""
+ if not clean_keypair():
+ raise RuntimeError("Cann't remove old keypair")
+
+ cmd = "ssh-keygen -t rsa -N "" -f {0} -q".format(PRIVATE_KEY)
+ os.system(cmd)
+
+ if path.isfile(PRIVATE_KEY) and path.isfile(PUBLIC_KEY):
+ return True
+ else:
+ return False
+
+
+def clean_file(file_path):
+ try:
+ if path.isfile(file_path):
+ os.remove(file_path)
+ print("Removed: " + file_path)
+ else:
+ print("Not exists: " + file_path)
+ except OSError, error:
+ print("Not able to Remove: " + file_path, error)
+ return False
+ return True
+
+
+def clean_keypair():
+ flag = True
+ flag &= clean_file(PRIVATE_KEY)
+ flag &= clean_file(PUBLIC_KEY)
+ return flag