1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
##############################################################################
# Copyright (c) 2015 Ericsson AB 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
##############################################################################
# ping scenario
import pkg_resources
import logging
import yardstick.ssh as ssh
from yardstick.benchmark.scenarios import base
LOG = logging.getLogger(__name__)
class Ping(base.Scenario):
"""Executes a ping benchmark between two hosts"""
__scenario_type__ = "Ping"
TARGET_SCRIPT = 'ping_benchmark.bash'
def __init__(self, context):
self.context = context
self.target_script = pkg_resources.resource_filename(
'yardstick.benchmark.scenarios.networking', Ping.TARGET_SCRIPT)
user = self.context.get('user', 'ubuntu')
host = self.context.get('host', None)
key_filename = self.context.get('key_filename', '~/.ssh/id_rsa')
LOG.debug("user:%s, host:%s", user, host)
self.connection = ssh.SSH(user, host, key_filename=key_filename)
self.connection.wait()
def run(self, args):
"""execute the benchmark"""
self.options = "-s %s" % args['options'].get("packetsize", '56')
self.ipaddr = args.get("ipaddr", '127.0.0.1')
LOG.debug("ping %s %s", self.options, self.ipaddr)
exit_status, stdout, stderr = self.connection.execute(
"/bin/sh -s {0} {1}".format(self.ipaddr, self.options),
stdin=open(self.target_script, "r"))
if exit_status != 0:
raise RuntimeError(stderr)
rtt = float(stdout)
if "sla" in args:
sla_max_rtt = int(args["sla"]["max_rtt"])
assert rtt <= sla_max_rtt, "rtt %f > sla_max_rtt" % rtt
return rtt
|