fluentd 0.14.4-x86-mingw32

data/test/plugin/test_filter_grep.rb

@@ -0,0 +1,119 @@
+require_relative '../helper'
+require 'fluent/plugin/filter_grep'
+require 'fluent/test/driver/filter'
+
+class GrepFilterTest < Test::Unit::TestCase
+  include Fluent
+
+  setup do
+    Fluent::Test.setup
+    @time = event_time
+  end
+
+  def create_driver(conf = '')
+    Fluent::Test::Driver::Filter.new(Fluent::Plugin::GrepFilter).configure(conf)
+  end
+
+  sub_test_case 'configure' do
+    test 'check default' do
+      d = create_driver
+      assert_empty(d.instance.regexps)
+      assert_empty(d.instance.excludes)
+    end
+
+    test "regexpN can contain a space" do
+      d = create_driver(%[regexp1 message  foo])
+      assert_equal(Regexp.compile(/ foo/), d.instance.regexps['message'])
+    end
+
+    test "excludeN can contain a space" do
+      d = create_driver(%[exclude1 message  foo])
+      assert_equal(Regexp.compile(/ foo/), d.instance.excludes['message'])
+    end
+  end
+
+  sub_test_case 'filter_stream' do
+    def messages
+      [
+        "2013/01/13T07:02:11.124202 INFO GET /ping",
+        "2013/01/13T07:02:13.232645 WARN POST /auth",
+        "2013/01/13T07:02:21.542145 WARN GET /favicon.ico",
+        "2013/01/13T07:02:43.632145 WARN POST /login",
+      ]
+    end
+
+    def filter(config, msgs)
+      d = create_driver(config)
+      d.run {
+        msgs.each { |msg|
+          d.feed("filter.test", @time, {'foo' => 'bar', 'message' => msg})
+        }
+      }
+      d.filtered_records
+    end
+
+    test 'empty config' do
+      filtered_records = filter('', messages)
+      assert_equal(4, filtered_records.size)
+    end
+
+    test 'regexpN' do
+      filtered_records = filter('regexp1 message WARN', messages)
+      assert_equal(3, filtered_records.size)
+      assert_block('only WARN logs') do
+        filtered_records.all? { |r|
+          !r['message'].include?('INFO')
+        }
+      end
+    end
+
+    test 'excludeN' do
+      filtered_records = filter('exclude1 message favicon', messages)
+      assert_equal(3, filtered_records.size)
+      assert_block('remove favicon logs') do
+        filtered_records.all? { |r|
+          !r['message'].include?('favicon')
+        }
+      end
+    end
+
+    sub_test_case 'with invalid sequence' do
+      def messages
+        [
+          "\xff".force_encoding('UTF-8'),
+        ]
+      end
+
+      test "don't raise an exception" do
+        assert_nothing_raised { 
+          filter(%[regexp1 message WARN], ["\xff".force_encoding('UTF-8')])
+        }
+      end
+    end
+  end
+
+  sub_test_case 'grep non-string jsonable values' do
+    def filter(msg, config = 'regexp1 message 0')
+      d = create_driver(config)
+      d.run do
+        d.feed("filter.test", @time, {'foo' => 'bar', 'message' => msg})
+      end
+      d.filtered_records
+    end
+
+    data(
+      'array' => ["0"],
+      'hash' => ["0" => "0"],
+      'integer' => 0,
+      'float' => 0.1)
+    test "value" do |data|
+      filtered_records = filter(data)
+      assert_equal(1, filtered_records.size)
+    end
+
+    test "value boolean" do
+      filtered_records = filter(true, %[regexp1 message true])
+      assert_equal(1, filtered_records.size)
+    end
+  end
+end

data/test/plugin/test_filter_record_transformer.rb

@@ -0,0 +1,600 @@
+require_relative '../helper'
+require 'timecop'
+require 'fluent/test/driver/filter'
+require 'fluent/plugin/filter_record_transformer'
+
+class RecordTransformerFilterTest < Test::Unit::TestCase
+  include Fluent
+
+  setup do
+    Test.setup
+    @hostname = Socket.gethostname.chomp
+    @tag = 'test.tag'
+    @tag_parts = @tag.split('.')
+    @time = event_time('2010-05-04 03:02:01 UTC')
+    Timecop.freeze(@time)
+  end
+
+  teardown do
+    Timecop.return
+  end
+
+  def create_driver(conf = '')
+    Fluent::Test::Driver::Filter.new(Fluent::Plugin::RecordTransformerFilter).configure(conf)
+  end
+
+  sub_test_case 'configure' do
+    test 'check default' do
+      assert_nothing_raised do
+        create_driver
+      end
+    end
+
+    test "keep_keys must be specified together with renew_record true" do
+      assert_raise(Fluent::ConfigError) do
+        create_driver(%[keep_keys a])
+      end
+    end
+  end
+
+  sub_test_case "test options" do
+    def filter(config, msgs = [''])
+      d = create_driver(config)
+      d.run {
+        msgs.each { |msg|
+          d.feed(@tag, @time, {'foo' => 'bar', 'message' => msg})
+        }
+      }
+      d.filtered
+    end
+
+    CONFIG = %[
+      <record>
+        hostname ${hostname}
+        tag ${tag}
+        time ${time}
+        message ${hostname} ${tag_parts[-1]} ${message}
+      </record>
+    ]
+
+    test 'typical usage' do
+      msgs = ['1', '2']
+      filtered = filter(CONFIG, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal('bar', r['foo'])
+        assert_equal(@hostname, r['hostname'])
+        assert_equal(@tag, r['tag'])
+        assert_equal(Time.at(@time).localtime.to_s, r['time'])
+        assert_equal("#{@hostname} #{@tag_parts[-1]} #{msgs[i]}", r['message'])
+      end
+    end
+
+    test 'remove_keys' do
+      config = CONFIG + %[remove_keys foo,message]
+      filtered = filter(config)
+      filtered.each_with_index do |(_t, r), i|
+        assert_not_include(r, 'foo')
+        assert_equal(@hostname, r['hostname'])
+        assert_equal(@tag, r['tag'])
+        assert_equal(Time.at(@time).localtime.to_s, r['time'])
+        assert_not_include(r, 'message')
+      end
+    end
+
+    test 'renew_record' do
+      config = CONFIG + %[renew_record true]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_not_include(r, 'foo')
+        assert_equal(@hostname, r['hostname'])
+        assert_equal(@tag, r['tag'])
+        assert_equal(Time.at(@time).localtime.to_s, r['time'])
+        assert_equal("#{@hostname} #{@tag_parts[-1]} #{msgs[i]}", r['message'])
+      end
+    end
+
+    test 'renew_time_key' do
+      config = %[renew_time_key message]
+      times = [ Time.local(2,2,3,4,5,2010,nil,nil,nil,nil), Time.local(3,2,3,4,5,2010,nil,nil,nil,nil) ]
+      msgs = times.map{|t| t.to_f.to_s }
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(time, _record), i|
+        assert_equal(times[i].to_i, time)
+        assert(time.is_a?(Fluent::EventTime))
+      end
+    end
+
+    test 'keep_keys' do
+      config = %[renew_record true\nkeep_keys foo,message]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal('bar', r['foo'])
+        assert_equal(msgs[i], r['message'])
+      end
+    end
+
+    test 'enable_ruby' do
+      config = %[
+        enable_ruby yes
+        <record>
+          message ${hostname} ${tag_parts.last} ${"'" + message + "'"}
+        </record>
+      ]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal("#{@hostname} #{@tag_parts[-1]} '#{msgs[i]}'", r['message'])
+      end
+    end
+
+    test 'hash_value' do
+      config = %[
+        <record>
+          hash_field {"k1":100, "k2":"foobar"}
+        </record>
+      %]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal({"k1"=>100, "k2"=>"foobar"}, r['hash_field'])
+      end
+    end
+
+    test 'array_value' do
+      config = %[
+        <record>
+          array_field [1, 2, 3]
+        </record>
+      %]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal([1,2,3], r['array_field'])
+      end
+    end
+
+    test 'array_hash_mixed' do
+      config = %[
+        <record>
+          mixed_field {"hello":[1,2,3], "world":{"foo":"bar"}}
+        </record>
+      %]
+      msgs = ['1', '2']
+      filtered = filter(config, msgs)
+      filtered.each_with_index do |(_t, r), i|
+        assert_equal({"hello"=>[1,2,3], "world"=>{"foo"=>"bar"}}, r['mixed_field'])
+      end
+    end
+  end
+
+  sub_test_case 'test placeholders' do
+    def filter(config, msgs = [''])
+      d = create_driver(config)
+      yield d if block_given?
+      d.run {
+        records = msgs.map do |msg|
+          next msg if msg.is_a?(Hash)
+          { 'eventType0' => 'bar', 'message' => msg }
+        end
+        records.each do |record|
+          d.feed(@tag, @time, record)
+        end
+      }
+      d.filtered
+    end
+
+    %w[yes no].each do |enable_ruby|
+      test "hostname with enble_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            message ${hostname}
+          </record>
+        ]
+        filtered = filter(config)
+        filtered.each do |t, r|
+          assert_equal(@hostname, r['message'])
+        end
+      end
+
+      test "tag with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            message ${tag}
+          </record>
+        ]
+        filtered = filter(config)
+        filtered.each do |t, r|
+          assert_equal(@tag, r['message'])
+        end
+      end
+
+      test "tag_parts with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            message ${tag_parts[0]} ${tag_parts[-1]}
+          </record>
+        ]
+        expected = "#{@tag.split('.').first} #{@tag.split('.').last}"
+        filtered = filter(config)
+        filtered.each do |t, r|
+          assert_equal(expected, r['message'])
+        end
+      end
+
+      test "${tag_prefix[N]} and ${tag_suffix[N]} with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            message ${tag_prefix[1]} ${tag_prefix[-2]} ${tag_suffix[2]} ${tag_suffix[-3]}
+          </record>
+        ]
+        @tag = 'prefix.test.tag.suffix'
+        expected = "prefix.test prefix.test.tag tag.suffix test.tag.suffix"
+        filtered = filter(config)
+        filtered.each do |t, r|
+          assert_equal(expected, r['message'])
+        end
+      end
+
+      test "time with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            message ${time}
+          </record>
+        ]
+        filtered = filter(config)
+        filtered.each do |t, r|
+          assert_equal(Time.at(@time).localtime.to_s, r['message'])
+        end
+      end
+
+      test "record keys with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          remove_keys eventType0
+          <record>
+            message bar ${message}
+            eventtype ${eventType0}
+          </record>
+        ]
+        msgs = ['1', '2']
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          assert_not_include(r, 'eventType0')
+          assert_equal("bar", r['eventtype'])
+          assert_equal("bar #{msgs[i]}", r['message'])
+        end
+      end
+
+      test "Prevent overwriting reserved keys such as tag with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            new_tag ${tag}
+            new_record_tag ${record["tag"]}
+          </record>
+        ]
+        records = [{'tag' => 'tag', 'time' => 'time'}]
+        filtered = filter(config, records)
+        filtered.each_with_index do |(_t, r), i|
+          assert_not_equal('tag', r['new_tag'])
+          assert_equal(@tag, r['new_tag'])
+          assert_equal('tag', r['new_record_tag'])
+        end
+      end
+
+      test "hash values with placeholders with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            hash_field {
+              "hostname":"${hostname}",
+              "tag":"${tag}",
+              "${tag}":100
+            }
+          </record>
+        ]
+        msgs = ['1', '2']
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          assert_equal({"hostname" => @hostname, "tag" => @tag, "#{@tag}" => 100}, r['hash_field'])
+        end
+      end
+
+      test "array values with placeholders with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            array_field ["${hostname}", "${tag}"]
+          </record>
+        ]
+        msgs = ['1', '2']
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          assert_equal([@hostname, @tag], r['array_field'])
+        end
+      end
+
+      test "array and hash values with placeholders with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          <record>
+            mixed_field [{"tag":"${tag}"}]
+          </record>
+        ]
+        msgs = ['1', '2']
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          assert_equal([{"tag" => @tag}], r['mixed_field'])
+        end
+      end
+
+      test "keys with placeholders with enable_ruby #{enable_ruby}" do
+        config = %[
+          enable_ruby #{enable_ruby}
+          renew_record true
+          <record>
+            ${hostname} hostname
+            foo.${tag}  tag
+          </record>
+        ]
+        msgs = ['1', '2']
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          assert_equal({@hostname=>'hostname',"foo.#{@tag}"=>'tag'}, r)
+        end
+      end
+
+      test "disabled typecasting of values with enable_ruby #{enable_ruby}" do
+        config = %[
+          auto_typecast false
+          enable_ruby #{enable_ruby}
+          <record>
+            single      ${source}
+            multiple    ${source}${source}
+            with_prefix prefix-${source}
+            with_suffix ${source}-suffix
+            with_quote  source[""]
+          </record>
+        ]
+        msgs = [
+          { "source" => "string" },
+          { "source" => 123 },
+          { "source" => [1, 2] },
+          { "source" => {a:1, b:2} },
+          { "source" => nil },
+        ]
+        expected_results = [
+          { single: "string",
+            multiple: "stringstring",
+            with_prefix: "prefix-string",
+            with_suffix: "string-suffix",
+            with_quote: %Q{source[""]} },
+          { single: 123.to_s,
+            multiple: "#{123.to_s}#{123.to_s}",
+            with_prefix: "prefix-#{123.to_s}",
+            with_suffix: "#{123.to_s}-suffix",
+            with_quote: %Q{source[""]} },
+          { single: [1, 2].to_s,
+            multiple: "#{[1, 2].to_s}#{[1, 2].to_s}",
+            with_prefix: "prefix-#{[1, 2].to_s}",
+            with_suffix: "#{[1, 2].to_s}-suffix",
+            with_quote: %Q{source[""]} },
+          { single: {a:1, b:2}.to_s,
+            multiple: "#{{a:1, b:2}.to_s}#{{a:1, b:2}.to_s}",
+            with_prefix: "prefix-#{{a:1, b:2}.to_s}",
+            with_suffix: "#{{a:1, b:2}.to_s}-suffix",
+            with_quote: %Q{source[""]} },
+          { single: nil.to_s,
+            multiple: "#{nil.to_s}#{nil.to_s}",
+            with_prefix: "prefix-#{nil.to_s}",
+            with_suffix: "#{nil.to_s}-suffix",
+            with_quote: %Q{source[""]} },
+        ]
+        actual_results = []
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          actual_results << {
+            single: r["single"],
+            multiple: r["multiple"],
+            with_prefix: r["with_prefix"],
+            with_suffix: r["with_suffix"],
+            with_quote: r["with_quote"],
+          }
+        end
+        assert_equal(expected_results, actual_results)
+      end
+
+      test "enabled typecasting of values with enable_ruby #{enable_ruby}" do
+        config = %[
+          auto_typecast yes
+          enable_ruby #{enable_ruby}
+          <record>
+            single      ${source}
+            multiple    ${source}${source}
+            with_prefix prefix-${source}
+            with_suffix ${source}-suffix
+          </record>
+        ]
+        msgs = [
+          { "source" => "string" },
+          { "source" => 123 },
+          { "source" => [1, 2] },
+          { "source" => {a:1, b:2} },
+          { "source" => nil },
+        ]
+        expected_results = [
+          { single: "string",
+            multiple: "stringstring",
+            with_prefix: "prefix-string",
+            with_suffix: "string-suffix" },
+          { single: 123,
+            multiple: "#{123.to_s}#{123.to_s}",
+            with_prefix: "prefix-#{123.to_s}",
+            with_suffix: "#{123.to_s}-suffix" },
+          { single: [1, 2],
+            multiple: "#{[1, 2].to_s}#{[1, 2].to_s}",
+            with_prefix: "prefix-#{[1, 2].to_s}",
+            with_suffix: "#{[1, 2].to_s}-suffix" },
+          { single: {a:1, b:2},
+            multiple: "#{{a:1, b:2}.to_s}#{{a:1, b:2}.to_s}",
+            with_prefix: "prefix-#{{a:1, b:2}.to_s}",
+            with_suffix: "#{{a:1, b:2}.to_s}-suffix" },
+          { single: nil,
+            multiple: "#{nil.to_s}#{nil.to_s}",
+            with_prefix: "prefix-#{nil.to_s}",
+            with_suffix: "#{nil.to_s}-suffix" },
+        ]
+        actual_results = []
+        filtered = filter(config, msgs)
+        filtered.each_with_index do |(_t, r), i|
+          actual_results << {
+            single: r["single"],
+            multiple: r["multiple"],
+            with_prefix: r["with_prefix"],
+            with_suffix: r["with_suffix"],
+          }
+        end
+        assert_equal(expected_results, actual_results)
+      end
+
+      test %Q[record["key"] with enable_ruby #{enable_ruby}] do
+        config = %[
+          enable_ruby #{enable_ruby}
+          auto_typecast yes
+          <record>
+            _timestamp ${record["@timestamp"]}
+            _foo_bar   ${record["foo.bar"]}
+          </record>
+        ]
+        d = create_driver(config)
+        record = {
+          "foo.bar"    => "foo.bar",
+          "@timestamp" => 10,
+        }
+        d.run { d.feed(@tag, @time, record) }
+        filtered = d.filtered
+        filtered.each do |t, r|
+          assert { r['_timestamp'] == record['@timestamp'] }
+          assert { r['_foo_bar'] == record['foo.bar'] }
+        end
+      end
+    end
+
+    test 'unknown placeholder (enable_ruby no)' do
+      config = %[
+        enable_ruby no
+        <record>
+          message ${unknown}
+        </record>
+      ]
+      filter(config) { |d|
+        mock(d.instance.log).warn("unknown placeholder `${unknown}` found")
+      }
+    end
+
+    test 'failed to expand (enable_ruby yes)' do
+      config = %[
+        enable_ruby yes
+        <record>
+          message ${unknown['bar']}
+        </record>
+      ]
+      filtered = filter(config) { |d|
+        mock(d.instance.log).warn("failed to expand `%Q[\#{unknown['bar']}]`", anything)
+      }
+      filtered.each do |t, r|
+        assert_nil(r['message'])
+      end
+    end
+
+    test 'expand fields starting with @ (enable_ruby no)' do
+      config = %[
+        enable_ruby no
+        <record>
+          foo ${@timestamp}
+        </record>
+      ]
+      d = create_driver(config)
+      message = {"@timestamp" => "foo"}
+      d.run { d.feed(@tag, @time, message) }
+      filtered = d.filtered
+      filtered.each do |t, r|
+        assert_equal(message["@timestamp"], r['foo'])
+      end
+    end
+
+    test 'auto_typecast placeholder containing {} (enable_ruby yes)' do
+      config = %[
+        tag tag
+        enable_ruby yes
+        auto_typecast yes
+        <record>
+          foo ${record.map{|k,v|v}}
+        </record>
+      ]
+      d = create_driver(config)
+      message = {"@timestamp" => "foo"}
+      d.run { d.feed(@tag, @time, message) }
+      filtered = d.filtered
+      filtered.each do |t, r|
+        assert_equal([message["@timestamp"]], r['foo'])
+      end
+    end
+
+    test 'expand fields starting with @ (enable_ruby yes)' do
+      config = %[
+        enable_ruby yes
+        <record>
+          foo ${__send__("@timestamp")}
+        </record>
+      ]
+      d = create_driver(config)
+      message = {"@timestamp" => "foo"}
+      d.run { d.feed(@tag, @time, message) }
+      filtered = d.filtered
+      filtered.each do |t, r|
+        assert_equal(message["@timestamp"], r['foo'])
+      end
+    end
+  end # test placeholders
+
+  test "compatibility test (enable_ruby yes)" do
+    config = %[
+      enable_ruby yes
+      auto_typecast yes
+      <record>
+        _message   prefix-${message}-suffix
+        _time      ${Time.at(time)}
+        _number    ${number == '-' ? 0 : number}
+        _match     ${/0x[0-9a-f]+/.match(hex)[0]}
+        _timestamp ${__send__("@timestamp")}
+        _foo_bar   ${__send__('foo.bar')}
+      </record>
+    ]
+    d = create_driver(config)
+    record = {
+      "number"     => "-",
+      "hex"        => "0x10",
+      "foo.bar"    => "foo.bar",
+      "@timestamp" => 10,
+      "message"    => "10",
+    }
+    d.run { d.feed(@tag, @time, record) }
+    filtered = d.filtered
+    filtered.each do |t, r|
+      assert { r['_message'] == "prefix-#{record['message']}-suffix" }
+      assert { r['_time'] == Time.at(@time) }
+      assert { r['_number'] == 0 }
+      assert { r['_match'] == record['hex'] }
+      assert { r['_timestamp'] == record['@timestamp'] }
+      assert { r['_foo_bar'] == record['foo.bar'] }
+    end
+  end
+end