Replaces yaml test runner with gem

Author: picandocodigo

Rakefile

@@ -45,42 +45,9 @@ namespace :spec do
     t.pattern = Dir.glob('spec/**/*_spec.rb')
   end
 
-  desc 'Download test suite to ./spec/tmp'
-  task :download_tests do
-    require 'open-uri'
-    path = 'spec/tmp'
-    filename = 'tests.zip'
-    url = 'https://api.github.com/repos/elastic/elasticsearch-clients-tests/zipball/main'
-
-    File.open(filename, 'w') do |downloaded_file|
-      URI.open(
-        url,
-        'Accept' => 'application/vnd.github+json'
-      ) do |artifact_file|
-        downloaded_file.write(artifact_file.read)
-      end
-    end
-    if File.exist?(filename)
-      puts "Successfully downloaded #{filename}"
-    else
-      warn "[!] Couldn't download #{filename}"
-      exit 1
-    end
-
-    puts "Unzipping files"
-    `unzip #{filename} -d spec/tmp/`
-    puts "Removing zip file"
-    File.delete(filename)
-  end
-
-  desc 'Clean tests folder'
-  task :clean_tests do
-    FileUtils.rm_rf(Dir.glob('./spec/tmp/**/*'), secure: true)
-  end
-
   desc 'Run YAML test runner'
   task :yaml do
-    ruby './spec/yaml-test-runner/runner.rb'
+    ruby './spec/yaml-test-runner/run.rb'
   end
 end
 
@@ -112,7 +79,7 @@ rescue StandardError => e
   abort "[!!!] #{e.class} : #{e.message}"
 end
 
-desc "Server info"
+desc 'Server info'
 task :info do
   require 'elasticsearch-serverless'
   client = ElasticsearchServerless::Client.new(url: ENV['ELASTICSEARCH_URL'], api_key: ENV['API_KEY'])

spec/yaml-test-runner/code_runner.rb

@@ -15,9 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 
-module Elastic
-  module TestRunner
-    class TestFailure < StandardError; end
-    class TestError < StandardError; end
-  end
-end
+require 'logger'
+require 'elasticsearch/tests/test_runner'
+require 'elasticsearch/tests/downloader'
+require_relative '../spec_helper'
+
+tests_path = File.expand_path('../../tmp', __dir__)
+
+logger = Logger.new($stdout)
+logger.level = Logger::WARN unless ENV['DEBUG']
+
+Elasticsearch::Tests::Downloader::run(tests_path)
+Elasticsearch::Tests::TestRunner.new(CLIENT, tests_path, logger).run