5d12c4acbe
Also, made some minor changes to imports orders to fit PEP8 Change-Id: I1596989cac61602a3f417c9d60774354f5650424
210 lines
8.5 KiB
Python
210 lines
8.5 KiB
Python
# Copyright 2015 Mirantis Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
from rally.common import cfg
|
|
from rally.common import logging
|
|
from rally.common import utils
|
|
from rally.common import validation
|
|
from rally import exceptions
|
|
from rally.task import context
|
|
|
|
from rally_openstack.cleanup import manager as resource_manager
|
|
from rally_openstack import consts as rally_consts
|
|
from rally_openstack.contexts.manila import consts
|
|
from rally_openstack.scenarios.manila import utils as manila_utils
|
|
|
|
|
|
CONF = cfg.CONF
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
CONTEXT_NAME = consts.SHARE_NETWORKS_CONTEXT_NAME
|
|
|
|
SHARE_NETWORKS_ARG_DESCR = """
|
|
This context arg will be used only when context arg "use_share_networks" is
|
|
set to True.
|
|
|
|
If context arg 'share_networks' has values then they will be used else share
|
|
networks will be autocreated - one for each tenant network. If networks do not
|
|
exist then will be created one share network for each tenant without network
|
|
data.
|
|
|
|
Expected value is dict of lists where tenant Name or ID is key and list of
|
|
share_network Names or IDs is value. Example:
|
|
|
|
.. code-block:: json
|
|
|
|
"context": {
|
|
"manila_share_networks": {
|
|
"use_share_networks": true,
|
|
"share_networks": {
|
|
"tenant_1_name_or_id": ["share_network_1_name_or_id",
|
|
"share_network_2_name_or_id"],
|
|
"tenant_2_name_or_id": ["share_network_3_name_or_id"]}
|
|
}
|
|
}
|
|
|
|
Also, make sure that all 'existing users' in appropriate registered deployment
|
|
have share networks if its usage is enabled, else Rally will randomly take
|
|
users that does not satisfy criteria.
|
|
"""
|
|
|
|
|
|
@validation.add("required_platform", platform="openstack", users=True)
|
|
@context.configure(name=CONTEXT_NAME, platform="openstack", order=450)
|
|
class ShareNetworks(context.Context):
|
|
"""This context creates share networks for Manila project."""
|
|
CONFIG_SCHEMA = {
|
|
"type": "object",
|
|
"$schema": rally_consts.JSON_SCHEMA,
|
|
"properties": {
|
|
"use_share_networks": {
|
|
"type": "boolean",
|
|
"description": "Specifies whether manila should use share "
|
|
"networks for share creation or not."},
|
|
|
|
"share_networks": {
|
|
"type": "object",
|
|
"description": SHARE_NETWORKS_ARG_DESCR,
|
|
"additionalProperties": True
|
|
},
|
|
},
|
|
"additionalProperties": False
|
|
}
|
|
DEFAULT_CONFIG = {
|
|
"use_share_networks": False,
|
|
"share_networks": {},
|
|
}
|
|
|
|
def _setup_for_existing_users(self):
|
|
if (self.config["use_share_networks"] and
|
|
not self.config["share_networks"]):
|
|
msg = ("Usage of share networks was enabled but for deployment "
|
|
"with existing users share networks also should be "
|
|
"specified via arg 'share_networks'")
|
|
raise exceptions.ContextSetupFailure(
|
|
ctx_name=self.get_name(), msg=msg)
|
|
|
|
for tenant_name_or_id, share_networks in self.config[
|
|
"share_networks"].items():
|
|
# Verify project existence
|
|
for tenant in self.context["tenants"].values():
|
|
if tenant_name_or_id in (tenant["id"], tenant["name"]):
|
|
tenant_id = tenant["id"]
|
|
existing_user = None
|
|
for user in self.context["users"]:
|
|
if user["tenant_id"] == tenant_id:
|
|
existing_user = user
|
|
break
|
|
break
|
|
else:
|
|
msg = ("Provided tenant Name or ID '%s' was not found in "
|
|
"existing tenants.") % tenant_name_or_id
|
|
raise exceptions.ContextSetupFailure(
|
|
ctx_name=self.get_name(), msg=msg)
|
|
self.context["tenants"][tenant_id][CONTEXT_NAME] = {}
|
|
self.context["tenants"][tenant_id][CONTEXT_NAME][
|
|
"share_networks"] = []
|
|
|
|
manila_scenario = manila_utils.ManilaScenario({
|
|
"user": existing_user,
|
|
"config": {
|
|
"api_versions": self.context["config"].get(
|
|
"api_versions", [])}
|
|
})
|
|
existing_sns = manila_scenario._list_share_networks(
|
|
detailed=False, search_opts={"project_id": tenant_id})
|
|
|
|
for sn_name_or_id in share_networks:
|
|
# Verify share network existence
|
|
for sn in existing_sns:
|
|
if sn_name_or_id in (sn.id, sn.name):
|
|
break
|
|
else:
|
|
msg = ("Specified share network '%(sn)s' does not "
|
|
"exist for tenant '%(tenant_id)s'"
|
|
% {"sn": sn_name_or_id, "tenant_id": tenant_id})
|
|
raise exceptions.ContextSetupFailure(
|
|
ctx_name=self.get_name(), msg=msg)
|
|
|
|
# Set share network for project
|
|
self.context["tenants"][tenant_id][CONTEXT_NAME][
|
|
"share_networks"].append(sn.to_dict())
|
|
|
|
def _setup_for_autocreated_users(self):
|
|
# Create share network for each network of tenant
|
|
for user, tenant_id in (utils.iterate_per_tenants(
|
|
self.context.get("users", []))):
|
|
networks = self.context["tenants"][tenant_id].get("networks")
|
|
manila_scenario = manila_utils.ManilaScenario({
|
|
"task": self.task,
|
|
"owner_id": self.get_owner_id(),
|
|
"user": user,
|
|
"config": {
|
|
"api_versions": self.context["config"].get(
|
|
"api_versions", [])}
|
|
})
|
|
manila_scenario.RESOURCE_NAME_FORMAT = self.RESOURCE_NAME_FORMAT
|
|
self.context["tenants"][tenant_id][CONTEXT_NAME] = {
|
|
"share_networks": []}
|
|
data = {}
|
|
|
|
def _setup_share_network(tenant_id, data):
|
|
share_network = manila_scenario._create_share_network(
|
|
**data).to_dict()
|
|
self.context["tenants"][tenant_id][CONTEXT_NAME][
|
|
"share_networks"].append(share_network)
|
|
for ss in self.context["tenants"][tenant_id].get(
|
|
consts.SECURITY_SERVICES_CONTEXT_NAME, {}).get(
|
|
"security_services", []):
|
|
manila_scenario._add_security_service_to_share_network(
|
|
share_network["id"], ss["id"])
|
|
|
|
if networks:
|
|
for network in networks:
|
|
if network.get("cidr"):
|
|
data["nova_net_id"] = network["id"]
|
|
elif network.get("subnets"):
|
|
data["neutron_net_id"] = network["id"]
|
|
data["neutron_subnet_id"] = network["subnets"][0]
|
|
else:
|
|
LOG.warning("Can't determine network service provider."
|
|
" Share network will have no data.")
|
|
_setup_share_network(tenant_id, data)
|
|
else:
|
|
_setup_share_network(tenant_id, data)
|
|
|
|
def setup(self):
|
|
self.context[CONTEXT_NAME] = {}
|
|
if not self.config["use_share_networks"]:
|
|
pass
|
|
elif self.context["config"].get("existing_users"):
|
|
self._setup_for_existing_users()
|
|
else:
|
|
self._setup_for_autocreated_users()
|
|
|
|
def cleanup(self):
|
|
if (not self.context["config"].get("existing_users") or
|
|
self.config["use_share_networks"]):
|
|
resource_manager.cleanup(
|
|
names=["manila.share_networks"],
|
|
users=self.context.get("users", []),
|
|
superclass=self.__class__,
|
|
api_versions=self.context["config"].get("api_versions"),
|
|
task_id=self.get_owner_id())
|
|
else:
|
|
# NOTE(vponomaryov): assume that share networks were not created
|
|
# by test run.
|
|
return
|