diff options
Diffstat (limited to 'src/onaptests/steps/onboard/pnf.py')
| -rw-r--r-- | src/onaptests/steps/onboard/pnf.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/onaptests/steps/onboard/pnf.py b/src/onaptests/steps/onboard/pnf.py index 4e3303e..75fdbcf 100644 --- a/src/onaptests/steps/onboard/pnf.py +++ b/src/onaptests/steps/onboard/pnf.py @@ -60,6 +60,12 @@ class PnfOnboardStep(BaseStep): ) pnf.onboard() + @BaseStep.store_state(cleanup=True) + def cleanup(self): + pnf: Pnf = Pnf(name=settings.PNF_NAME) + pnf.delete() + super().cleanup() + class YamlTemplatePnfOnboardStep(YamlTemplateBaseStep): """PNF onboard using YAML template step.""" @@ -124,3 +130,11 @@ class YamlTemplatePnfOnboardStep(YamlTemplateBaseStep): artifact=pnf["pnf_artifact_file_path"] ) pnf_obj.onboard() + + @YamlTemplateBaseStep.store_state(cleanup=True) + def cleanup(self): + if "pnfs" in self.yaml_template: + for pnf in self.yaml_template["pnfs"]: + pnf_obj: Pnf = Pnf(name=pnf["pnf_name"]) + pnf_obj.delete() + super().cleanup() |