yardstick/benchmark/scenarios/networking/ping.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Ericsson AB and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 # ping scenario
  11
  12 import pkg_resources
  13 import logging
  14
  15 import yardstick.ssh as ssh
  16 from yardstick.benchmark.scenarios import base
  17
  18 LOG = logging.getLogger(__name__)
  19
  20
  21 class Ping(base.Scenario):
  22     """Execute ping between two hosts
  23
  24   Parameters
  25     packetsize - number of data bytes to send
  26         type:    int
  27         unit:    bytes
  28         default: 56
  29     """
  30
  31     __scenario_type__ = "Ping"
  32
  33     TARGET_SCRIPT = 'ping_benchmark.bash'
  34
  35     def __init__(self, scenario_cfg, context_cfg):
  36         self.scenario_cfg = scenario_cfg
  37         self.context_cfg = context_cfg
  38         self.target_script = pkg_resources.resource_filename(
  39             'yardstick.benchmark.scenarios.networking', Ping.TARGET_SCRIPT)
  40         host = self.context_cfg['host']
  41         user = host.get('user', 'ubuntu')
  42         ssh_port = host.get("ssh_port", ssh.DEFAULT_PORT)
  43         ip = host.get('ip', None)
  44         key_filename = host.get('key_filename', '/root/.ssh/id_rsa')
  45         password = host.get('password', None)
  46
  47         if password is not None:
  48             LOG.info("Log in via pw, user:%s, host:%s, pw:%s",
  49                      user, ip, password)
  50             self.connection = ssh.SSH(user, ip, password=password,
  51                                       port=ssh_port)
  52         else:
  53             LOG.info("Log in via key, user:%s, host:%s, key_filename:%s",
  54                      user, ip, key_filename)
  55             self.connection = ssh.SSH(user, ip, key_filename=key_filename,
  56                                       port=ssh_port)
  57
  58         self.connection.wait(timeout=600)
  59
  60     def run(self, result):
  61         """execute the benchmark"""
  62
  63         if "options" in self.scenario_cfg:
  64             options = "-s %s" % \
  65                 self.scenario_cfg['options'].get("packetsize", '56')
  66         else:
  67             options = ""
  68
  69         destination = self.context_cfg['target'].get('ipaddr', '127.0.0.1')
  70         dest_list = [s.strip() for s in destination.split(',')]
  71
  72         result["rtt"] = {}
  73         rtt_result = result["rtt"]
  74
  75         for pos, dest in enumerate(dest_list):
  76             if 'targets' in self.scenario_cfg:
  77                 target_vm = self.scenario_cfg['targets'][pos]
  78             else:
  79                 target_vm = self.scenario_cfg['target']
  80
  81             LOG.debug("ping '%s' '%s'", options, dest)
  82             with open(self.target_script, "r") as stdin_file:
  83                 exit_status, stdout, stderr = self.connection.execute(
  84                     "/bin/sh -s {0} {1}".format(dest, options),
  85                     stdin=stdin_file)
  86
  87             if exit_status != 0:
  88                 raise RuntimeError(stderr)
  89
  90             if stdout:
  91                 target_vm_name = target_vm.split('.')[0]
  92                 rtt_result[target_vm_name] = float(stdout)
  93                 if "sla" in self.scenario_cfg:
  94                     sla_max_rtt = int(self.scenario_cfg["sla"]["max_rtt"])
  95                     assert rtt_result[target_vm_name] <= sla_max_rtt,\
  96                         "rtt %f > sla: max_rtt(%f); " % \
  97                         (rtt_result[target_vm_name], sla_max_rtt)
  98             else:
  99                 LOG.error("ping '%s' '%s' timeout", options, target_vm)
 100
 101
 102 def _test():    # pragma: no cover
 103     '''internal test function'''
 104     key_filename = pkg_resources.resource_filename("yardstick.resources",
 105                                                    "files/yardstick_key")
 106     ctx = {
 107         "host": {
 108             "ip": "10.229.47.137",
 109             "user": "root",
 110             "key_filename": key_filename
 111         },
 112         "target": {
 113             "ipaddr": "10.229.17.105",
 114         }
 115     }
 116
 117     logger = logging.getLogger("yardstick")
 118     logger.setLevel(logging.DEBUG)
 119
 120     args = {}
 121     result = {}
 122
 123     p = Ping(args, ctx)
 124     p.run(result)
 125     print result
 126
 127 if __name__ == '__main__':    # pragma: no cover
 128     _test()