Introduce JUnitTreeOrException and ParsedJUnitFile

EnricoMi · Jun 21, 2022 · 7067fa3 · 7067fa3
1 parent 0bcc218
commit 7067fa3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/python/publish/junit.py b/python/publish/junit.py
@@ -120,13 +120,15 @@ def close(self) -> Element:
 
 
 JUnitTree = etree.ElementTree
+JUnitTreeOrException = Union[JUnitTree, BaseException]
+ParsedJUnitFile = Tuple[str, JUnitTreeOrException]
 
 
 def parse_junit_xml_files(files: Iterable[str],
                           drop_testcases: bool = False,
-                          progress: Callable[[Tuple[str, Union[JUnitTree, BaseException]]], Tuple[str, Union[JUnitTree, BaseException]]] = lambda x: x) -> Iterable[Tuple[str, Union[JUnitTree, BaseException]]]:
-    """Parses junit xml files and returns aggregated statistics as a ParsedUnitTestResults."""
-    def parse(path: str) -> Union[JUnitXml, BaseException]:
+                          progress: Callable[[ParsedJUnitFile], ParsedJUnitFile] = lambda x: x) -> Iterable[ParsedJUnitFile]:
+    def parse(path: str) -> JUnitTreeOrException:
+        """Parses a junit xml file and returns either a JUnitTree or an Exception."""
         if not os.path.exists(path):
             return FileNotFoundError(f'File does not exist.')
         if os.stat(path).st_size == 0:
@@ -143,8 +145,7 @@ def parse(path: str) -> Union[JUnitXml, BaseException]:
     return [progress((result_file, parse(result_file))) for result_file in files]
 
 
-def process_junit_xml_elems(trees: Iterable[Tuple[str, Union[JUnitTree, BaseException]]],
-                            time_factor: float = 1.0) -> ParsedUnitTestResults:
+def process_junit_xml_elems(trees: Iterable[ParsedJUnitFile], time_factor: float = 1.0) -> ParsedUnitTestResults:
     # TODO: move upstream into JUnitTree
     def create_junitxml(filepath: str, tree: JUnitTree) -> Union[JUnitXml, JUnitXmlError]:
         root_elem = tree.getroot()

diff --git a/python/publish/trx.py b/python/publish/trx.py
@@ -1,19 +1,19 @@
 import os
 import pathlib
-from typing import Iterable, Tuple, Union, Callable
+from typing import Iterable, Callable
 
 from lxml import etree
 
-from publish.junit import JUnitTree
+from publish.junit import JUnitTreeOrException, ParsedJUnitFile
 
 with (pathlib.Path(__file__).parent / 'xslt' / 'trx-to-junit.xslt').open('r', encoding='utf-8') as r:
     transform_trx_to_junit = etree.XSLT(etree.parse(r))
 
 
 def parse_trx_files(files: Iterable[str],
-                    progress: Callable[[Tuple[str, Union[JUnitTree, BaseException]]], Tuple[str, Union[JUnitTree, BaseException]]] = lambda x: x) -> Iterable[Tuple[str, Union[JUnitTree, BaseException]]]:
-    """Parses trx files and returns aggregated statistics as a ParsedUnitTestResults."""
-    def parse(path: str) -> Union[JUnitTree, BaseException]:
+                    progress: Callable[[ParsedJUnitFile], ParsedJUnitFile] = lambda x: x) -> Iterable[ParsedJUnitFile]:
+    def parse(path: str) -> JUnitTreeOrException:
+        """Parses a trx file and returns either a JUnitTree or an Exception."""
         if not os.path.exists(path):
             return FileNotFoundError(f'File does not exist.')
         if os.stat(path).st_size == 0: