test: add more fields to xml report for test aggregation tool (#519)

Author: ykim-akamai

.github/workflows/e2e-suite.yml

@@ -46,6 +46,20 @@ jobs:
         env:
           LINODE_CLI_TOKEN: ${{ secrets.LINODE_TOKEN }}
 
+      - name: Set release version env
+        run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
+
+      - name: Add additional information to XML report
+        run: |
+          echo $RELEASE_VERSION
+          echo ${{ env.RELEASE_VERSION }}
+          filename=$(ls | grep -E '^[0-9]{12}_cli_test_report\.xml$') 
+          python scripts/add_to_xml_test_report.py \
+          --branch_name "${{ env.RELEASE_VERSION }}" \
+          --gha_run_id "$GITHUB_RUN_ID" \
+          --gha_run_number "$GITHUB_RUN_NUMBER" \
+          --xmlfile "${filename}"
+
       - name: Upload test results
         run: |
           filename=$(ls | grep -E '^[0-9]{12}_cli_test_report\.xml$')

scripts/add_to_xml_test_report.py

@@ -0,0 +1,37 @@
+import argparse
+import xml.etree.ElementTree as ET
+
+# Parse command-line arguments
+parser = argparse.ArgumentParser(description='Modify XML with workflow information')
+parser.add_argument('--branch_name', required=True)
+parser.add_argument('--gha_run_id', required=True)
+parser.add_argument('--gha_run_number', required=True)
+parser.add_argument('--xmlfile', required=True)  # Added argument for XML file path
+
+args = parser.parse_args()
+
+# Open and parse the XML file
+xml_file_path = args.xmlfile
+tree = ET.parse(xml_file_path)
+root = tree.getroot()
+
+# Create new elements for the information
+branch_name_element = ET.Element('branch_name')
+branch_name_element.text = args.branch_name
+
+gha_run_id_element = ET.Element('gha_run_id')
+gha_run_id_element.text = args.gha_run_id
+
+gha_run_number_element = ET.Element('gha_run_number')
+gha_run_number_element.text = args.gha_run_number
+
+# Add the new elements to the root of the XML
+root.append(branch_name_element)
+root.append(gha_run_id_element)
+root.append(gha_run_number_element)
+
+# Save the modified XML
+modified_xml_file_path = xml_file_path  # Overwrite it
+tree.write(modified_xml_file_path)
+
+print(f'Modified XML saved to {modified_xml_file_path}')