add CheckConnectivity scenario

Change-Id: I9d246828790467c2a57ba410826ee9751fff89c5
Signed-off-by: JingLu5 <lvjing5@huawei.com>
(cherry picked from commit 4712d72a570dc9e2799227d489ee41768881a06d)

1 files changed, 99 insertions, 0 deletions

diff --git a/yardstick/benchmark/scenarios/lib/check_connectivity.py b/yardstick/benchmark/scenarios/lib/check_connectivity.py
new file mode 100644
index 000000000..bdf52d4c8
--- /dev/null
+++ b/yardstick/benchmark/scenarios/lib/check_connectivity.py
@@ -0,0 +1,99 @@
+##############################################################################
+# Copyright (c) 2017 Huawei Technologies Co.,Ltd 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
+##############################################################################
+
+from __future__ import print_function
+from __future__ import absolute_import
+
+import logging
+import yardstick.ssh as ssh
+
+from yardstick.benchmark.scenarios import base
+
+LOG = logging.getLogger(__name__)
+
+
+class CheckConnectivity(base.Scenario):
+    """Check connectivity between two VMs"""
+
+    __scenario_type__ = "CheckConnectivity"
+
+    def __init__(self, scenario_cfg, context_cfg):
+        self.scenario_cfg = scenario_cfg
+        self.context_cfg = context_cfg
+        self.options = self.scenario_cfg['options']
+
+        try:
+            self.source_ip_addr = self.options['src_ip_addr']
+            self.dest_ip_addr = self.options['dest_ip_addr']
+        except KeyError:
+            host = self.context_cfg['host']
+            target = self.context_cfg['target']
+            self.ssh_user = host.get('user', 'ubuntu')
+            self.ssh_port = host.get("ssh_port", 22)
+            self.source_ip_addr = host.get('ip', None)
+            self.dest_ip_addr = target.get('ipaddr', None)
+            self.ssh_key = host.get('key_filename', '/root/.ssh/id_rsa')
+            self.ssh_passwd = host.get('password', None)
+            self.ssh_timeout = 600
+        else:
+            self.ssh_user = self.options.get("ssh_user", 'ubuntu')
+            self.ssh_port = self.options.get("ssh_port", 22)
+            self.ssh_key = self.options.get("ssh_key", '/root/.ssh/id_rsa')
+            self.ssh_passwd = self.options.get("ssh_passwd", None)
+            self.ssh_timeout = self.options.get("ssh_timeout", 600)
+
+        self.connection = None
+        self.setup_done = False
+
+    def setup(self):
+        """scenario setup"""
+
+        if self.ssh_passwd is not None:
+            LOG.info("Log in via pw, user:%s, host:%s, pw:%s",
+                     self.ssh_user, self.source_ip_addr, self.ssh_passwd)
+            self.connection = ssh.SSH(self.ssh_user,
+                                      self.source_ip_addr,
+                                      password=self.ssh_passwd,
+                                      port=self.ssh_port)
+        else:
+            LOG.info("Log in via key, user:%s, host:%s, key_filename:%s",
+                     self.ssh_user, self.source_ip_addr, self.ssh_key)
+            self.connection = ssh.SSH(self.ssh_user,
+                                      self.source_ip_addr,
+                                      key_filename=self.ssh_key,
+                                      port=self.ssh_port)
+
+        self.connection.wait(timeout=self.ssh_timeout)
+        self.setup_done = True
+
+    def run(self, result):     # pragma: no cover
+        """execute the test"""
+
+        if not self.setup_done:
+            self.setup()
+
+        cmd = 'ping -c 4 ' + self.dest_ip_addr
+        parameter = self.options.get('ping_parameter', None)
+        if parameter:
+            cmd += (" %s" % parameter)
+
+        LOG.info("Executing command: %s", cmd)
+        LOG.info("ping %s ==> %s", self.source_ip_addr, self.dest_ip_addr)
+        status, stdout, stderr = self.connection.execute(cmd)
+
+        conn_status = self.scenario_cfg['sla']['status']
+
+        if bool(status) != bool(conn_status):
+            LOG.info("%s ===> %s connectivity check passed!" % (self.source_ip_addr,
+                                                                self.dest_ip_addr))
+            result['Check_Connectivity'] = 'PASS'
+        else:
+            LOG.info("%s ===> %s connectivity check failed!" % (self.source_ip_addr,
+                                                                self.dest_ip_addr))
+            result['Check_Connectivity'] = 'FAIL'