diff options
Diffstat (limited to 'src/onaptests/steps')
-rw-r--r-- | src/onaptests/steps/instantiate/vl_ala_carte.py | 153 | ||||
-rw-r--r-- | src/onaptests/steps/onboard/service.py | 27 | ||||
-rw-r--r-- | src/onaptests/steps/onboard/vf.py | 9 | ||||
-rw-r--r-- | src/onaptests/steps/onboard/vsp.py | 15 |
4 files changed, 188 insertions, 16 deletions
diff --git a/src/onaptests/steps/instantiate/vl_ala_carte.py b/src/onaptests/steps/instantiate/vl_ala_carte.py new file mode 100644 index 0000000..3a77c73 --- /dev/null +++ b/src/onaptests/steps/instantiate/vl_ala_carte.py @@ -0,0 +1,153 @@ +import time +import re +from typing import Iterable +from uuid import uuid4 +from yaml import load + +from onapsdk.aai.business import Customer, ServiceInstance, ServiceSubscription +from onapsdk.configuration import settings +from onapsdk.so.instantiation import Subnet +from onapsdk.sdc.service import Service +from onapsdk.vid import LineOfBusiness, Platform + +import onaptests.utils.exceptions as onap_test_exceptions +from ..base import YamlTemplateBaseStep +from .service_ala_carte import YamlTemplateServiceAlaCarteInstantiateStep + + +class YamlTemplateVlAlaCarteInstantiateStep(YamlTemplateBaseStep): + """Instantiate vl a'la carte using YAML template.""" + + def __init__(self, cleanup=False): + """Initialize step. + + Substeps: + - YamlTemplateServiceAlaCarteInstantiateStep. + """ + super().__init__(cleanup=cleanup) + self._yaml_template: dict = None + self._service_instance_name: str = None + self._service_instance: ServiceInstance = None + self.add_step(YamlTemplateServiceAlaCarteInstantiateStep(cleanup)) + + @property + def yaml_template(self) -> dict: + """Step YAML template. + + Load from file if it's a root step, get from parent otherwise. + + Returns: + dict: Step YAML template + + """ + if self.is_root: + if not self._yaml_template: + with open(settings.SERVICE_YAML_TEMPLATE, "r") as yaml_template: + self._yaml_template: dict = load(yaml_template) + return self._yaml_template + return self.parent.yaml_template + + @property + def service_name(self) -> str: + """Service name. + + Get from YAML template if it's a root step, get from parent otherwise. + + Returns: + str: Service name + + """ + if self.is_root: + return next(iter(self.yaml_template.keys())) + return self.parent.service_name + + @property + def service_instance_name(self) -> str: + """Service instance name. + + Generate using `service_name` and `uuid4()` function if it's a root step, + get from parent otherwise. + + Returns: + str: Service instance name + + """ + if self.is_root: + if not self._service_instance_name: + self._service_instance_name: str = f"{self.service_name}-{str(uuid4())}" + return self._service_instance_name + return self.parent.service_instance_name + + def get_subnets(self, network_name: str) -> Iterable[Subnet]: + """Get Network parameters from YAML template. + + Args: + network_name (str): Network name to get parameters for. + + Yields: + Iterator[Iterable[Subnet]]: Subnets + + """ + # workaround, as Network name differs from model name (added " 0") + network_name=re.sub(r"\s\d$", r"", network_name) + for net in self.yaml_template[self.service_name]["networks"]: + if net["vl_name"] == network_name: + if net['subnets'] is None: + print("No Subnet defined") + else: + for subnet in net['subnets']: + yield Subnet( + name=subnet['subnet-name'], + start_address=subnet['start-address'], + gateway_address=subnet['gateway-address'], + cidr_mask=subnet['cidr-mask'], + ip_version=subnet['ip-version'], + dhcp_enabled=subnet['dhcp-enabled']) + + @YamlTemplateBaseStep.store_state + def execute(self) -> None: + """Instantiate Vl. + + Use settings values: + - GLOBAL_CUSTOMER_ID. + + Raises: + Exception: Vl instantiation failed + + """ + super().execute() + service: Service = Service(self.service_name) + customer: Customer = Customer.get_by_global_customer_id(settings.GLOBAL_CUSTOMER_ID) + service_subscription: ServiceSubscription = customer.get_service_subscription_by_service_type(self.service_name) + service_instance: ServiceInstance = service_subscription.get_service_instance_by_name(self.service_instance_name) + self._service_instance = service_instance + line_of_business: LineOfBusiness = LineOfBusiness(settings.LINE_OF_BUSINESS) + platform: Platform = Platform(settings.PLATFORM) + for idx, network in enumerate(service.networks): + #for network in self.yaml_template[self.service_name]["networks"]: + net_instantiation = service_instance.add_network( + network, + line_of_business, + platform, + network_instance_name=f"{self.service_instance_name}_net_{idx}", + subnets=self.get_subnets(network.name)) + while not net_instantiation.finished: + time.sleep(10) + if net_instantiation.failed: + raise onap_test_exceptions.NetworkInstantiateException + + def cleanup(self) -> None: + """Cleanup VL. + + Raises: + Exception: VL cleaning failed + """ + if settings.CLEANUP_FLAG: + for net_instance in self._service_instance.network_instances: + self._logger.info("Start network deletion %s",net_instance.name) + net_deletion = net_instance.delete() + try: + net_deletion.wait_for_finish() + except TimeoutError: + raise onap_test_exceptions.NetworkCleanupException + super().cleanup() diff --git a/src/onaptests/steps/onboard/service.py b/src/onaptests/steps/onboard/service.py index 87211bd..86a2e6a 100644 --- a/src/onaptests/steps/onboard/service.py +++ b/src/onaptests/steps/onboard/service.py @@ -3,6 +3,7 @@ from yaml import load from onapsdk.configuration import settings from onapsdk.sdc.service import Service from onapsdk.sdc.vf import Vf +from onapsdk.sdc.vl import Vl from ..base import BaseStep, YamlTemplateBaseStep from .vf import VfOnboardStep, YamlTemplateVfOnboardStep @@ -25,13 +26,21 @@ class ServiceOnboardStep(BaseStep): """Onboard service. Use settings values: + - VL_NAME, - VF_NAME, - SERVICE_NAME. """ super().execute() - vf: Vf = Vf(name=settings.VF_NAME) - service: Service = Service(name=settings.SERVICE_NAME, resources=[vf]) + service: Service = Service(name=settings.SERVICE_NAME) + service.create() + if settings.VL_NAME != "": + vl: Vl = Vl(name=settings.VL_NAME) + service.add_resource(vl) + if settings.VF_NAME != "": + vf: Vf = Vf(name=settings.VF_NAME) + service.add_resource(vf) + service.checkin() service.onboard() @@ -84,7 +93,15 @@ class YamlTemplateServiceOnboardStep(YamlTemplateBaseStep): def execute(self): """Onboard service.""" super().execute() - service: Service = Service(name=self.service_name, - resources=[Vf(name=vnf["vnf_name"]) \ - for vnf in self.yaml_template[self.service_name]["vnfs"]]) + service: Service = Service(name=settings.SERVICE_NAME) + service.create() + if "networks" in self.yaml_template[self.service_name]: + for net in self.yaml_template[self.service_name]["networks"]: + vl: Vl = Vl(name=net['vl_name']) + service.add_resource(vl) + if "vnfs" in self.yaml_template[self.service_name]: + for vnf in self.yaml_template[self.service_name]["vnfs"]: + vf: Vf = Vf(name=vnf["vnf_name"]) + service.add_resource(vf) + service.checkin() service.onboard() diff --git a/src/onaptests/steps/onboard/vf.py b/src/onaptests/steps/onboard/vf.py index f26d66d..18b56f3 100644 --- a/src/onaptests/steps/onboard/vf.py +++ b/src/onaptests/steps/onboard/vf.py @@ -61,7 +61,8 @@ class YamlTemplateVfOnboardStep(YamlTemplateBaseStep): def execute(self): """Onboard Vfs from YAML template.""" super().execute() - for vnf in self.yaml_template["vnfs"]: - vsp: Vsp = Vsp(name=f"{vnf['vnf_name']}_VSP") - vf: Vf = Vf(name=vnf['vnf_name'], vsp=vsp) - vf.onboard() + if "vnfs" in self.yaml_template: + for vnf in self.yaml_template["vnfs"]: + vsp: Vsp = Vsp(name=f"{vnf['vnf_name']}_VSP") + vf: Vf = Vf(name=vnf['vnf_name'], vsp=vsp) + vf.onboard() diff --git a/src/onaptests/steps/onboard/vsp.py b/src/onaptests/steps/onboard/vsp.py index ba6020a..e0761ab 100644 --- a/src/onaptests/steps/onboard/vsp.py +++ b/src/onaptests/steps/onboard/vsp.py @@ -68,10 +68,11 @@ class YamlTemplateVspOnboardStep(YamlTemplateBaseStep): """ super().execute() vendor: Vendor = Vendor(name=settings.VENDOR_NAME) - for vnf in self.yaml_template["vnfs"]: - with open( - sys.path[-1] + "/" + vnf["heat_files_to_upload"], "rb") as package: - vsp: Vsp = Vsp(name=f"{vnf['vnf_name']}_VSP", - vendor=vendor, - package=package) - vsp.onboard() + if "vnfs" in self.yaml_template: + for vnf in self.yaml_template["vnfs"]: + with open( + sys.path[-1] + "/" + vnf["heat_files_to_upload"], "rb") as package: + vsp: Vsp = Vsp(name=f"{vnf['vnf_name']}_VSP", + vendor=vendor, + package=package) + vsp.onboard() |