diff options
Diffstat (limited to 'src/onaptests/steps/onboard/vf.py')
| -rw-r--r-- | src/onaptests/steps/onboard/vf.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/onaptests/steps/onboard/vf.py b/src/onaptests/steps/onboard/vf.py index 852cb82..f77f9af 100644 --- a/src/onaptests/steps/onboard/vf.py +++ b/src/onaptests/steps/onboard/vf.py @@ -47,6 +47,12 @@ class VfOnboardStep(BaseStep): if not vf.created(): vf.onboard() + @BaseStep.store_state(cleanup=True) + def cleanup(self): + vf: Vf = Vf(name=settings.VF_NAME) + vf.delete() + super().cleanup() + class YamlTemplateVfOnboardStep(YamlTemplateBaseStep): """Vf onboard using YAML template step.""" @@ -122,3 +128,11 @@ class YamlTemplateVfOnboardStep(YamlTemplateBaseStep): ) time.sleep(10) vf.onboard() + + @YamlTemplateBaseStep.store_state(cleanup=True) + def cleanup(self): + if "vnfs" in self.yaml_template: + for vnf in self.yaml_template["vnfs"]: + vf_obj: Vf = Vf(name=vnf["vnf_name"]) + vf_obj.delete() + super().cleanup() |