require "bundler/gem_tasks"
def __current__
Pathname( File.expand_path('..', __FILE__) )
end
def git_specs(command, options={})
sh "git --git-dir=#{__current__.join('../support/elasticsearch/.git')} --work-tree=#{__current__.join('../support/elasticsearch')} #{command}", options
end
task(:default) { system "rake --tasks" }
task :test => 'test:unit'
# ----- Test tasks ------------------------------------------------------------
require 'rake/testtask'
namespace :test do
desc "Update the repository with YAML tests"
task :update do
git_specs "fetch origin --verbose", :verbose => true
end
task :ci_reporter do
ENV['CI_REPORTS'] ||= 'tmp/reports'
if defined?(RUBY_VERSION) && RUBY_VERSION < '1.9'
require 'ci/reporter/rake/test_unit'
Rake::Task['ci:setup:testunit'].invoke
else
require 'ci/reporter/rake/minitest'
Rake::Task['ci:setup:minitest'].invoke
end
end
Rake::TestTask.new(:unit) do |test|
Rake::Task['test:ci_reporter'].invoke if ENV['CI']
test.libs << 'lib' << 'test'
test.test_files = FileList["test/unit/**/*_test.rb"]
# test.verbose = true
# test.warning = true
end
desc "Run integration tests"
task :integration do
require 'elasticsearch'
branches = `git --git-dir=#{__current__.join('../support/elasticsearch/.git')} --work-tree=#{__current__.join('../support/elasticsearch')} branch --no-color`
current_branch = branches.
split("\n").
select { |b| b =~ /^\*/ }.
reject { |b| b =~ /no branch|detached/ }.
map { |b| b.gsub(/^\*\s*/, '') }.
first
unless current_branch
STDERR.puts "[!] Unable to determine current branch, defaulting to 'master'"
current_branch = 'master'
end
# Define the task
t = Rake::TestTask.new(:integration) do |test|
Rake::Task['test:ci_reporter'].invoke if ENV['CI']
test.libs << 'lib' << 'test'
test.test_files = FileList["test/integration/yaml_test_runner.rb", "test/integration/**/*_test.rb"]
end
# Check if a test cluster is running
begin
client = Elasticsearch::Client.new :host => "localhost:#{ENV['TEST_CLUSTER_PORT'] || 9250}"
es_version_info = client.info['version']
build_hash = es_version_info['build_hash']
cluster_running = true
rescue Faraday::Error::ConnectionFailed => e
STDERR.puts "[!] Test cluster not running?"
cluster_running = false
end
checkout_specs_version = ENV['TEST_NO_CHECKOUT'].nil? ? true : false
checkout_build_hash = ENV['TEST_BUILD_REF'] || build_hash
ENV['TEST_BUILD_REF'] = checkout_build_hash
begin
unless checkout_specs_version
STDERR.puts '-'*80, "YAML tests: Not switching, TEST_NO_CHECKOUT=y", '-'*80
end
if checkout_specs_version && !checkout_build_hash
STDERR.puts "[!] Cannot determine checkout build hash -- server not running or TEST_BUILD_REF not specified"
exit(1)
end
if checkout_specs_version && checkout_build_hash
# Checkout the commit corresponding to the running server build, or passed TEST_BUILD_REF
name = ENV['CI'] ? checkout_build_hash : "[\e[1m#{checkout_build_hash}\e[0m]"
STDERR.puts '-'*80, "YAML tests: Switching to #{name} from #{current_branch}", '-'*80
git_specs "checkout #{checkout_build_hash} --force --quiet"
end
# Path to the API specs
ENV['TEST_REST_API_SPEC'] = __current__.join('../support/elasticsearch/rest-api-spec/test/').to_s
# Run the task
args = [t.ruby_opts_string, t.run_code, t.file_list_string, t.option_list].join(' ')
ruby args do |ok, status|
if !ok && status.respond_to?(:signaled?) && status.signaled?
raise SignalException.new(status.termsig)
elsif !ok
fail "Command failed with status (#{status.exitstatus}): " + "[ruby #{args}]"
end
end
ensure
git_specs "checkout #{current_branch} --force --quiet" if checkout_specs_version && current_branch
end
end
desc "Run unit and integration tests"
task :all do
Rake::Task['test:ci_reporter'].invoke if ENV['CI']
Rake::Task['test:unit'].invoke
Rake::Task['test:integration'].invoke
end
namespace :cluster do
desc "Start Elasticsearch nodes for tests"
task :start do
$LOAD_PATH << File.expand_path('../../elasticsearch-transport/lib', __FILE__) << File.expand_path('../test', __FILE__)
require 'elasticsearch/extensions/test/cluster'
Elasticsearch::Extensions::Test::Cluster.start
end
desc "Stop Elasticsearch nodes for tests"
task :stop do
$LOAD_PATH << File.expand_path('../../elasticsearch-transport/lib', __FILE__) << File.expand_path('../test', __FILE__)
require 'elasticsearch/extensions/test/cluster'
Elasticsearch::Extensions::Test::Cluster.stop
end
end
end
# ----- Documentation tasks ---------------------------------------------------
require 'yard'
YARD::Rake::YardocTask.new(:doc) do |t|
t.options = %w| --embed-mixins --markup=markdown |
end
# ----- Code analysis tasks ---------------------------------------------------
if defined?(RUBY_VERSION) && RUBY_VERSION > '1.9'
require 'cane/rake_task'
Cane::RakeTask.new(:quality) do |cane|
cane.abc_max = 15
cane.no_style = true
end
end