YAML Tests: Download branch when in a release branch

.buildkite/pipeline.yml

@@ -67,6 +67,7 @@ steps:
     env:
       RUBY_VERSION: "3.3"
       STACK_VERSION: 8.15.0-SNAPSHOT
+      ES_YAML_TESTS_BRANCH: 8.15
       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 @@
   CLIENT = 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

rake_tasks/automation.rake

@@ -103,6 +103,12 @@ namespace :automation do
     regexp = Regexp.new(/([0-9]{1,2}\.[0-9]{1,2}\.[0-9]{1,2}?+(-SNAPSHOT)?)/)
     files.each do |file|
       content = File.read(file)
+      if file == '.buildkite/pipeline.yml'
+        branch = version.match(/([0-9]+\.[0-9]+)\.[0-9]+.*/)[1]
+        old_branch = content.match(/ES_YAML_TESTS_BRANCH: ([0-9.]+)/)[1]
+        content.gsub!(old_branch, branch)
+        puts "[#{old_branch}] -> [#{branch}] in #{file.gsub('./', '')}"
+      end
       match = content.match(regexp)
       next if match.nil?