Remove JunitCaseException exception object

src/junit_report/_junit_test_case.py

@@ -8,7 +8,7 @@
 from . import _utils
 from ._junit_decorator import JunitDecorator
 from ._junit_test_suite import JunitTestSuite
-from ._test_case_data import CaseFailure, JunitCaseException, TestCaseCategories, TestCaseData
+from ._test_case_data import CaseFailure, TestCaseCategories, TestCaseData, is_case_exception_already_raised
 
 
 class JunitTestCase(JunitDecorator):
@@ -45,10 +45,11 @@ def _add_failure(self, e: BaseException, message_prefix: str = ""):
 
     def _on_exception(self, e: BaseException):
         self._case_data.had_exception = True
-        if isinstance(e, JunitCaseException):
+        if is_case_exception_already_raised(e):
             raise  # already registered on son test case
+        setattr(e, "__is_junit_exception__", True)
         self._add_failure(e)
-        raise JunitCaseException(exception=e)
+        raise e
 
     def _on_wrapper_end(self) -> bool:
         self._case_data.set_fin_time()

src/junit_report/_junit_test_suite.py

@@ -8,7 +8,7 @@
 from junit_xml import TestCase, TestSuite, to_xml_report_string
 
 from ._junit_decorator import JunitDecorator
-from ._test_case_data import CaseFailure, JunitCaseException, TestCaseCategories, TestCaseData
+from ._test_case_data import CaseFailure, TestCaseCategories, TestCaseData, is_case_exception_already_raised
 
 
 class DuplicateSuiteError(KeyError):
@@ -196,8 +196,9 @@ def _collect_yield(self):
         self._export(self.suite)
 
     def _on_exception(self, e: BaseException):
-        if isinstance(e, JunitCaseException):
-            raise e.exception
+        if is_case_exception_already_raised(e):
+            raise e
+
         self._handle_in_suite_exception(e)
 
     def _handle_in_suite_exception(self, exception: BaseException):

src/junit_report/_test_case_data.py

@@ -6,6 +6,9 @@
 from junit_xml import TestCase
 
 
+JUNIT_EXCEPTION_ON_TAG = "__is_junit_exception__"
+
+
 class TestCaseCategories(Enum):
     FUNCTION = "function"
     FIXTURE = "fixture"
@@ -17,18 +20,6 @@ class MainRunner(Enum):
     NONE = "none"
 
 
-class JunitCaseException(BaseException):
-    def __init__(self, exception: BaseException, *args: object) -> None:
-        super().__init__(*args)
-        self.exception: BaseException = exception
-
-    def __str__(self) -> str:
-        return self.exception.__str__()
-
-    def __repr__(self) -> str:
-        return self.exception.__repr__()
-
-
 @dataclass
 class CaseFailure:
     message: str
@@ -73,3 +64,7 @@ def set_parametrize(self, params: List[Tuple[str, Any]]):
     def set_parent(self, case_parent_name: str):
         self._has_parent = True
         self.case.classname = f"{self.case.classname}.{case_parent_name}"
+
+
+def is_case_exception_already_raised(exception: BaseException) -> bool:
+    return hasattr(exception, JUNIT_EXCEPTION_ON_TAG)