mirror of
https://github.com/elastic/logstash.git
synced 2025-04-24 14:47:19 -04:00
120 lines
3.1 KiB
Ruby
120 lines
3.1 KiB
Ruby
require "insist"
|
|
require "logstash/agent"
|
|
require "logstash/pipeline"
|
|
require "logstash/event"
|
|
require "logstash/logging"
|
|
require "insist"
|
|
require "stud/try"
|
|
|
|
$TESTING = true
|
|
if RUBY_VERSION < "1.9.2"
|
|
$stderr.puts "Ruby 1.9.2 or later is required. (You are running: " + RUBY_VERSION + ")"
|
|
$stderr.puts "Options for fixing this: "
|
|
$stderr.puts " * If doing 'ruby bin/logstash ...' add --1.9 flag to 'ruby'"
|
|
$stderr.puts " * If doing 'java -jar ... ' add -Djruby.compat.version=RUBY1_9 to java flags"
|
|
raise LoadError
|
|
end
|
|
|
|
$logger = LogStash::Logger.new(STDOUT)
|
|
if ENV["TEST_DEBUG"]
|
|
$logger.level = :debug
|
|
else
|
|
$logger.level = :error
|
|
end
|
|
|
|
module LogStash
|
|
module RSpec
|
|
def config(configstr)
|
|
let(:config) { configstr }
|
|
end # def config
|
|
|
|
def type(default_type)
|
|
let(:default_type) { default_type }
|
|
end
|
|
|
|
def tags(*tags)
|
|
let(:default_tags) { tags }
|
|
puts "Setting default tags: #{@default_tags}"
|
|
end
|
|
|
|
def sample(sample_event, &block)
|
|
name = sample_event.is_a?(String) ? sample_event : sample_event.to_json
|
|
name = name[0..50] + "..." if name.length > 50
|
|
|
|
describe "\"#{name}\"" do
|
|
let(:pipeline) { LogStash::Pipeline.new(config) }
|
|
let(:event) do
|
|
sample_event = [sample_event] unless sample_event.is_a?(Array)
|
|
next sample_event.collect do |e|
|
|
e = { "message" => e } if e.is_a?(String)
|
|
next LogStash::Event.new(e)
|
|
end
|
|
end
|
|
|
|
let(:results) do
|
|
results = []
|
|
count = 0
|
|
pipeline.instance_eval { @filters.each(&:register) }
|
|
event.each do |e|
|
|
extra = []
|
|
pipeline.filter(e) do |new_event|
|
|
extra << new_event
|
|
end
|
|
results << e if !e.cancelled?
|
|
results += extra.reject(&:cancelled?)
|
|
end
|
|
|
|
# TODO(sissel): pipeline flush needs to be implemented.
|
|
#results += pipeline.flush
|
|
next results
|
|
end
|
|
|
|
subject { results.length > 1 ? results: results.first }
|
|
|
|
it("when processed", &block)
|
|
end
|
|
end # def sample
|
|
|
|
def input(&block)
|
|
it "inputs" do
|
|
queue = Queue.new
|
|
pipeline = LogStash::Pipeline.new(config)
|
|
#(class << pipeline; self; end).send(:define_method, :output) do |event|
|
|
#p :event => event
|
|
#queue << event
|
|
#end
|
|
#p pipeline.method(:output)
|
|
block.call(pipeline, queue)
|
|
pipeline.shutdown
|
|
end
|
|
end # def input
|
|
|
|
def agent(&block)
|
|
@agent_count ||= 0
|
|
require "logstash/pipeline"
|
|
|
|
# scoping is hard, let's go shopping!
|
|
describe "agent(#{@agent_count}) #{caller[1]}" do
|
|
before :each do
|
|
start = ::Time.now
|
|
pipeline = LogStash::Pipeline.new(config)
|
|
pipeline.run
|
|
@duration = ::Time.now - start
|
|
end
|
|
it("looks good", &block)
|
|
end
|
|
@agent_count += 1
|
|
end # def agent
|
|
|
|
end # module RSpec
|
|
end # module LogStash
|
|
|
|
class Shiftback
|
|
def initialize(&block)
|
|
@block = block
|
|
end
|
|
|
|
def <<(event)
|
|
@block.call(event)
|
|
end
|
|
end # class Shiftback
|