934eaf78af
The aim of this patch is to not only spawn a bunch of vm's but also run some stress test inside the vm's.The patch creates jumphost on network with fip and all other vm's on the same neutron network so that jumphost can access the other vm's and run the stress tests. You need to set the following params to run this plugin: - specify the external network in browbeat-config.yaml - In the group_vars/all.yml install_browbeat_workloads: true, browbeat_network, in browbeat-workloads enable stress-ng Change-Id: Ica15af5de0c60916fd89661fb411e4bd10382b8f
120 lines
5.0 KiB
Python
120 lines
5.0 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import logging
|
|
|
|
from rally_openstack import consts
|
|
from rally_openstack.scenarios.vm import utils as vm_utils
|
|
from rally_openstack.scenarios.neutron import utils as neutron_utils
|
|
from rally.common import sshutils
|
|
from rally.task import scenario
|
|
from rally.task import types
|
|
from rally.task import validation
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
@types.convert(image={"type": "glance_image"},
|
|
flavor={"type": "nova_flavor"})
|
|
@validation.add("image_valid_on_flavor", flavor_param="flavor",
|
|
image_param="image")
|
|
@validation.add("number", param_name="port", minval=1, maxval=65535,
|
|
nullable=True, integer_only=True)
|
|
@validation.add("external_network_exists", param_name="floating_network")
|
|
@validation.add("required_services", services=[consts.Service.NOVA])
|
|
@validation.add("required_param_or_context",
|
|
param_name="image", ctx_name="image_command_customizer")
|
|
@validation.add("required_platform", platform="openstack", users=True)
|
|
@scenario.configure(context={"cleanup@openstack": ["nova", "neutron"],
|
|
"keypair@openstack": {},
|
|
"allow_ssh@openstack": None},
|
|
name="BrowbeatPlugin.stress_ng",
|
|
platform="openstack")
|
|
class BrowbeatStressNg(vm_utils.VMScenario, neutron_utils.NeutronScenario):
|
|
|
|
def run(self, flavor, username, ssh_timeout, num_clients,
|
|
command, image=None, floating_network=None, port=22,
|
|
use_floating_ip=True, **kwargs):
|
|
"""Create a jumphost on network with fip and all
|
|
other vm's on the same neutron network so that jumphost
|
|
can access the other vm's and run the stress tests
|
|
|
|
:param flavor: VM flavor name
|
|
:param username: ssh username on server
|
|
:param ssh_timeout: ssh timeout in seconds
|
|
:param num_clients: no.of clients
|
|
:param command: command that runs inside the client vm's
|
|
:param floating_network: external network name, for floating ip
|
|
:param port: ssh port for SSH connection
|
|
:param use_floating_ip: bool, floating or fixed IP for SSH connection
|
|
:param kwargs: optional args to create a VM
|
|
"""
|
|
jump_host, fip = self._boot_server_with_fip(
|
|
image, flavor, use_floating_ip=use_floating_ip,
|
|
floating_network=floating_network,
|
|
key_name=self.context["user"]["keypair"]["name"],
|
|
**kwargs)
|
|
ssh = sshutils.SSH(username, fip["ip"], port=port, pkey=self.context[
|
|
"user"]["keypair"]["private"])
|
|
self._wait_for_ssh(ssh, timeout=ssh_timeout)
|
|
|
|
# Write id_rsa to get to guests.
|
|
self._run_command_over_ssh(ssh, {'remote_path': "rm -rf ~/.ssh"})
|
|
self._run_command_over_ssh(ssh, {'remote_path': "mkdir ~/.ssh"})
|
|
ssh.run(
|
|
"cat > ~/.ssh/id_rsa",
|
|
stdin=self.context["user"]["keypair"]["private"])
|
|
ssh.execute("chmod 0600 ~/.ssh/id_rsa")
|
|
|
|
_clients = self.create_clients(
|
|
ssh, num_clients, image, flavor, username, **kwargs)
|
|
|
|
# Run stress test
|
|
for sip in _clients:
|
|
cmd = " {} 'ssh {}@{}' ".format(command, username, sip)
|
|
exitcode, stdout, stderr = ssh.execute(cmd)
|
|
LOG.error(" couldn't run the stress-ng command: {}".format(stderr))
|
|
|
|
def create_clients(self, jump_ssh, num_clients, image, flavor, user, **kwargs):
|
|
"""Creates Client VM's
|
|
|
|
:param jump_ssh: ssh connection
|
|
:param num_clients: no.of clients
|
|
:param image: VM image name
|
|
:param flavor: VM flavor name
|
|
:param user: ssh username on server
|
|
:param kwargs: optional args to create a VM
|
|
"""
|
|
_clients = []
|
|
for i in range(num_clients):
|
|
LOG.info("Launching Client : {}".format(i))
|
|
server = self._boot_server(
|
|
image,
|
|
flavor,
|
|
key_name=self.context["user"]["keypair"]["name"],
|
|
**kwargs)
|
|
for net in server.addresses:
|
|
network_name = net
|
|
break
|
|
if network_name is None:
|
|
return False
|
|
# IP Address
|
|
_clients.append(
|
|
str(server.addresses[network_name][0]["addr"]))
|
|
|
|
for sip in _clients:
|
|
cmd = "ssh -o StrictHostKeyChecking=no {}@{} /bin/true".format(
|
|
user, sip)
|
|
s1_exitcode, s1_stdout, s1_stderr = jump_ssh.execute(cmd)
|
|
return _clients
|