Contents

# encoding: UTF-8
require_relative 'spec_helper'

shared_context 'keep_forward_init' do
  before { Fluent::Test.setup }
  CONFIG = %[
    <server>
      host localhost
      port 24224
    </server>
    <server>
      host localhost
      port 24225
    </server>
  ]
  let(:tag) { 'syslog.host1' }
  let(:es)  { Object.new }
  let(:driver) { Fluent::Test::OutputTestDriver.new(Fluent::KeepForwardOutput, tag).configure(CONFIG).instance }
end

shared_context 'keep_forward_try_once' do
  before do
    # simpler version of Fluent::ForwardOutput#start method
    driver.instance_variable_set(:@rand_seed, Random.new.seed)
    driver.send(:rebuild_weight_array)
    driver.instance_variable_set(:@rr, 0)
    # try send once to cache keep_node
    driver.stub(:send_data) # stub
    driver.write_objects(tag, es)
  end
  let!(:keep_node) { driver.instance_variable_get(:@node)[tag] }
  let!(:unkeep_node) { (driver.instance_variable_get(:@nodes) - [keep_node]).first }
end

describe Fluent::KeepForwardOutput do
  include_context 'keep_forward_init'
  include_context 'keep_forward_try_once'

  describe 'keep forwarding if no problem?' do
    before { driver.should_receive(:send_data).with(keep_node, tag, es) }
    it { driver.write_objects(tag, es) }
  end

  describe 'switch if not available?' do
    before { keep_node.available = false }

    before { driver.should_receive(:send_data).with(unkeep_node, tag, es) }
    it { driver.write_objects(tag, es) }
  end

  describe 'switch if not included in weight_array?' do
    before { driver.instance_variable_set(:@weight_array, [unkeep_node]) }

    before { driver.should_receive(:send_data).with(unkeep_node, tag, es) }
    it { driver.write_objects(tag, es) }
  end

  describe 'switch if send_data to keep_node raises?' do
    before { driver.stub(:send_data).with(keep_node, tag, es).and_raise(StandardError) }

    before { driver.should_receive(:send_data).with(unkeep_node, tag, es) }
    it { driver.write_objects(tag, es) }
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
fluent-plugin-keep-forward-0.0.1	spec/out_keep_forward_spec.rb