1 ##############################################################################
2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd 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 ##############################################################################
13 import yardstick.ssh as ssh
14 from yardstick.benchmark.scenarios import base
16 LOG = logging.getLogger(__name__)
19 class Ping6(base.Scenario): # pragma: no cover
20 """Execute ping6 between two hosts
22 read link below for more ipv6 info description:
23 http://wiki.opnfv.org/ipv6_opnfv_project
25 __scenario_type__ = "Ping6"
27 TARGET_SCRIPT = 'ping6_benchmark.bash'
28 PRE_SETUP_SCRIPT = 'ping6_pre_setup.bash'
29 SETUP_SCRIPT = 'ping6_setup.bash'
30 SETUP_ODL_SCRIPT = 'ping6_setup_with_odl.bash'
31 FIND_HOST_SCRIPT = 'ping6_find_host.bash'
32 TEARDOWN_SCRIPT = 'ping6_teardown.bash'
33 METADATA_SCRIPT = 'ping6_metadata.txt'
34 RADVD_SCRIPT = 'ping6_radvd.conf'
35 POST_TEARDOWN_SCRIPT = 'ping6_post_teardown.bash'
37 def __init__(self, scenario_cfg, context_cfg):
38 self.scenario_cfg = scenario_cfg
39 self.context_cfg = context_cfg
40 self.nodes = context_cfg['nodes']
41 self.options = scenario_cfg['options']
42 self.setup_done = False
44 self.external_network = self.options.get("external_network", "ext-net")
45 self.ping_options = "-s %s -c %s" % \
46 (self.options.get("packetsize", '56'),
47 self.options.get("ping_count", '5'))
48 self.openrc = self.options.get("openrc", "/opt/admin-openrc.sh")
50 def _ssh_host(self, node_name):
52 node = self.nodes.get(node_name, None)
53 user = node.get('user', 'ubuntu')
54 ip = node.get('ip', None)
55 pwd = node.get('password', None)
56 key_fname = node.get('key_filename', '/root/.ssh/id_rsa')
59 LOG.debug("Log in via pw, user:%s, host:%s, password:%s",
61 self.client = ssh.SSH(user, ip, password=pwd)
63 LOG.debug("Log in via key, user:%s, host:%s, key_filename:%s",
65 self.client = ssh.SSH(user, ip, key_filename=key_fname)
66 self.client.wait(timeout=60)
69 for node_name in self.host_list:
70 self._ssh_host(node_name)
71 self.client.run("cat > ~/pre_setup.sh",
72 stdin=open(self.pre_setup_script, "rb"))
73 status, stdout, stderr = self.client.execute(
74 "sudo bash pre_setup.sh")
78 self.setup_script = pkg_resources.resource_filename(
79 'yardstick.benchmark.scenarios.networking',
82 self.setup_odl_script = pkg_resources.resource_filename(
83 'yardstick.benchmark.scenarios.networking',
84 Ping6.SETUP_ODL_SCRIPT)
86 self.pre_setup_script = pkg_resources.resource_filename(
87 'yardstick.benchmark.scenarios.networking',
88 Ping6.PRE_SETUP_SCRIPT)
90 self.ping6_metadata_script = pkg_resources.resource_filename(
91 'yardstick.benchmark.scenarios.networking',
92 Ping6.METADATA_SCRIPT)
94 self.ping6_radvd_script = pkg_resources.resource_filename(
95 'yardstick.benchmark.scenarios.networking',
98 host_str = self.options.get("host", 'host1')
99 self.host_list = host_str.split(',')
100 self.host_list.sort()
101 pre_setup = self.options.get("pre_setup", True)
106 self._ssh_host(self.host_list[0])
108 self.client.run("cat > ~/metadata.txt",
109 stdin=open(self.ping6_metadata_script, "rb"))
111 # run script to setup ipv6 with nosdn or odl
112 sdn = self.options.get("sdn", 'nosdn')
114 self.client.run("cat > ~/br-ex.radvd.conf",
115 stdin=open(self.ping6_radvd_script, "rb"))
116 self.client.run("cat > ~/setup_odl.sh",
117 stdin=open(self.setup_odl_script, "rb"))
118 setup_bash_file = "setup_odl.sh"
120 self.client.run("cat > ~/setup.sh",
121 stdin=open(self.setup_script, "rb"))
122 setup_bash_file = "setup.sh"
123 cmd = "sudo bash %s %s %s" % \
124 (setup_bash_file, self.openrc, self.external_network)
125 LOG.debug("Executing setup command: %s", cmd)
126 status, stdout, stderr = self.client.execute(cmd)
128 self.setup_done = True
130 def run(self, result):
131 """execute the benchmark"""
133 self.ping6_script = pkg_resources.resource_filename(
134 'yardstick.benchmark.scenarios.networking',
137 self.ping6_find_host_script = pkg_resources.resource_filename(
138 'yardstick.benchmark.scenarios.networking',
139 Ping6.FIND_HOST_SCRIPT)
140 if not self.setup_done:
141 host_str = self.options.get("host", 'host1')
142 self.host_list = host_str.split(',')
143 self.host_list.sort()
144 self._ssh_host(self.host_list[0])
146 # find ipv4-int-network1 to ssh VM
147 self.client.run("cat > ~/find_host.sh",
148 stdin=open(self.ping6_find_host_script, "rb"))
149 cmd = "sudo bash find_host.sh %s" % self.openrc
150 LOG.debug("Executing find_host command: %s", cmd)
151 status, stdout, stderr = self.client.execute(cmd)
152 host_name = stdout.strip()
154 # copy vRouterKey to target host
155 self.client.run("cat ~/vRouterKey",
156 stdout=open("/tmp/vRouterKey", "w"))
157 self._ssh_host(host_name)
158 self.client.run("cat > ~/vRouterKey",
159 stdin=open("/tmp/vRouterKey", "rb"))
161 # run ping6 benchmark
162 self.client.run("cat > ~/ping6.sh",
163 stdin=open(self.ping6_script, "rb"))
164 cmd = "sudo bash ping6.sh %s %s" % (self.openrc, self.ping_options)
165 LOG.debug("Executing ping6 command: %s", cmd)
166 status, stdout, stderr = self.client.execute(cmd)
169 raise RuntimeError(stderr)
173 result["rtt"] = float(stdout)
174 if "sla" in self.scenario_cfg:
175 sla_max_rtt = int(self.scenario_cfg["sla"]["max_rtt"])
176 assert result["rtt"] <= sla_max_rtt, \
177 "rtt %f > sla:max_rtt(%f); " % (result["rtt"], sla_max_rtt)
179 LOG.error("ping6 timeout!!!")
183 """teardown the benchmark"""
185 self.post_teardown_script = pkg_resources.resource_filename(
186 'yardstick.benchmark.scenarios.networking',
187 Ping6.POST_TEARDOWN_SCRIPT)
189 host_str = self.options.get("host", 'node1')
190 self.host_list = host_str.split(',')
191 self.host_list.sort()
193 if not self.run_done:
194 self._ssh_host(self.host_list[0])
196 self.teardown_script = pkg_resources.resource_filename(
197 'yardstick.benchmark.scenarios.networking',
198 Ping6.TEARDOWN_SCRIPT)
199 self.client.run("cat > ~/teardown.sh",
200 stdin=open(self.teardown_script, "rb"))
201 cmd = "sudo bash teardown.sh %s %s" % \
202 (self.openrc, self.external_network)
203 status, stdout, stderr = self.client.execute(cmd)
205 post_teardown = self.options.get("post_teardown", True)
207 self._post_teardown()
210 raise RuntimeError(stderr)
215 LOG.error("ping6 teardown failed")
217 def _post_teardown(self):
218 for node_name in self.host_list:
219 self._ssh_host(node_name)
220 self.client.run("cat > ~/post_teardown.sh",
221 stdin=open(self.post_teardown_script, "rb"))
222 status, stdout, stderr = self.client.execute(
223 "sudo bash post_teardown.sh")