1 ##############################################################################
2 # Copyright (c) 2015 Ericsson AB and others.
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 ##############################################################################
15 import yardstick.ssh as ssh
16 from yardstick.benchmark.scenarios import base
18 LOG = logging.getLogger(__name__)
21 class Ping(base.Scenario):
22 """Execute ping between two hosts
25 packetsize - number of data bytes to send
31 __scenario_type__ = "Ping"
33 TARGET_SCRIPT = 'ping_benchmark.bash'
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 ip = host.get('ip', None)
43 key_filename = host.get('key_filename', '~/.ssh/id_rsa')
45 LOG.info("user:%s, host:%s", user, ip)
47 self.connection = ssh.SSH(user, ip, key_filename=key_filename)
48 self.connection.wait()
50 def run(self, result):
51 """execute the benchmark"""
53 if "options" in self.scenario_cfg:
55 self.scenario_cfg['options'].get("packetsize", '56')
59 destination = self.context_cfg['target'].get("ipaddr", '127.0.0.1')
61 LOG.debug("ping '%s' '%s'", options, destination)
63 exit_status, stdout, stderr = self.connection.execute(
64 "/bin/sh -s {0} {1}".format(destination, options),
65 stdin=open(self.target_script, "r"))
68 raise RuntimeError(stderr)
70 result["rtt"] = float(stdout)
72 if "sla" in self.scenario_cfg:
73 sla_max_rtt = int(self.scenario_cfg["sla"]["max_rtt"])
74 assert result["rtt"] <= sla_max_rtt, "rtt %f > sla:max_rtt(%f); " % \
75 (result["rtt"], sla_max_rtt)
79 '''internal test function'''
80 key_filename = pkg_resources.resource_filename("yardstick.resources",
81 "files/yardstick_key")
84 "ip": "10.229.47.137",
86 "key_filename": key_filename
89 "ipaddr": "10.229.17.105",
93 logger = logging.getLogger("yardstick")
94 logger.setLevel(logging.DEBUG)
103 if __name__ == '__main__':