ci: write test shard name for testmo report

ci: write test shard name for testmo report

Pull Request resolved: #249

1 files changed, 22 insertions, 0 deletions

diff --git a/.github/scripts/tests/junit_utils.py b/.github/scripts/tests/junit_utils.py
new file mode 100644
index 0000000000..c1aea12f80
--- /dev/null
+++ b/.github/scripts/tests/junit_utils.py
@@ -0,0 +1,22 @@
+from xml.etree import ElementTree as ET
+
+
+def get_or_create_properties(testcase):
+    props = testcase.find("properties")
+    if props is None:
+        props = ET.Element("properties")
+        testcase.append(props)
+    return props
+
+
+def add_junit_link_property(testcase, name, url):
+    add_junit_property(testcase, f"url:{name}", url)
+
+
+def add_junit_property(testcase, name, value):
+    props = get_or_create_properties(testcase)
+    props.append(ET.Element("property", dict(name=name, value=value)))
+
+
+def add_junit_log_property(testcase, url):
+    add_junit_link_property(testcase, "Log", url)