Merge pull request #101 from simpsonjulian/huge-junits-2

Don't send annotations if they're too large for the API

README.md

@@ -53,7 +53,7 @@ the build to fail.
 
 ## Developing
 
-To test the junit parser (in Ruby) and plugin hooks (in Bash):
+To test the plugin hooks (in Bash) and the junit parser (in Ruby):
 
 ```bash
 docker-compose run --rm plugin &&

hooks/command

@@ -3,6 +3,7 @@
 set -euo pipefail
 
 PLUGIN_DIR="$(cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd)/.."
+MAX_SIZE=100 # in KB
 
 echo "--- :junit: Download the junits"
 
@@ -15,6 +16,12 @@ function cleanup {
   rm -rf "${annotation_dir}"
 }
 
+function check_size {
+  local size_in_kb
+  size_in_kb=$(du -k "${annotation_path}" | cut -f 1)
+  [ "${size_in_kb}" -lt "${MAX_SIZE}" ]
+}
+
 trap cleanup EXIT
 
 buildkite-agent artifact download \
@@ -37,6 +44,13 @@ docker \
 cat "$annotation_path"
 
 if grep -q "<details>" "$annotation_path"; then
+
+  if ! check_size; then
+      echo "--- :warning: Failures too large to annotate"
+      echo "The failures are too large to create a build annotation. Please inspect the failed JUnit artifacts manually."
+      exit 0
+  fi
+
   echo "--- :buildkite: Creating annotation"
   # shellcheck disable=SC2002
   cat "$annotation_path" | buildkite-agent annotate --context junit --style error

tests/command.bats

@@ -6,6 +6,7 @@ load "$BATS_PATH/load.bash"
 # export MKTEMP_STUB_DEBUG=/dev/tty
 # export BUILDKITE_AGENT_STUB_DEBUG=/dev/tty
 # export DOCKER_STUB_DEBUG=/dev/tty
+# export DU_STUB_DEBUG=/dev/tty
 
 @test "runs the annotator and creates the annotation" {
   export BUILDKITE_PLUGIN_JUNIT_ANNOTATE_ARTIFACTS="junits/*.xml"
@@ -118,3 +119,32 @@ load "$BATS_PATH/load.bash"
 
   assert_output --partial "BUILDKITE_PLUGIN_JUNIT_ANNOTATE_ARTIFACTS: unbound variable"
 }
+
+@test "fails if the annotation is larger than " {
+  export BUILDKITE_PLUGIN_JUNIT_ANNOTATE_ARTIFACTS="junits/*.xml"
+
+  artifacts_tmp="tests/tmp/$PWD/junit-artifacts"
+  annotation_tmp="tests/tmp/$PWD/junit-annotation"
+
+  stub mktemp \
+    "-d junit-annotate-plugin-artifacts-tmp.XXXXXXXXXX : mkdir -p $artifacts_tmp; echo $artifacts_tmp" \
+    "-d junit-annotate-plugin-annotation-tmp.XXXXXXXXXX : mkdir -p $annotation_tmp; echo $annotation_tmp"
+
+  # 1kb over the 100k size limit of annotations
+  stub du "-k /plugin/tests/tmp//plugin/junit-annotation/annotation.md : echo 101 /plugin/tests/tmp//plugin/junit-annotation/annotation.md"
+
+  stub buildkite-agent "artifact download junits/*.xml /plugin/tests/tmp//plugin/junit-artifacts : echo Downloaded artifacts"
+
+  stub docker "--log-level error run --rm --volume /plugin/tests/tmp//plugin/junit-artifacts:/junits --volume /plugin/hooks/../ruby:/src --env BUILDKITE_PLUGIN_JUNIT_ANNOTATE_JOB_UUID_FILE_PATTERN= --env BUILDKITE_PLUGIN_JUNIT_ANNOTATE_FAILURE_FORMAT= ruby:2.5-alpine /src/bin/annotate /junits : echo '<details>Failure</details>'"
+
+  run "$PWD/hooks/command"
+
+  assert_success
+
+  assert_output --partial "Failures too large to annotate"
+
+  unstub docker
+  unstub buildkite-agent
+  unstub du
+  unstub mktemp
+}