aboutsummaryrefslogtreecommitdiffstats
path: root/functest/opnfv_tests/OpenStack/vPing/vping_ssh.py
diff options
context:
space:
mode:
Diffstat (limited to 'functest/opnfv_tests/OpenStack/vPing/vping_ssh.py')
-rw-r--r--functest/opnfv_tests/OpenStack/vPing/vping_ssh.py175
1 files changed, 175 insertions, 0 deletions
diff --git a/functest/opnfv_tests/OpenStack/vPing/vping_ssh.py b/functest/opnfv_tests/OpenStack/vPing/vping_ssh.py
new file mode 100644
index 00000000..6fb35a7a
--- /dev/null
+++ b/functest/opnfv_tests/OpenStack/vPing/vping_ssh.py
@@ -0,0 +1,175 @@
+#!/usr/bin/python
+#
+# Copyright (c) 2015 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 re
+import sys
+import time
+
+import argparse
+import paramiko
+from scp import SCPClient
+
+import functest.utils.functest_logger as ft_logger
+import functest.utils.openstack_utils as os_utils
+import vping_base
+from functest.core.TestCasesBase import TestCasesBase
+
+
+class VPingSSH(vping_base.VPingBase):
+
+ def __init__(self):
+ super(VPingSSH, self).__init__()
+ self.case_name = 'vping_ssh'
+ self.logger = ft_logger.Logger(self.case_name).getLogger()
+
+ def do_vping(self, vm, test_ip):
+ floatip = self.add_float_ip(vm)
+ if not floatip:
+ return TestCasesBase.EX_RUN_ERROR
+ ssh = self.establish_ssh(vm, floatip)
+ if not ssh:
+ return TestCasesBase.EX_RUN_ERROR
+ if not self.transfer_ping_script(ssh, floatip):
+ return TestCasesBase.EX_RUN_ERROR
+ return self.do_vping_ssh(ssh, test_ip)
+
+ def add_float_ip(self, vm):
+ self.logger.info("Creating floating IP for VM '%s'..." % self.vm2_name)
+ floatip_dic = os_utils.create_floating_ip(self.neutron_client)
+ floatip = floatip_dic['fip_addr']
+
+ if floatip is None:
+ self.logger.error("Cannot create floating IP.")
+ return None
+ self.logger.info("Floating IP created: '%s'" % floatip)
+
+ self.logger.info("Associating floating ip: '%s' to VM '%s' "
+ % (floatip, self.vm2_name))
+ if not os_utils.add_floating_ip(self.nova_client, vm.id, floatip):
+ self.logger.error("Cannot associate floating IP to VM.")
+ return None
+
+ return floatip
+
+ def establish_ssh(self, vm, floatip):
+ self.logger.info("Trying to establish SSH connection to %s..."
+ % floatip)
+ username = 'cirros'
+ password = 'cubswin:)'
+ ssh = paramiko.SSHClient()
+ ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+ timeout = 50
+ nolease = False
+ got_ip = False
+ discover_count = 0
+ cidr_first_octet = self.private_subnet_cidr.split('.')[0]
+ while timeout > 0:
+ try:
+ ssh.connect(floatip, username=username,
+ password=password, timeout=2)
+ self.logger.debug("SSH connection established to %s."
+ % floatip)
+ break
+ except:
+ self.logger.debug("Waiting for %s..." % floatip)
+ time.sleep(6)
+ timeout -= 1
+
+ console_log = vm.get_console_output()
+
+ # print each "Sending discover" captured on the console log
+ if (len(re.findall("Sending discover", console_log)) >
+ discover_count and not got_ip):
+ discover_count += 1
+ self.logger.debug("Console-log '%s': Sending discover..."
+ % self.vm2_name)
+
+ # check if eth0 got an ip,the line looks like this:
+ # "inet addr:192.168."....
+ # if the dhcp agent fails to assing ip, this line will not appear
+ if "inet addr:" + cidr_first_octet in console_log and not got_ip:
+ got_ip = True
+ self.logger.debug("The instance '%s' succeeded to get the IP "
+ "from the dhcp agent." % self.vm2_name)
+
+ # if dhcp not work,it shows "No lease, failing".The test will fail
+ if "No lease, failing" in console_log \
+ and not nolease \
+ and not got_ip:
+ nolease = True
+ self.logger.debug("Console-log '%s': No lease, failing..."
+ % self.vm2_name)
+ self.logger.info("The instance failed to get an IP from DHCP "
+ "agent. The test will probably timeout...")
+
+ if timeout == 0: # 300 sec timeout (5 min)
+ self.logger.error("Cannot establish connection to IP '%s'. "
+ "Aborting" % floatip)
+ return None
+ return ssh
+
+ def transfer_ping_script(self, ssh, floatip):
+ self.logger.info("Trying to transfer ping.sh to %s..." % floatip)
+ scp = SCPClient(ssh.get_transport())
+
+ ping_script = os.path.join(self.repo + "ping.sh")
+ try:
+ scp.put(ping_script, "~/")
+ except:
+ self.logger.error("Cannot SCP the file '%s' to VM '%s'"
+ % (ping_script, floatip))
+ return False
+
+ cmd = 'chmod 755 ~/ping.sh'
+ (stdin, stdout, stderr) = ssh.exec_command(cmd)
+ for line in stdout.readlines():
+ print line
+
+ return True
+
+ def do_vping_ssh(self, ssh, test_ip):
+ EXIT_CODE = -1
+ self.logger.info("Waiting for ping...")
+
+ sec = 0
+ cmd = '~/ping.sh ' + test_ip
+ flag = False
+
+ while True:
+ time.sleep(1)
+ (stdin, stdout, stderr) = ssh.exec_command(cmd)
+ output = stdout.readlines()
+
+ for line in output:
+ if "vPing OK" in line:
+ self.logger.info("vPing detected!")
+ EXIT_CODE = 0
+ flag = True
+ break
+
+ elif sec == self.ping_timeout:
+ self.logger.info("Timeout reached.")
+ flag = True
+ break
+ if flag:
+ break
+ self.logger.debug("Pinging %s. Waiting for response..." % test_ip)
+ sec += 1
+ return EXIT_CODE
+
+
+if __name__ == '__main__':
+ args_parser = argparse.ArgumentParser()
+ args_parser.add_argument("-r", "--report",
+ help="Create json result file",
+ action="store_true")
+ args = vars(args_parser.parse_args())
+ sys.exit(vping_base.VPingMain(VPingSSH).main(**args))