Contents

#
# Fluent
#
# Copyright (C) 2011 FURUHASHI Sadayuki
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
module Fluent
module Test


class InputTestDriver < TestDriver
  def initialize(klass, &block)
    super(klass, &block)
    @emit_streams = []
    @expects = nil
  end

  def expect_emit(tag, time, record)
    (@expects ||= []) << [tag, time, record]
    self
  end

  def expected_emits
    @expects ||= []
  end

  attr_reader :emit_streams

  def emits
    all = []
    @emit_streams.each {|tag,events|
      events.each {|time,record|
        all << [tag, time, record]
      }
    }
    all
  end

  def events
    all = []
    @emit_streams.each {|tag,events|
      all.concat events
    }
    all
  end

  def records
    all = []
    @emit_streams.each {|tag,events|
      events.each {|time,record|
        all << record
      }
    }
    all
  end

  def run(&block)
    m = method(:emit_stream)
    super {
      Engine.define_singleton_method(:emit_stream) {|tag,es|
        m.call(tag, es)
      }

      block.call if block

      if @expects
        i = 0
        @emit_streams.each {|tag,events|
          events.each {|time,record|
            assert_equal(@expects[i], [tag, time, record])
            i += 1
          }
        }
        assert_equal @expects.length, i
      end
    }
    self
  end

  private
  def emit_stream(tag, es)
    @emit_streams << [tag, es.to_a]
  end
end


end
end

Version data entries

31 entries across 31 versions & 1 rubygems

Version	Path
fluentd-0.10.36	lib/fluent/test/input_test.rb
fluentd-0.10.35	lib/fluent/test/input_test.rb
fluentd-0.10.34	lib/fluent/test/input_test.rb
fluentd-0.10.33	lib/fluent/test/input_test.rb
fluentd-0.10.32	lib/fluent/test/input_test.rb
fluentd-0.10.31	lib/fluent/test/input_test.rb
fluentd-0.10.30	lib/fluent/test/input_test.rb
fluentd-0.10.29	lib/fluent/test/input_test.rb
fluentd-0.10.28	lib/fluent/test/input_test.rb
fluentd-0.10.27	lib/fluent/test/input_test.rb
fluentd-0.10.26	lib/fluent/test/input_test.rb
fluentd-0.10.25	lib/fluent/test/input_test.rb
fluentd-0.10.24	lib/fluent/test/input_test.rb
fluentd-0.10.23	lib/fluent/test/input_test.rb
fluentd-0.10.22	lib/fluent/test/input_test.rb
fluentd-0.10.21	lib/fluent/test/input_test.rb
fluentd-0.10.20	lib/fluent/test/input_test.rb
fluentd-0.10.19	lib/fluent/test/input_test.rb
fluentd-0.10.18	lib/fluent/test/input_test.rb
fluentd-0.10.17	lib/fluent/test/input_test.rb