logstash/bin/logstash-test
Jordan Sissel e8f43c15d2 - add additional library checks
- add todo
2011-02-12 22:40:18 -08:00

90 lines
2.8 KiB
Ruby
Executable file

#!/usr/bin/env ruby
require "rubygems"
require "optparse"
$:.unshift "#{File.dirname(__FILE__)}/../lib"
$:.unshift "#{File.dirname(__FILE__)}/../test"
#require "logstash/test_syntax"
#require "logstash/filters/test_date"
#require "logstash/filters/test_multiline"
def check_lib(lib, provider, optional=true, message=nil)
begin
require lib
puts "+ Found #{optional ? "optional" : "required"} library '#{lib}'"
return { :optional => optional, :found => true }
rescue LoadError => e
puts "- Missing #{optional ? "optional" : "required"} library '#{lib}' - try 'gem install #{provider}'#{optional ? " if you want this library" : ""}. #{message}"
return { :optional => optional, :found => false }
end
end
def report_ruby_version
puts "Running #{RUBY_VERSION}p#{RUBY_PATCHLEVEL} on #{RUBY_PLATFORM}"
end
def check_libraries
results = []
results << check_lib("em-http-request", "em-http-request", true,
"needed for ElasticSearch input, output, and logstash-web support.")
results << check_lib("em-mongo", "em-mongo", true,
"needed for the mongodb output.")
results << check_lib("grok", "jls-grok", true,
"needed for the grok filter.")
results << check_lib("em-websocket", "em-websocket", true,
"needed for websocket output")
results << check_lib("rack", "rack", true,
"needed for logstash-web")
results << check_lib("thin", "thin", true,
"needed for logstash-web")
results << check_lib("amqp", "amqp", true,
"needed for AMQP input and output")
results << check_lib("sinatra/async", "async_sinatra", true,
"needed for logstash-web")
results << check_lib("uuidtools", "uuidtools", true,
"needed for AMQP input and output")
results << check_lib("ap", "awesome_print", true,
"improve logstash debug logging output")
results << check_lib("eventmachine", "eventmachine", false,
"required for logstash to function")
results << check_lib("json", "json", false,
"required for logstash to function")
missing_required = results.count { |r| !r[:optional] and !r[:found] }
if missing_required == 0
puts "All required libraries found :)"
else
suffix = (missing_required > 1) ? "ies" : "y"
puts "FATAL: Missing #{missing_required} required librar#{suffix}"
return false
end
return true
end
def run_tests
require "#{File.dirname(__FILE__)}/../test/run"
return Test::Unit::AutoRunner.run
end
def main(args)
report_ruby_version
# TODO(sissel): Add a way to call out specific things to test, like
# logstash-web, elasticsearch, mongodb, syslog, etc.
if !check_libraries
puts "Library check failed."
return 1
end
if !run_tests
puts "Test suite failed."
return 1
end
return 0
end
exit(main(ARGV))