SCSTestExceptions must be raised inside with block

Signed-off-by: Katharina Trentau <[email protected]>

Author: fraugabel

Tests/kaas/k8s-default-storage-class/helper.py

@@ -31,6 +31,7 @@ class SCSTestException(Exception):
     """Raised when an Specific test did not pass"""
 
     def __init__(self, *args, return_code: int):
+        super().__init__(*args)
         self.return_code = return_code
 
 

Tests/kaas/k8s-default-storage-class/k8s-default-storage-class-check.py

@@ -261,7 +261,13 @@ def __exit__(self, exc_type, exc_value, traceback):
         self.clean()
         if self.return_code == 0:
             self.return_message = "all tests passed"
-
+        if isinstance(exc_value, SCSTestException):
+            # Get the return_code from the exception
+            self.return_code = exc_value.return_code
+            print(f"SCSTestException occurred with return_code: {self.return_code}")
+        else:
+            # No specific exception, handle normally
+            print(f"Exiting the context with return_code: {self.return_code}")
         logger.debug(f"return_code:{self.return_code} {self.return_message}")
 
         gen_sonobuoy_result_file(self.return_code, self.return_message, os.path.basename(__file__))