1 files changed, 114 insertions, 0 deletions
diff --git a/src/onaptests/steps/onboard/pnf.py b/src/onaptests/steps/onboard/pnf.py
new file mode 100644
index 0000000..12f2561
--- /dev/null
+++ b/src/onaptests/steps/onboard/pnf.py
@@ -0,0 +1,114 @@
+"""PNF onboarding step module."""
+
+from onapsdk.configuration import settings
+from onapsdk.sdc.pnf import Pnf
+from onapsdk.sdc.vendor import Vendor
+from ..base import BaseStep, YamlTemplateBaseStep
+from .vendor import VendorOnboardStep
+
+
+class PnfOnboardStep(BaseStep):
+    """PNF onboard step."""
+
+    def __init__(self, cleanup: bool = False) -> None:
+        """Step initialization.
+
+        Substeps:
+            - VendorOnboardStep.
+
+        Args:
+            cleanup(bool, optional): Determines if cleanup action should be called.
+
+        """
+        super().__init__(cleanup=cleanup)
+        self.add_step(VendorOnboardStep(cleanup=cleanup))
+
+    @property
+    def description(self) -> str:
+        """Step description."""
+        return "Onboard pnf in SDC."
+
+    @property
+    def component(self) -> str:
+        """Component name."""
+        return "SDC"
+
+    @BaseStep.store_state
+    def execute(self) -> None:
+        """Onboard PNF in SDC.
+
+        Use settings values:
+         - VENDOR_NAME,
+         - PNF_NAME,
+         - PNF_ARTIFACT_TYPE,
+         - PNF_ARTIFACT_NAME,
+         - PNF_ARTIFACT_LABEL,
+         - PNF_ARTIFACT_FILE_PATH
+
+        """
+        vendor: Vendor = Vendor(name=settings.VENDOR_NAME)
+        pnf: Pnf = Pnf(name=settings.PNF_NAME, vendor=vendor)
+        pnf.create()
+        pnf.add_deployment_artifact(
+            artifact_type=settings.PNF_ARTIFACT_TYPE,
+            artifact_name=settings.PNF_ARTIFACT_NAME,
+            artifact_label=settings.PNF_ARTIFACT_LABEL,
+            artifact=settings.PNF_ARTIFACT_FILE_PATH
+        )
+        pnf.onboard()
+
+
+class YamlTemplatePnfOnboardStep(YamlTemplateBaseStep):
+    """PNF onboard using YAML template step."""
+
+    def __init__(self, cleanup: bool = False) -> None:
+        """Step initialization.
+
+        Substeps:
+            - VendorOnboardStep.
+
+        Args:
+            cleanup(bool, optional): Determines if cleanup action should be called.
+
+        """
+        super().__init__(cleanup=cleanup)
+        self.add_step(VendorOnboardStep(cleanup=cleanup))
+
+    @property
+    def description(self) -> str:
+        """Step description."""
+        return "Onboard pnf using YAML template in SDC."
+
+    @property
+    def component(self) -> str:
+        """Component name."""
+        return "SDC"
+
+    @property
+    def yaml_template(self) -> dict:
+        """YAML template.
+
+        Get YAML template from parent using it's name.
+
+        Returns:
+            dict: YAML template
+
+        """
+        return self.parent.yaml_template[self.parent.service_name]
+
+    @YamlTemplateBaseStep.store_state
+    def execute(self):
+        """Onboard PNFs from YAML template."""
+        super().execute()
+        if "pnfs" in self.yaml_template:
+            vendor: Vendor = Vendor(name=settings.VENDOR_NAME)
+            for pnf in self.yaml_template["pnfs"]:
+                pnf_obj: Pnf = Pnf(name=pnf["pnf_name"], vendor=vendor)
+                pnf_obj.create()
+                pnf_obj.add_deployment_artifact(
+                    artifact_type=pnf["pnf_artifact_type"],
+                    artifact_name=pnf["pnf_artifact_name"],
+                    artifact_label=pnf["pnf_artifact_label"],
+                    artifact=pnf["pnf_artifact_file_path"]
+                )
+                pnf_obj.onboard()