Contents

require 'helper'
require 'json'

class SendmailInputTest < Test::Unit::TestCase
  def setup
    Fluent::Test.setup
  end

  TMP_DIR = File.dirname(__FILE__) + "/../tmp"
  DATA_DIR = File.dirname(__FILE__) + "/../data"

  CONFIG_UNBUNDLE = %[
    path #{TMP_DIR}/sendmaillog
    tag sendmail
    unbundle yes
  ]

  CONFIG_BUNDLE = %[
    path #{TMP_DIR}/sendmaillog
    tag sendmail
    unbundle no
  ]

  def setup
    Fluent::Test.setup
    FileUtils.rm_rf(TMP_DIR)
    FileUtils.mkdir_p(TMP_DIR)
  end

  def create_driver(conf = CONFIG_UNBUNDLE, tag='test')
    driver = Fluent::Test::InputTestDriver.new(Fluent::SendmailInput)
    driver.configure(conf)
    driver
  end

  def test_configure
    #### set configurations
    # d = create_driver %[
    #   path test_path
    #   compress gz
    # ]
    #### check configurations
    # assert_equal 'test_path', d.instance.path
    # assert_equal :gz, d.instance.compress
  end

  def test_unbundled
    data_file = "#{DATA_DIR}/data1"
    expect_file = "#{DATA_DIR}/data1_unbundle_result_expect"
    driver = create_driver
    do_test(driver, data_file, expect_file)
  end

  def test_bundled
    data_file = "#{DATA_DIR}/data1"
    expect_file = "#{DATA_DIR}/data1_bundle_result_expect"
    driver = create_driver(conf=CONFIG_BUNDLE)
    do_test(driver, data_file, expect_file)
  end

  def do_test(driver, data_file, expect_file)
    lines = nil
    path = driver.instance.paths[0].to_s

    # touch tail file
    File.open(path, "a") {}

    # result_expect
    expects = []
    File.open(expect_file, "r") {|expectfile|
      expectfile.each_line {|line|
        expects.push(JSON.parse(line))
      }
    }

    driver.run do
      sleep 1
      File.open(data_file, "r") {|srcfile|
        File.open(path, "a") {|dstfile|
          srcfile.each_line {|line|
            dstfile.puts(line)
          }
        }
      }
      sleep 1
    end

    emits = driver.emits
    emits.each_index {|i|
      assert_equal(expects[i], emits[i][2])
    }
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
fluent-plugin-sendmail-0.1.0	test/plugin/test_in_sendmail.rb