from onaptests.steps.instantiate.vf_module_ala_carte import YamlTemplateVfModuleAlaCarteInstantiateStep
 
 class BasicVm(testcase.TestCase):
-    """Onboard then instantiate a simple VM though ONAP."""
+    """Onboard then instantiate a simple VM with ONAP."""
 
     __logger = logging.getLogger(__name__)
 
         self.__logger.debug("start time")
         self.test.execute()
         self.__logger.info("VNF basic_vm successfully created")
-        if not settings.CLEANUP_FLAG:
-            self.result = 100
-            self.stop_time = time.time()
-            return testcase.TestCase.EX_OK
-
-    def clean(self):
-        """Clean VNF."""
+        self.stop_time = time.time()
+        # The cleanup is part of the test, not only a teardown action
         if settings.CLEANUP_FLAG:
+            self.__logger.info("VNF basic_vm cleanup called")
             time.sleep(settings.CLEANUP_ACTIVITY_TIMER)
-        try:
             self.test.cleanup()
-        except ValueError as error:
-            self.__logger.info("service instance deleted as expected {0}".format(error))
             self.result = 100
-            self.stop_time = time.time()
-            return testcase.TestCase.EX_OK
+        else:
+            self.__logger.info("No cleanup requested. Test completed.")
+            self.result = 100
+
+
+    def clean(self):
+        """Clean Additional resources if needed."""
+        pass
 
 from onaptests.steps.instantiate.vf_module_ala_carte import YamlTemplateVfModuleAlaCarteInstantiateStep
 
 class ClearwaterIms(testcase.TestCase):
-    """Onboard then instantiate a simple VM though ONAP."""
+    """Onboard then instantiate a clearwater vIMS with ONAP."""
 
     __logger = logging.getLogger(__name__)
 
         self.start_time = time.time()
         self.__logger.debug("start time")
         self.test.execute()
-        self.__logger.info("vIMS successfully created")
-        # Space for running additional tests on the deployed VNF here
-        if not settings.CLEANUP_FLAG:
-            self.result = 100
-            self.stop_time = time.time()
-            return testcase.TestCase.EX_OK
-
-    def clean(self):
-        """Clean VNF."""
+        self.__logger.info("VNF clearwater IMS successfully created")
+        self.stop_time = time.time()
+        # The cleanup is part of the test, not only a teardown action
         if settings.CLEANUP_FLAG:
+            self.__logger.info("VNF clearwater IMS cleanup called")
             time.sleep(settings.CLEANUP_ACTIVITY_TIMER)
-        try:
             self.test.cleanup()
-        except ValueError as error:
-            self.__logger.info("service instance deleted as expected {0}".format(error))
             self.result = 100
-            self.stop_time = time.time()
-            return testcase.TestCase.EX_OK
+        else:
+            self.__logger.info("No cleanup requested. Test completed.")
+            self.result = 100
+
+    def clean(self):
+        """Clean Additional resources if needed."""
+        pass