Decouple TestSuite from JUnitXml for inheritance

weiwei · Feb 25, 2025 · 16553e8 · 16553e8
1 parent d540e64
commit 16553e8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/junitparser/junitparser.py b/junitparser/junitparser.py
@@ -509,6 +509,7 @@ class TestSuite(Element):
     __test__ = False
 
     testcase = TestCase
+    root = None
 
     def __init__(self, name=None):
         super().__init__(self._tag)
@@ -552,7 +553,7 @@ def __add__(self, other):
             result.update_statistics()
         else:
             # Create a new test result containing two testsuites
-            result = JUnitXml()
+            result = JUnitXml() if TestSuite.root is None else TestSuite.root()
             result.add_testsuite(self)
             result.add_testsuite(other)
         return result
@@ -566,7 +567,7 @@ def __iadd__(self, other):
             self.update_statistics()
             return self
 
-        result = JUnitXml()
+        result = JUnitXml() if TestSuite.root is None else TestSuite.root()
         result.filepath = self.filepath
         result.add_testsuite(self)
         result.add_testsuite(other)

diff --git a/junitparser/xunit2.py b/junitparser/xunit2.py
@@ -168,6 +168,10 @@ class TestSuite(junitparser.TestSuite):
 
     testcase = TestCase
 
+    def __init__(self, name=None):
+        super().__init__(name)
+        TestSuite.root = JUnitXml
+
     @property
     def system_out(self):
         """<system-out>"""