YAML Tests: Download branch when in a release branch

Author: picandocodigo

.buildkite/pipeline.yml

@@ -68,6 +68,7 @@ steps:
     env:
       RUBY_VERSION: "3.3"
       STACK_VERSION: 8.16.0-SNAPSHOT
+      ES_YAML_TESTS_BRANCH: main
       TRANSPORT_VERSION: "8.3"
       RUBY_SOURCE: "ruby"
       TEST_SUITE: "platinum"

.buildkite/run-yaml-tests.sh

@@ -36,6 +36,7 @@ docker run \
        --env "BUILDKITE=true" \
        --env "TRANSPORT_VERSION=${TRANSPORT_VERSION}" \
        --env "STACK_VERSION=${STACK_VERSION}" \
+       --env "ES_YAML_TESTS_BRANCH=${ES_YAML_TESTS_BRANCH}" \
        --volume $repo:/usr/src/app \
        --name elasticsearch-ruby \
        --rm \

elasticsearch-api/spec/yaml-test-runner/run.rb

@@ -55,11 +55,13 @@
            Elasticsearch::Client.new(host: host, transport_options: transport_options)
          end
 
-
 tests_path = File.expand_path('./tmp', __dir__)
 
 logger = Logger.new($stdout)
 logger.level = Logger::WARN unless ENV['DEBUG']
 
-Elasticsearch::Tests::Downloader::run(tests_path)
+# If we're running in a release branch, download the corresponding branch for tests
+current_branch = `git rev-parse --abbrev-ref HEAD`.strip
+branch = current_branch.match(/[0-9]\.[0-9]+/)&.[](0) || ENV['ES_YAML_TESTS_BRANCH'] || nil
+Elasticsearch::Tests::Downloader::run(tests_path, branch)
 Elasticsearch::Tests::TestRunner.new(CLIENT, tests_path, logger).run