Merge pull request #1 from Phoenix7351/CA-4183

CA-4183: Generate failed tests artifacts file

Author: Phoenix7351

hooks/command

@@ -13,6 +13,7 @@ RUBY_IMAGE="${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_RUBY_IMAGE:-ruby:3.1-alpine@sha256
 
 artifacts_dir="$(pwd)/$(mktemp -d "junit-annotate-plugin-artifacts-tmp.XXXXXXXXXX")"
 annotation_dir="$(pwd)/$(mktemp -d "junit-annotate-plugin-annotation-tmp.XXXXXXXXXX")"
+failures_file_path="./failures_artifacts_names.txt"
 annotation_path="${annotation_dir}/annotation.md"
 annotation_style="info"
 fail_build=0
@@ -39,6 +40,25 @@ if ! buildkite-agent artifact download "${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_ARTIFA
   exit "${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_FAILED_DOWNLOAD_EXIT_CODE:-2}"
 fi
 
+echo "--- :junit: Building the failures file"
+
+if [[ "${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_RUN_IN_DOCKER:-true}" =~ "true" ]]; then
+  docker \
+    --log-level "error" \
+    run \
+      --rm \
+      --volume "$artifacts_dir:/junits" \
+      --volume "$PLUGIN_DIR/ruby:/src" \
+      --env "BUILDKITE_PLUGIN_JUNIT_ANNOTATE_JOB_UUID_FILE_PATTERN=${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_JOB_UUID_FILE_PATTERN:-}" \
+      --env "BUILDKITE_PLUGIN_JUNIT_ANNOTATE_FAILURE_FORMAT=${BUILDKITE_PLUGIN_JUNIT_ANNOTATE_FAILURE_FORMAT:-}" \
+      "${RUBY_IMAGE}" ruby /src/bin/buildfailuresfile /junits \
+        > "$failures_file_path"
+else
+  ruby "${PLUGIN_DIR}/ruby/bin/buildfailuresfile" "${artifacts_dir}" > "$failures_file_path"
+fi
+
+buildkite-agent artifact upload "${failures_file_path}"
+
 echo "--- :junit: Processing the junits"
 
 set +e

ruby/bin/buildfailuresfile

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+
+require 'rexml/document'
+require 'rexml/element'
+require 'set'
+
+junits_dir = ARGV[0]
+abort("Usage: annotate <junits-dir>") unless junits_dir
+abort("#{junits_dir} does not exist") unless Dir.exist?(junits_dir)
+
+job_pattern = ENV['BUILDKITE_PLUGIN_JUNIT_ANNOTATE_JOB_UUID_FILE_PATTERN']
+job_pattern = '-(.*).xml' if !job_pattern || job_pattern.empty?
+
+failure_format = ENV['BUILDKITE_PLUGIN_JUNIT_ANNOTATE_FAILURE_FORMAT']
+failure_format = 'classname' if !failure_format || failure_format.empty?
+
+class Failure < Struct.new(:unit_name)
+end
+
+junit_report_files = Dir.glob(File.join(junits_dir, "**", "*"), File::FNM_DOTMATCH)
+testcases = 0
+tests = Set.new
+
+junit_report_files.sort.each do |file|
+  next if File.directory?(file)
+
+  job = File.basename(file)[/#{job_pattern}/, 1]
+  xml = File.read(file)
+  doc = REXML::Document.new(xml)
+
+  REXML::XPath.each(doc, '//testsuite/testcase') do |testcase|
+    testcases += 1
+    unit_name = testcase.attributes[failure_format].to_s
+    testcase.elements.each("failure | error | skipped") do |elem|
+      tests.add(unit_name)
+    end
+  end
+end
+
+tests.each do |elem|
+  puts "*#{elem}*"
+end