# 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 consts as rally_consts from rally import exceptions from rally_openstack.cleanup import manager as resource_manager from rally_openstack.contexts.manila import consts from rally_openstack.scenarios.manila import utils as manila_utils from rally.task import context 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