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 FIND_HOST_SCRIPT = 'ping6_find_host.bash'
31 TEARDOWN_SCRIPT = 'ping6_teardown.bash'
32 METADATA_SCRIPT = 'ping6_metadata.txt'
33 POST_TEARDOWN_SCRIPT = 'ping6_post_teardown.bash'
35 def __init__(self, scenario_cfg, context_cfg):
36 self.scenario_cfg = scenario_cfg
37 self.context_cfg = context_cfg
38 self.setup_done = False
42 for node_name in self.host_list:
43 self._ssh_host(node_name)
44 self.client.run("cat > ~/pre_setup.sh",
45 stdin=open(self.pre_setup_script, "rb"))
46 status, stdout, stderr = self.client.execute(
47 "sudo bash pre_setup.sh")
49 def _ssh_host(self, node_name):
52 nodes = self.context_cfg['nodes']
53 node = nodes.get(node_name, None)
54 host_user = node.get('user', 'ubuntu')
55 host_ip = node.get('ip', None)
56 host_pwd = node.get('password', 'root')
57 LOG.debug("user:%s, host:%s", host_user, host_ip)
58 self.client = ssh.SSH(host_user, host_ip, password=host_pwd)
59 self.client.wait(timeout=600)
63 self.setup_script = pkg_resources.resource_filename(
64 'yardstick.benchmark.scenarios.networking',
67 self.pre_setup_script = pkg_resources.resource_filename(
68 'yardstick.benchmark.scenarios.networking',
69 Ping6.PRE_SETUP_SCRIPT)
71 self.ping6_metadata_script = pkg_resources.resource_filename(
72 'yardstick.benchmark.scenarios.networking',
73 Ping6.METADATA_SCRIPT)
75 options = self.scenario_cfg['options']
76 host_str = options.get("host", 'host1')
77 self.host_list = host_str.split(',')
79 pre_setup = options.get("pre_setup", True)
84 self._ssh_host(self.host_list[0])
85 # run script to setup ipv6
86 self.client.run("cat > ~/setup.sh",
87 stdin=open(self.setup_script, "rb"))
88 self.client.run("cat > ~/metadata.txt",
89 stdin=open(self.ping6_metadata_script, "rb"))
90 cmd = "sudo bash setup.sh"
91 status, stdout, stderr = self.client.execute(cmd)
93 self.setup_done = True
95 def run(self, result):
96 """execute the benchmark"""
98 self.ping6_script = pkg_resources.resource_filename(
99 'yardstick.benchmark.scenarios.networking',
102 self.ping6_find_host_script = pkg_resources.resource_filename(
103 'yardstick.benchmark.scenarios.networking',
104 Ping6.FIND_HOST_SCRIPT)
106 if not self.setup_done:
107 options = self.scenario_cfg['options']
108 host_str = options.get("host", 'host1')
109 self.host_list = host_str.split(',')
110 self.host_list.sort()
111 self._ssh_host(self.host_list[0])
112 self.client.run("cat > ~/find_host.sh",
113 stdin=open(self.ping6_find_host_script, "rb"))
114 cmd = "sudo bash find_host.sh"
115 LOG.debug("Executing command: %s", cmd)
116 status, stdout, stderr = self.client.execute(cmd)
117 host_name = stdout.strip()
119 self.client.run("cat ~/vRouterKey",
120 stdout=open("/tmp/vRouterKey", "w"))
121 self._ssh_host(host_name)
123 self.client.run("cat > ~/vRouterKey",
124 stdin=open("/tmp/vRouterKey", "rb"))
126 self.client.run("cat > ~/ping6.sh",
127 stdin=open(self.ping6_script, "rb"))
128 cmd = "sudo bash ping6.sh"
129 LOG.debug("Executing command: %s", cmd)
130 status, stdout, stderr = self.client.execute(cmd)
133 raise RuntimeError(stderr)
136 result["rtt"] = float(stdout)
138 if "sla" in self.scenario_cfg:
139 sla_max_rtt = int(self.scenario_cfg["sla"]["max_rtt"])
140 assert result["rtt"] <= sla_max_rtt, \
141 "rtt %f > sla:max_rtt(%f); " % (result["rtt"], sla_max_rtt)
143 LOG.error("ping6 timeout")
147 """teardown the benchmark"""
149 self.post_teardown_script = pkg_resources.resource_filename(
150 'yardstick.benchmark.scenarios.networking',
151 Ping6.POST_TEARDOWN_SCRIPT)
153 options = self.scenario_cfg['options']
154 host_str = options.get("host", 'node1')
155 self.host_list = host_str.split(',')
156 self.host_list.sort()
158 if not self.run_done:
159 self._ssh_host(self.host_list[0])
161 self.teardown_script = pkg_resources.resource_filename(
162 'yardstick.benchmark.scenarios.networking',
163 Ping6.TEARDOWN_SCRIPT)
164 self.client.run("cat > ~/teardown.sh",
165 stdin=open(self.teardown_script, "rb"))
166 cmd = "sudo bash teardown.sh"
167 status, stdout, stderr = self.client.execute(cmd)
169 post_teardown = options.get("post_teardown", True)
171 self._post_teardown()
174 raise RuntimeError(stderr)
179 LOG.error("ping6 teardown failed")
181 def _post_teardown(self):
182 for node_name in self.host_list:
183 self._ssh_host(node_name)
184 self.client.run("cat > ~/post_teardown.sh",
185 stdin=open(self.post_teardown_script, "rb"))
186 status, stdout, stderr = self.client.execute(
187 "sudo bash post_teardown.sh")