fluentd 1.14.6 → 1.15.0

data/test/plugin/test_in_tail.rb

@@ -12,15 +12,21 @@ require 'securerandom'
 class TailInputTest < Test::Unit::TestCase
   include FlexMock::TestCase
 
+  def tmp_dir
+    File.join(File.dirname(__FILE__), "..", "tmp", "tail#{ENV['TEST_ENV_NUMBER']}", SecureRandom.hex(10))
+  end
+
   def setup
     Fluent::Test.setup
-    cleanup_directory(TMP_DIR)
+    @tmp_dir = tmp_dir
+    cleanup_directory(@tmp_dir)
   end
 
   def teardown
     super
-    cleanup_directory(TMP_DIR)
+    cleanup_directory(@tmp_dir)
     Fluent::Engine.stop
+    Timecop.return
   end
 
   def cleanup_directory(path)
@@ -29,89 +35,48 @@ class TailInputTest < Test::Unit::TestCase
       return
     end
 
-    if Fluent.windows?
-      Dir.glob("*", base: path).each do |name|
-        begin
-          cleanup_file(File.join(path, name))
-        rescue
-          # expect test driver block release already owned file handle.
-        end
-      end
-    else
-      begin
-        FileUtils.rm_f(path, secure:true)
-      rescue ArgumentError
-        FileUtils.rm_f(path) # For Ruby 2.6 or before.
-      end
-      if File.exist?(path)
-        FileUtils.remove_entry_secure(path, true)
-      end
-    end
-    FileUtils.mkdir_p(path)
+    FileUtils.remove_entry_secure(path, true)
   end
 
   def cleanup_file(path)
-    if Fluent.windows?
-      # On Windows, when the file or directory is removed and created
-      # frequently, there is a case that creating file or directory will
-      # fail. This situation is caused by pending file or directory
-      # deletion which is mentioned on win32 API document [1]
-      # As a workaround, execute rename and remove method.
-      #
-      # [1] https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilea#files
-      #
-      file = File.join(Dir.tmpdir, SecureRandom.hex(10))
-      begin
-        FileUtils.mv(path, file)
-        FileUtils.rm_rf(file, secure: true)
-      rescue ArgumentError
-        FileUtils.rm_rf(file) # For Ruby 2.6 or before.
-      end
-      if File.exist?(file)
-        # ensure files are closed for Windows, on which deleted files
-        # are still visible from filesystem
-        GC.start(full_mark: true, immediate_mark: true, immediate_sweep: true)
-        FileUtils.remove_entry_secure(file, true)
-      end
-    else
-      begin
-        FileUtils.rm_f(path, secure: true)
-      rescue ArgumentError
-        FileUtils.rm_f(path) # For Ruby 2.6 or before.
-      end
-      if File.exist?(path)
-        FileUtils.remove_entry_secure(path, true)
-      end
-    end
+    FileUtils.remove_entry_secure(path, true)
   end
 
   def create_target_info(path)
     Fluent::Plugin::TailInput::TargetInfo.new(path, Fluent::FileWrapper.stat(path).ino)
   end
 
-  TMP_DIR = File.dirname(__FILE__) + "/../tmp/tail#{ENV['TEST_ENV_NUMBER']}"
-
-  CONFIG = config_element("ROOT", "", {
-                            "path" => "#{TMP_DIR}/tail.txt",
+  ROOT_CONFIG = config_element("ROOT", "", {
                             "tag" => "t1",
                             "rotate_wait" => "2s",
                             "refresh_interval" => "1s"
-                          })
-  COMMON_CONFIG = CONFIG + config_element("", "", { "pos_file" => "#{TMP_DIR}/tail.pos" })
+                               })
+
+  def base_config
+    ROOT_CONFIG + config_element("", "", { "path" => "#{@tmp_dir}/tail.txt" })
+  end
+
+  def common_config
+    base_config + config_element("", "", { "pos_file" => "#{@tmp_dir}/tail.pos" })
+  end
+
+  def common_follow_inode_config
+    config_element("ROOT", "", {
+                     "path" => "#{@tmp_dir}/tail.txt*",
+                     "pos_file" => "#{@tmp_dir}/tail.pos",
+                     "tag" => "t1",
+                     "refresh_interval" => "1s",
+                     "read_from_head" => "true",
+                     "format" => "none",
+                     "rotate_wait" => "1s",
+                     "follow_inodes" => "true"
+                   })
+  end
+
   CONFIG_READ_FROM_HEAD = config_element("", "", { "read_from_head" => true })
   CONFIG_DISABLE_WATCH_TIMER = config_element("", "", { "enable_watch_timer" => false })
   CONFIG_DISABLE_STAT_WATCHER = config_element("", "", { "enable_stat_watcher" => false })
   CONFIG_OPEN_ON_EVERY_UPDATE = config_element("", "", { "open_on_every_update" => true })
-  COMMON_FOLLOW_INODE_CONFIG = config_element("ROOT", "", {
-                                                "path" => "#{TMP_DIR}/tail.txt*",
-                                                "pos_file" => "#{TMP_DIR}/tail.pos",
-                                                "tag" => "t1",
-                                                "refresh_interval" => "1s",
-                                                "read_from_head" => "true",
-                                                "format" => "none",
-                                                "rotate_wait" => "1s",
-                                                "follow_inodes" => "true"
-                                              })
   SINGLE_LINE_CONFIG = config_element("", "", { "format" => "/(?<message>.*)/" })
   PARSE_SINGLE_LINE_CONFIG = config_element("", "", {}, [config_element("parse", "", { "@type" => "/(?<message>.*)/" })])
   MULTILINE_CONFIG = config_element(
@@ -144,21 +109,67 @@ class TailInputTest < Test::Unit::TestCase
                     })
     ])
 
+  EX_ROTATE_WAIT = 0
+  EX_FOLLOW_INODES = false
+
+  def ex_config
+    config_element("", "", {
+                     "tag" => "tail",
+                     "path" => "test/plugin/*/%Y/%m/%Y%m%d-%H%M%S.log,test/plugin/data/log/**/*.log",
+                     "format" => "none",
+                     "pos_file" => "#{@tmp_dir}/tail.pos",
+                     "read_from_head" => true,
+                     "refresh_interval" => 30,
+                     "rotate_wait" => "#{EX_ROTATE_WAIT}s",
+                     "follow_inodes" => "#{EX_FOLLOW_INODES}",
+                   })
+  end
+
+  def tailing_group_pattern
+    "/#{@tmp_dir}\/(?<podname>[a-z0-9]([-a-z0-9]*[a-z0-9])?(\/[a-z0-9]([-a-z0-9]*[a-z0-9])?)*)_(?<namespace>[^_]+)_(?<container>.+)-(?<docker_id>[a-z0-9]{6})\.log$/"
+  end
+
+  DEBUG_LOG_LEVEL = config_element("", "", {
+    "@log_level" => "debug"
+  })
+
+  def create_group_directive(pattern, rate_period, *rules)
+    config_element("", "", {}, [
+      config_element("group", "", {
+        "pattern" => pattern,
+        "rate_period" => rate_period
+      }, rules)
+    ])
+  end
+
+  def create_rule_directive(match_named_captures, limit)
+    params = {
+      "limit" => limit,
+      "match" => match_named_captures,
+    }
+
+    config_element("rule", "", params)
+  end
+
+  def create_path_element(path)
+    config_element("source", "", { "path" => "#{@tmp_dir}/#{path}" })
+  end
+
   def create_driver(conf = SINGLE_LINE_CONFIG, use_common_conf = true)
-    config = use_common_conf ? COMMON_CONFIG + conf : conf
+    config = use_common_conf ? common_config + conf : conf
     Fluent::Test::Driver::Input.new(Fluent::Plugin::TailInput).configure(config)
   end
 
   sub_test_case "configure" do
     test "plain single line" do
       d = create_driver
-      assert_equal ["#{TMP_DIR}/tail.txt"], d.instance.paths
-      assert_equal "t1", d.instance.tag
-      assert_equal 2, d.instance.rotate_wait
-      assert_equal "#{TMP_DIR}/tail.pos", d.instance.pos_file
-      assert_equal 1000, d.instance.read_lines_limit
-      assert_equal -1, d.instance.read_bytes_limit_per_second
-      assert_equal false, d.instance.ignore_repeated_permission_error
+      assert_equal(["#{@tmp_dir}/tail.txt"], d.instance.paths)
+      assert_equal("t1", d.instance.tag)
+      assert_equal(2, d.instance.rotate_wait)
+      assert_equal("#{@tmp_dir}/tail.pos", d.instance.pos_file)
+      assert_equal(1000, d.instance.read_lines_limit)
+      assert_equal(-1, d.instance.read_bytes_limit_per_second)
+      assert_equal(false, d.instance.ignore_repeated_permission_error)
       assert_nothing_raised do
         d.instance.have_read_capability?
       end
@@ -175,13 +186,13 @@ class TailInputTest < Test::Unit::TestCase
     test "multi paths with path_delimiter" do
       c = config_element("ROOT", "", { "path" => "tail.txt|test2|tmp,dev", "tag" => "t1", "path_delimiter" => "|" })
       d = create_driver(c + PARSE_SINGLE_LINE_CONFIG, false)
-      assert_equal ["tail.txt", "test2", "tmp,dev"], d.instance.paths
+      assert_equal(["tail.txt", "test2", "tmp,dev"], d.instance.paths)
     end
 
     test "multi paths with same path configured twice" do
       c = config_element("ROOT", "", { "path" => "test1.txt,test2.txt,test1.txt", "tag" => "t1", "path_delimiter" => "," })
       d = create_driver(c + PARSE_SINGLE_LINE_CONFIG, false)
-      assert_equal ["test2.txt","test1.txt"].sort, d.instance.paths.sort
+      assert_equal(["test2.txt","test1.txt"].sort, d.instance.paths.sort)
     end
 
     test "multi paths with invaid path_delimiter" do
@@ -193,7 +204,7 @@ class TailInputTest < Test::Unit::TestCase
 
     test "follow_inodes w/o pos file" do
       assert_raise(Fluent::ConfigError) do
-        create_driver(CONFIG + config_element('', '', {'follow_inodes' => 'true'}))
+        create_driver(base_config + config_element('', '', {'follow_inodes' => 'true'}))
       end
     end
 
@@ -223,7 +234,7 @@ class TailInputTest < Test::Unit::TestCase
       test "valid" do
         conf = SINGLE_LINE_CONFIG + config_element("", "", { "encoding" => "utf-8" })
         d = create_driver(conf)
-        assert_equal Encoding::UTF_8, d.instance.encoding
+        assert_equal(Encoding::UTF_8, d.instance.encoding)
       end
 
       test "invalid" do
@@ -263,12 +274,171 @@ class TailInputTest < Test::Unit::TestCase
     end
   end
 
+  sub_test_case "configure group" do
+    test "<rule> required" do
+      conf = create_group_directive('.', '1m') + SINGLE_LINE_CONFIG
+      assert_raise(Fluent::ConfigError) do
+        create_driver(conf)
+      end
+    end
+
+    test "valid configuration" do
+      rule1 = create_rule_directive({
+        "namespace"=> "/namespace-a/",
+        "podname"=> "/podname-[b|c]/"
+      }, 100)
+      rule2 = create_rule_directive({
+        "namespace"=> "/namespace-[d|e]/",
+        "podname"=> "/podname-f/",
+      }, 50)
+      rule3 = create_rule_directive({
+        "podname"=> "/podname-g/",
+      }, -1)
+      rule4 = create_rule_directive({
+        "namespace"=> "/namespace-h/",
+      }, 0)
+
+      conf = create_group_directive(tailing_group_pattern, '1m', rule1, rule2, rule3, rule4) + SINGLE_LINE_CONFIG
+      assert_nothing_raised do
+        create_driver(conf)
+      end
+    end
+
+    test "limit should be greater than DEFAULT_LIMIT (-1)" do
+      rule1 = create_rule_directive({
+        "namespace"=> "/namespace-a/",
+        "podname"=> "/podname-[b|c]/",
+      }, -100)
+      rule2 = create_rule_directive({
+        "namespace"=> "/namespace-[d|e]/",
+        "podname"=> "/podname-f/",
+      }, 50)
+      conf = create_group_directive(tailing_group_pattern, '1m', rule1, rule2) + SINGLE_LINE_CONFIG
+      assert_raise(RuntimeError) do
+        create_driver(conf)
+      end
+    end
+  end
+
+  sub_test_case "group rules line limit resolution" do
+    test "valid" do
+      rule1 = create_rule_directive({
+        "namespace"=> "/namespace-a/",
+        "podname"=> "/podname-[b|c]/",
+      }, 50)
+      rule2 = create_rule_directive({
+        "podname"=> "/podname-[b|c]/",
+      }, 400)
+      rule3 = create_rule_directive({
+        "namespace"=> "/namespace-a/",
+      }, 100)
+
+      conf = create_group_directive(tailing_group_pattern, '1m', rule3, rule1, rule2) + SINGLE_LINE_CONFIG
+      assert_nothing_raised do
+        d = create_driver(conf)
+        instance = d.instance
+
+        metadata = {
+          "namespace"=> "namespace-a",
+          "podname"=> "podname-b",
+        }
+        assert_equal(50, instance.find_group(metadata).limit)
+
+        metadata = {
+          "namespace" => "namespace-a",
+          "podname" => "podname-c",
+        }
+        assert_equal(50, instance.find_group(metadata).limit)
+
+        metadata = {
+          "namespace" => "namespace-a",
+          "podname" => "podname-d",
+        }
+        assert_equal(100, instance.find_group(metadata).limit)
+
+        metadata = {
+          "namespace" => "namespace-f",
+          "podname" => "podname-b",
+        }
+        assert_equal(400, instance.find_group(metadata).limit)
+
+        metadata = {
+          "podname" => "podname-c",
+        }
+        assert_equal(400, instance.find_group(metadata).limit)
+
+        assert_equal(-1, instance.find_group({}).limit)
+      end
+    end
+  end
+
+  sub_test_case "files should be placed in groups" do
+    test "invalid regex pattern places files in default group" do
+      rule1 = create_rule_directive({}, 100) ## limits default groups
+      conf = ROOT_CONFIG + DEBUG_LOG_LEVEL + create_group_directive(tailing_group_pattern, '1m', rule1) + create_path_element("test*.txt") + SINGLE_LINE_CONFIG
+
+      d = create_driver(conf, false)
+      File.open("#{@tmp_dir}/test1.txt", 'w')
+      File.open("#{@tmp_dir}/test2.txt", 'w')
+      File.open("#{@tmp_dir}/test3.txt", 'w')
+
+      d.run do
+        ## checking default group_watcher's paths
+        instance = d.instance
+        key = instance.default_group_key
+
+        assert_equal(3, instance.log.logs.count{|a| a.match?("Cannot find group from metadata, Adding file in the default group\n")})
+        assert_equal(3, instance.group_watchers[key].size)
+        assert_true(instance.group_watchers[key].include? File.join(@tmp_dir, 'test1.txt'))
+        assert_true(instance.group_watchers[key].include? File.join(@tmp_dir, 'test2.txt'))
+        assert_true(instance.group_watchers[key].include? File.join(@tmp_dir, 'test3.txt'))
+      end
+    end
+
+    test "valid regex pattern places file in their respective groups" do
+      rule1 = create_rule_directive({
+        "namespace"=> "/test-namespace1/",
+        "podname"=> "/test-podname1/",
+      }, 100)
+      rule2 = create_rule_directive({
+        "namespace"=> "/test-namespace1/",
+      }, 200)
+      rule3 = create_rule_directive({
+        "podname"=> "/test-podname2/",
+      }, 300)
+      rule4 = create_rule_directive({}, 400)
+
+      path_element = create_path_element("test-podname*.log")
+
+      conf = ROOT_CONFIG + create_group_directive(tailing_group_pattern, '1m', rule4, rule3, rule2, rule1) + path_element + SINGLE_LINE_CONFIG
+      d = create_driver(conf, false)
+
+      file1 = File.join(@tmp_dir, "test-podname1_test-namespace1_test-container-15fabq.log")
+      file2 = File.join(@tmp_dir, "test-podname3_test-namespace1_test-container-15fabq.log")
+      file3 = File.join(@tmp_dir, "test-podname2_test-namespace2_test-container-15fabq.log")
+      file4 = File.join(@tmp_dir, "test-podname4_test-namespace3_test-container-15fabq.log")
+
+      d.run do
+        File.open(file1, 'w')
+        File.open(file2, 'w')
+        File.open(file3, 'w')
+        File.open(file4, 'w')
+
+        instance = d.instance
+        assert_equal(100, instance.find_group_from_metadata(file1).limit)
+        assert_equal(200, instance.find_group_from_metadata(file2).limit)
+        assert_equal(300, instance.find_group_from_metadata(file3).limit)
+        assert_equal(400, instance.find_group_from_metadata(file4).limit)
+      end
+    end
+  end
+
   sub_test_case "singleline" do
     data(flat: SINGLE_LINE_CONFIG,
          parse: PARSE_SINGLE_LINE_CONFIG)
     def test_emit(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -276,7 +446,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config)
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3\ntest4"
         }
       end
@@ -292,11 +462,11 @@ class TailInputTest < Test::Unit::TestCase
 
     def test_emit_with_emit_unmatched_lines_true
       config = config_element("", "", { "format" => "/^(?<message>test.*)/", "emit_unmatched_lines" => true })
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test line 1"
           f.puts "test line 2"
           f.puts "bad line 1"
@@ -328,7 +498,7 @@ class TailInputTest < Test::Unit::TestCase
       msg = 'test' * 2000 # in_tail reads 8192 bytes at once.
 
       d.run(expect_emits: num_events, timeout: 2) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts msg
           f.puts msg
         }
@@ -343,7 +513,7 @@ class TailInputTest < Test::Unit::TestCase
 
     sub_test_case "log throttling per file" do
       teardown do
-        cleanup_file("#{TMP_DIR}/tail.txt")
+        cleanup_file("#{@tmp_dir}/tail.txt")
       end
 
       sub_test_case "reads_bytes_per_second w/o throttled" do
@@ -374,7 +544,7 @@ class TailInputTest < Test::Unit::TestCase
 
           d = create_driver(config)
           d.run(expect_emits: 2) do
-            File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+            File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
               100.times do
                 f.puts msg
               end
@@ -397,7 +567,7 @@ class TailInputTest < Test::Unit::TestCase
           start_time = Fluent::Clock.now
           d = create_driver(config)
           d.run(expect_emits: 2) do
-            File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+            File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
               8000.times do
                 f.puts msg
               end
@@ -436,7 +606,7 @@ class TailInputTest < Test::Unit::TestCase
             io_handler
           end
 
-          File.open("#{TMP_DIR}/tail.txt", "ab") do |f|
+          File.open("#{@tmp_dir}/tail.txt", "ab") do |f|
             100.times do
               f.puts msg
             end
@@ -446,7 +616,7 @@ class TailInputTest < Test::Unit::TestCase
           d.run do
             start_time = Fluent::Clock.now
             while Fluent::Clock.now - start_time < 0.8 do
-              File.open("#{TMP_DIR}/tail.txt", "ab") do |f|
+              File.open("#{@tmp_dir}/tail.txt", "ab") do |f|
                 f.puts msg
                 f.flush
               end
@@ -464,7 +634,7 @@ class TailInputTest < Test::Unit::TestCase
           num_lines = 1024 * 3
           msg = "08bytes"
 
-          File.open("#{TMP_DIR}/tail.txt", "wb") do |f|
+          File.open("#{@tmp_dir}/tail.txt", "wb") do |f|
             f.write("#{msg}\n" * num_lines)
           end
 
@@ -481,8 +651,8 @@ class TailInputTest < Test::Unit::TestCase
           d.run(timeout: 10) do
             while d.events.size < num_lines do
               if d.events.size > 0 && !rotated
-                cleanup_file("#{TMP_DIR}/tail.txt")
-                FileUtils.touch("#{TMP_DIR}/tail.txt")
+                cleanup_file("#{@tmp_dir}/tail.txt")
+                FileUtils.touch("#{@tmp_dir}/tail.txt")
                 rotated = true
               end
               sleep 0.3
@@ -500,7 +670,7 @@ class TailInputTest < Test::Unit::TestCase
           num_lines = 1024 * 2
           msg = "08bytes"
 
-          File.open("#{TMP_DIR}/tail.txt", "wb") do |f|
+          File.open("#{@tmp_dir}/tail.txt", "wb") do |f|
             f.write("#{msg}\n" * num_lines)
           end
 
@@ -527,8 +697,8 @@ class TailInputTest < Test::Unit::TestCase
           d.run(timeout: 10) do
             until detached do
               if d.events.size > 0 && !rotated
-                cleanup_file("#{TMP_DIR}/tail.txt")
-                FileUtils.touch("#{TMP_DIR}/tail.txt")
+                cleanup_file("#{@tmp_dir}/tail.txt")
+                FileUtils.touch("#{@tmp_dir}/tail.txt")
                 rotated = true
               end
               sleep 0.3
@@ -548,7 +718,7 @@ class TailInputTest < Test::Unit::TestCase
          parse: CONFIG_READ_FROM_HEAD + PARSE_SINGLE_LINE_CONFIG)
     def test_emit_with_read_from_head(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -556,7 +726,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config)
 
       d.run(expect_emits: 2) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -574,7 +744,7 @@ class TailInputTest < Test::Unit::TestCase
          parse: CONFIG_DISABLE_WATCH_TIMER + PARSE_SINGLE_LINE_CONFIG)
     def test_emit_without_watch_timer(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -582,7 +752,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config)
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -601,12 +771,12 @@ class TailInputTest < Test::Unit::TestCase
       omit "need inotify" unless Fluent.linux?
 
       config = config_element("ROOT", "", {
-                                "path" => "#{TMP_DIR}/tail*.txt",
+                                "path" => "#{@tmp_dir}/tail*.txt",
                                 "tag" => "t1",
                               })
       config = config + CONFIG_DISABLE_WATCH_TIMER + SINGLE_LINE_CONFIG
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -614,7 +784,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config, false)
 
       d.run(expect_emits: 1, timeout: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -632,7 +802,7 @@ class TailInputTest < Test::Unit::TestCase
          parse: CONFIG_DISABLE_STAT_WATCHER + PARSE_SINGLE_LINE_CONFIG)
     def test_emit_with_disable_stat_watcher(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -640,7 +810,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config)
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -656,7 +826,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(SINGLE_LINE_CONFIG)
 
       d.run(expect_emits: 1, timeout: 3) do
-        File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
           f.puts "test1\ntest2\n"
         }
       end
@@ -682,11 +852,14 @@ class TailInputTest < Test::Unit::TestCase
 
     def setup
       omit "NTFS doesn't support UNIX like permissions" if Fluent.windows?
+      super
       # Store default permission
       @default_permission = system_config.instance_variable_get(:@file_permission)
     end
 
     def teardown
+      return if Fluent.windows?
+      super
       # Restore default permission
       system_config.instance_variable_set(:@file_permission, @default_permission)
     end
@@ -700,7 +873,7 @@ class TailInputTest < Test::Unit::TestCase
       system_conf = parse_system(CONFIG_SYSTEM)
       sc = Fluent::SystemConfig.new(system_conf)
       Fluent::Engine.init(sc)
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -708,7 +881,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -728,16 +901,18 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   sub_test_case "rotate file" do
+    def create_driver(conf = SINGLE_LINE_CONFIG)
+      config = common_config + conf
+      Fluent::Test::Driver::Input.new(Fluent::Plugin::TailInput).configure(config)
+    end
+
     data(flat: SINGLE_LINE_CONFIG,
          parse: PARSE_SINGLE_LINE_CONFIG)
     def test_rotate_file(data)
       config = data
       events = sub_test_rotate_file(config, expect_emits: 2)
-      assert_equal(4, events.length)
-      assert_equal({"message" => "test3"}, events[0][2])
-      assert_equal({"message" => "test4"}, events[1][2])
-      assert_equal({"message" => "test5"}, events[2][2])
-      assert_equal({"message" => "test6"}, events[3][2])
+      assert_equal(3.upto(6).collect { |i| {"message" => "test#{i}"} },
+                   events.collect { |event| event[2] })
     end
 
     data(flat: CONFIG_READ_FROM_HEAD + SINGLE_LINE_CONFIG,
@@ -745,13 +920,8 @@ class TailInputTest < Test::Unit::TestCase
     def test_rotate_file_with_read_from_head(data)
       config = data
       events = sub_test_rotate_file(config, expect_records: 6)
-      assert_equal(6, events.length)
-      assert_equal({"message" => "test1"}, events[0][2])
-      assert_equal({"message" => "test2"}, events[1][2])
-      assert_equal({"message" => "test3"}, events[2][2])
-      assert_equal({"message" => "test4"}, events[3][2])
-      assert_equal({"message" => "test5"}, events[4][2])
-      assert_equal({"message" => "test6"}, events[5][2])
+      assert_equal(1.upto(6).collect { |i| {"message" => "test#{i}"} },
+                   events.collect { |event| event[2] })
     end
 
     data(flat: CONFIG_OPEN_ON_EVERY_UPDATE + CONFIG_READ_FROM_HEAD + SINGLE_LINE_CONFIG,
@@ -759,13 +929,8 @@ class TailInputTest < Test::Unit::TestCase
     def test_rotate_file_with_open_on_every_update(data)
       config = data
       events = sub_test_rotate_file(config, expect_records: 6)
-      assert_equal(6, events.length)
-      assert_equal({"message" => "test1"}, events[0][2])
-      assert_equal({"message" => "test2"}, events[1][2])
-      assert_equal({"message" => "test3"}, events[2][2])
-      assert_equal({"message" => "test4"}, events[3][2])
-      assert_equal({"message" => "test5"}, events[4][2])
-      assert_equal({"message" => "test6"}, events[5][2])
+      assert_equal(1.upto(6).collect { |i| {"message" => "test#{i}"} },
+                   events.collect { |event| event[2] })
     end
 
     data(flat: SINGLE_LINE_CONFIG,
@@ -773,26 +938,21 @@ class TailInputTest < Test::Unit::TestCase
     def test_rotate_file_with_write_old(data)
       config = data
       events = sub_test_rotate_file(config, expect_emits: 3) { |rotated_file|
-        File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+        File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
         rotated_file.puts "test7"
         rotated_file.puts "test8"
         rotated_file.flush
 
         sleep 1
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "test5"
           f.puts "test6"
         }
       }
       # This test sometimes fails and it shows a potential bug of in_tail
       # https://github.com/fluent/fluentd/issues/1434
-      assert_equal(6, events.length)
-      assert_equal({"message" => "test3"}, events[0][2])
-      assert_equal({"message" => "test4"}, events[1][2])
-      assert_equal({"message" => "test7"}, events[2][2])
-      assert_equal({"message" => "test8"}, events[3][2])
-      assert_equal({"message" => "test5"}, events[4][2])
-      assert_equal({"message" => "test6"}, events[5][2])
+      assert_equal([3, 4, 7, 8, 5, 6].collect { |i| {"message" => "test#{i}"} },
+                   events.collect { |event| event[2] })
     end
 
     data(flat: SINGLE_LINE_CONFIG,
@@ -804,21 +964,19 @@ class TailInputTest < Test::Unit::TestCase
         rotated_file.puts "test8"
         rotated_file.flush
       }
-      assert_equal(4, events.length)
-      assert_equal({"message" => "test3"}, events[0][2])
-      assert_equal({"message" => "test4"}, events[1][2])
-      assert_equal({"message" => "test7"}, events[2][2])
-      assert_equal({"message" => "test8"}, events[3][2])
+      assert_equal([3, 4, 7, 8].collect { |i| {"message" => "test#{i}"} },
+                   events.collect { |event| event[2] })
     end
 
-    def sub_test_rotate_file(config = nil, expect_emits: nil, expect_records: nil, timeout: nil)
-      file = Fluent::FileWrapper.open("#{TMP_DIR}/tail.txt", "wb")
+    def sub_test_rotate_file(config = nil, expect_emits: nil, expect_records: nil, timeout: 5)
+      file = Fluent::FileWrapper.open("#{@tmp_dir}/tail.txt", "wb")
       file.puts "test1"
       file.puts "test2"
       file.flush
 
       d = create_driver(config)
       d.run(expect_emits: expect_emits, expect_records: expect_records, timeout: timeout) do
+        sleep(0.1) while d.instance.instance_variable_get(:@startup)
         size = d.emit_count
         file.puts "test3"
         file.puts "test4"
@@ -828,18 +986,18 @@ class TailInputTest < Test::Unit::TestCase
 
         if Fluent.windows?
           file.close
-          FileUtils.mv("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail2.txt", force: true)
-          file = File.open("#{TMP_DIR}/tail.txt", "ab")
+          FileUtils.mv("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail2.txt", force: true)
+          file = Fluent::FileWrapper.open("#{@tmp_dir}/tail2.txt", "ab")
         else
-          FileUtils.mv("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail2.txt")
+          FileUtils.mv("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail2.txt")
         end
         if block_given?
           yield file
         else
-          File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+          File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
           sleep 1
 
-          File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+          File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
             f.puts "test5"
             f.puts "test6"
           }
@@ -853,10 +1011,8 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   def test_truncate_file
-    omit "Permission denied error happen on Windows. Need fix" if Fluent.windows?
-
     config = SINGLE_LINE_CONFIG
-    File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+    File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
       f.puts "test1"
       f.puts "test2"
       f.flush
@@ -865,12 +1021,18 @@ class TailInputTest < Test::Unit::TestCase
     d = create_driver(config)
 
     d.run(expect_emits: 2) do
-      File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
         f.puts "test3\ntest4"
         f.flush
       }
       waiting(2) { sleep 0.1 until d.events.length == 2 }
-      File.truncate("#{TMP_DIR}/tail.txt", 6)
+      if Fluent.windows?
+        Fluent::FileWrapper.open("#{@tmp_dir}/tail.txt", "wb") { |f|
+          f.puts("test1");
+        }
+      else
+        File.truncate("#{@tmp_dir}/tail.txt", 6)
+      end
     end
 
     expected = {
@@ -889,10 +1051,8 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   def test_move_truncate_move_back
-    omit "Permission denied error happen on Windows. Need fix" if Fluent.windows?
-
     config = SINGLE_LINE_CONFIG
-    File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+    File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
       f.puts "test1"
       f.puts "test2"
     }
@@ -901,17 +1061,23 @@ class TailInputTest < Test::Unit::TestCase
 
     d.run(expect_emits: 1) do
       if Fluent.windows?
-        FileUtils.mv("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail2.txt", force: true)
+        FileUtils.mv("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail2.txt", force: true)
       else
-        FileUtils.mv("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail2.txt")
+        FileUtils.mv("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail2.txt")
       end
       sleep(1)
-      File.truncate("#{TMP_DIR}/tail2.txt", 6)
+      if Fluent.windows?
+        Fluent::FileWrapper.open("#{@tmp_dir}/tail2.txt", "wb") { |f|
+          f.puts("test1");
+        }
+      else
+        File.truncate("#{@tmp_dir}/tail2.txt", 6)
+      end
       sleep(1)
       if Fluent.windows?
-        FileUtils.mv("#{TMP_DIR}/tail2.txt", "#{TMP_DIR}/tail.txt", force: true)
+        FileUtils.mv("#{@tmp_dir}/tail2.txt", "#{@tmp_dir}/tail.txt", force: true)
       else
-        FileUtils.mv("#{TMP_DIR}/tail2.txt", "#{TMP_DIR}/tail.txt")
+        FileUtils.mv("#{@tmp_dir}/tail2.txt", "#{@tmp_dir}/tail.txt")
       end
     end
 
@@ -923,17 +1089,17 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   def test_lf
-    File.open("#{TMP_DIR}/tail.txt", "wb") {|f| }
+    File.open("#{@tmp_dir}/tail.txt", "wb") {|f| }
 
     d = create_driver
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
         f.print "test3"
       }
       sleep 1
 
-      File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
         f.puts "test4"
       }
     end
@@ -944,12 +1110,12 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   def test_whitespace
-    File.open("#{TMP_DIR}/tail.txt", "wb") {|f| }
+    File.open("#{@tmp_dir}/tail.txt", "wb") {|f| }
 
     d = create_driver
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
         f.puts "    "		# 4 spaces
         f.puts "    4 spaces"
         f.puts "4 spaces    "
@@ -980,7 +1146,7 @@ class TailInputTest < Test::Unit::TestCase
     d = create_driver(CONFIG_READ_FROM_HEAD + encoding_config)
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test"
       }
     end
@@ -1002,7 +1168,7 @@ class TailInputTest < Test::Unit::TestCase
     utf8_message = cp932_message.encode(Encoding::UTF_8)
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "w:cp932") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "w:cp932") {|f|
         f.puts cp932_message
       }
     end
@@ -1025,7 +1191,7 @@ class TailInputTest < Test::Unit::TestCase
     utf8_message = utf16_message.encode(Encoding::UTF_8).strip
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "w:utf-16le") { |f|
+      File.open("#{@tmp_dir}/tail.txt", "w:utf-16le") { |f|
         f.write utf16_message
       }
     end
@@ -1046,7 +1212,7 @@ class TailInputTest < Test::Unit::TestCase
     d = create_driver(conf)
 
     d.run(expect_emits: 1) do
-      File.open("#{TMP_DIR}/tail.txt", "w") { |f|
+      File.open("#{@tmp_dir}/tail.txt", "w") { |f|
         f.write "te\x86st\n"
       }
     end
@@ -1061,11 +1227,11 @@ class TailInputTest < Test::Unit::TestCase
          parse: PARSE_MULTILINE_CONFIG)
     def test_multiline(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "f test1"
           f.puts "s test2"
           f.puts "f test3"
@@ -1089,11 +1255,11 @@ class TailInputTest < Test::Unit::TestCase
          parse: PARSE_MULTILINE_CONFIG)
     def test_multiline_with_emit_unmatched_lines_true(data)
       config = data + config_element("", "", { "emit_unmatched_lines" => true })
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "f test1"
           f.puts "s test2"
           f.puts "f test3"
@@ -1119,11 +1285,11 @@ class TailInputTest < Test::Unit::TestCase
       parse: PARSE_MULTILINE_CONFIG_WITH_NEWLINE)
     def test_multiline_with_emit_unmatched_lines2(data)
       config = data + config_element("", "", { "emit_unmatched_lines" => true })
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       d = create_driver(config)
       d.run(expect_emits: 0, timeout: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "s test0"
           f.puts "f test1"
           f.puts "f test2"
@@ -1145,15 +1311,15 @@ class TailInputTest < Test::Unit::TestCase
     data(flat: MULTILINE_CONFIG,
          parse: PARSE_MULTILINE_CONFIG)
     def test_multiline_with_flush_interval(data)
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       config = data + config_element("", "", { "multiline_flush_interval" => "2s" })
       d = create_driver(config)
 
-      assert_equal 2, d.instance.multiline_flush_interval
+      assert_equal(2, d.instance.multiline_flush_interval)
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "f test1"
           f.puts "s test2"
           f.puts "f test3"
@@ -1188,7 +1354,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(config + encoding_config)
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "wb") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "wb") { |f|
           f.puts "s test"
         }
       end
@@ -1211,7 +1377,7 @@ class TailInputTest < Test::Unit::TestCase
       cp932_message = "s \x82\xCD\x82\xEB\x81\x5B\x82\xED\x81\x5B\x82\xE9\x82\xC7".force_encoding(Encoding::CP932)
       utf8_message = "\x82\xCD\x82\xEB\x81\x5B\x82\xED\x81\x5B\x82\xE9\x82\xC7".encode(Encoding::UTF_8, Encoding::CP932)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "w:cp932") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "w:cp932") { |f|
           f.puts cp932_message
         }
       end
@@ -1243,11 +1409,11 @@ class TailInputTest < Test::Unit::TestCase
         )
     def test_multiline_with_multiple_formats(data)
       config = data
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "f test1"
           f.puts "s test2"
           f.puts "f test3"
@@ -1283,7 +1449,7 @@ class TailInputTest < Test::Unit::TestCase
            ])
         )
     def test_multilinelog_with_multiple_paths(data)
-      files = ["#{TMP_DIR}/tail1.txt", "#{TMP_DIR}/tail2.txt"]
+      files = ["#{@tmp_dir}/tail1.txt", "#{@tmp_dir}/tail2.txt"]
       files.each { |file| File.open(file, "wb") { |f| } }
 
       config = data + config_element("", "", {
@@ -1328,12 +1494,12 @@ class TailInputTest < Test::Unit::TestCase
            ])
         )
     def test_multiline_without_firstline(data)
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       config = data
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "foo 1"
           f.puts "bar 1"
           f.puts "baz 1"
@@ -1353,35 +1519,22 @@ class TailInputTest < Test::Unit::TestCase
   sub_test_case "path" do
     # * path test
     # TODO: Clean up tests
-    EX_ROTATE_WAIT = 0
-    EX_FOLLOW_INODES = false
-
-    EX_CONFIG = config_element("", "", {
-                                 "tag" => "tail",
-                                 "path" => "test/plugin/*/%Y/%m/%Y%m%d-%H%M%S.log,test/plugin/data/log/**/*.log",
-                                 "format" => "none",
-                                 "pos_file" => "#{TMP_DIR}/tail.pos",
-                                 "read_from_head" => true,
-                                 "refresh_interval" => 30,
-                                 "rotate_wait" => "#{EX_ROTATE_WAIT}s",
-                                 "follow_inodes" => "#{EX_FOLLOW_INODES}",
-                               })
     def test_expand_paths
       ex_paths = [
         create_target_info('test/plugin/data/2010/01/20100102-030405.log'),
         create_target_info('test/plugin/data/log/foo/bar.log'),
         create_target_info('test/plugin/data/log/test.log')
       ]
-      plugin = create_driver(EX_CONFIG, false).instance
+      plugin = create_driver(ex_config, false).instance
       flexstub(Time) do |timeclass|
         timeclass.should_receive(:now).with_no_args.and_return(Time.new(2010, 1, 2, 3, 4, 5))
-        assert_equal ex_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+        assert_equal(ex_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
       end
 
       # Test exclusion
-      exclude_config = EX_CONFIG + config_element("", "", { "exclude_path" => %Q(["#{ex_paths.last.path}"]) })
+      exclude_config = ex_config + config_element("", "", { "exclude_path" => %Q(["#{ex_paths.last.path}"]) })
       plugin = create_driver(exclude_config, false).instance
-      assert_equal ex_paths - [ex_paths.last], plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+      assert_equal(ex_paths - [ex_paths.last], plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
     end
 
     def test_expand_paths_with_duplicate_configuration
@@ -1389,10 +1542,10 @@ class TailInputTest < Test::Unit::TestCase
         create_target_info('test/plugin/data/log/foo/bar.log'),
         create_target_info('test/plugin/data/log/test.log')
       ]
-      duplicate_config = EX_CONFIG.dup
+      duplicate_config = ex_config.dup
       duplicate_config["path"]="test/plugin/data/log/**/*.log, test/plugin/data/log/**/*.log"
-      plugin = create_driver(EX_CONFIG, false).instance
-      assert_equal expanded_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+      plugin = create_driver(ex_config, false).instance
+      assert_equal(expanded_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
     end
 
     def test_expand_paths_with_timezone
@@ -1402,7 +1555,7 @@ class TailInputTest < Test::Unit::TestCase
         create_target_info('test/plugin/data/log/test.log')
       ]
       ['Asia/Taipei', '+08'].each do |tz_type|
-        taipei_config = EX_CONFIG + config_element("", "", {"path_timezone" => tz_type})
+        taipei_config = ex_config + config_element("", "", {"path_timezone" => tz_type})
         plugin = create_driver(taipei_config, false).instance
 
         # Test exclude
@@ -1414,8 +1567,8 @@ class TailInputTest < Test::Unit::TestCase
             # env : 2010-01-01 19:04:05 (UTC), tail path : 2010-01-02 03:04:05 (Asia/Taipei)
             timeclass.should_receive(:now).with_no_args.and_return(Time.new(2010, 1, 1, 19, 4, 5))
 
-            assert_equal ex_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
-            assert_equal ex_paths - [ex_paths.first], exclude_plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+            assert_equal(ex_paths, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
+            assert_equal(ex_paths - [ex_paths.first], exclude_plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
           end
         end
       end
@@ -1433,35 +1586,44 @@ class TailInputTest < Test::Unit::TestCase
         "tag" => "tail",
         "path" => "test/plugin/data/log/**/*",
         "format" => "none",
-        "pos_file" => "#{TMP_DIR}/tail.pos"
+        "pos_file" => "#{@tmp_dir}/tail.pos"
       })
 
       plugin = create_driver(config, false).instance
-      assert_equal expected_files, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+      assert_equal(expected_files, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
     end
 
     def test_unwatched_files_should_be_removed
       config = config_element("", "", {
                                 "tag" => "tail",
-                                "path" => "#{TMP_DIR}/*.txt",
+                                "path" => "#{@tmp_dir}/*.txt",
                                 "format" => "none",
-                                "pos_file" => "#{TMP_DIR}/tail.pos",
+                                "pos_file" => "#{@tmp_dir}/tail.pos",
                                 "read_from_head" => true,
                                 "refresh_interval" => 1,
                               })
       d = create_driver(config, false)
       d.end_if { d.instance.instance_variable_get(:@tails).keys.size >= 1 }
       d.run(expect_emits: 1, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f| f.puts "test3\n" }
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f| f.puts "test3\n" }
       end
 
-      cleanup_directory(TMP_DIR)
-      waiting(20) { sleep 0.1 until Dir.glob("#{TMP_DIR}/*.txt").size == 0 } # Ensure file is deleted on Windows
+      cleanup_directory(@tmp_dir)
+      waiting(20) { sleep 0.1 until Dir.glob("#{@tmp_dir}/*.txt").size == 0 } # Ensure file is deleted on Windows
       waiting(5) { sleep 0.1 until d.instance.instance_variable_get(:@tails).keys.size <= 0 }
 
-      assert_equal 0, d.instance.instance_variable_get(:@tails).keys.size
-
-      d.instance_shutdown
+      assert_equal(
+        {
+          files: [],
+          tails: []
+        },
+        {
+          files: Dir.glob("#{@tmp_dir}/*.txt"),
+          tails: d.instance.instance_variable_get(:@tails).keys
+        }
+      )
+    ensure
+      d.instance_shutdown if d && d.instance
     end
 
     def count_timer_object
@@ -1511,55 +1673,55 @@ class TailInputTest < Test::Unit::TestCase
                             "rotate_wait" => "2s"
                        }) + PARSE_SINGLE_LINE_CONFIG, false)
 
-      assert_equal readable_paths, d.instance.expand_paths.length
-      assert_equal result, d.instance.have_read_capability?
+      assert_equal(readable_paths, d.instance.expand_paths.length)
+      assert_equal(result, d.instance.have_read_capability?)
     end
   end
 
   def test_pos_file_dir_creation
     config = config_element("", "", {
       "tag" => "tail",
-      "path" => "#{TMP_DIR}/*.txt",
+      "path" => "#{@tmp_dir}/*.txt",
       "format" => "none",
-      "pos_file" => "#{TMP_DIR}/pos/tail.pos",
+      "pos_file" => "#{@tmp_dir}/pos/tail.pos",
       "read_from_head" => true,
       "refresh_interval" => 1
     })
 
-    assert_path_not_exist("#{TMP_DIR}/pos")
+    assert_path_not_exist("#{@tmp_dir}/pos")
     d = create_driver(config, false)
     d.run
-    assert_path_exist("#{TMP_DIR}/pos")
-    assert_equal '755', File.stat("#{TMP_DIR}/pos").mode.to_s(8)[-3, 3]
+    assert_path_exist("#{@tmp_dir}/pos")
+    assert_equal('755', File.stat("#{@tmp_dir}/pos").mode.to_s(8)[-3, 3])
   ensure
-    cleanup_directory(TMP_DIR)
+    cleanup_directory(@tmp_dir)
   end
 
   def test_pos_file_dir_creation_with_system_dir_permission
     config = config_element("", "", {
       "tag" => "tail",
-      "path" => "#{TMP_DIR}/*.txt",
+      "path" => "#{@tmp_dir}/*.txt",
       "format" => "none",
-      "pos_file" => "#{TMP_DIR}/pos/tail.pos",
+      "pos_file" => "#{@tmp_dir}/pos/tail.pos",
       "read_from_head" => true,
       "refresh_interval" => 1
     })
 
-    assert_path_not_exist("#{TMP_DIR}/pos")
+    assert_path_not_exist("#{@tmp_dir}/pos")
 
     Fluent::SystemConfig.overwrite_system_config({ "dir_permission" => "744" }) do
       d = create_driver(config, false)
       d.run
     end
 
-    assert_path_exist("#{TMP_DIR}/pos")
+    assert_path_exist("#{@tmp_dir}/pos")
     if Fluent.windows?
-      assert_equal '755', File.stat("#{TMP_DIR}/pos").mode.to_s(8)[-3, 3]
+      assert_equal('755', File.stat("#{@tmp_dir}/pos").mode.to_s(8)[-3, 3])
     else
-      assert_equal '744', File.stat("#{TMP_DIR}/pos").mode.to_s(8)[-3, 3]
+      assert_equal('744', File.stat("#{@tmp_dir}/pos").mode.to_s(8)[-3, 3])
     end
   ensure
-    cleanup_directory(TMP_DIR)
+    cleanup_directory(@tmp_dir)
   end
 
   def test_z_refresh_watchers
@@ -1568,7 +1730,7 @@ class TailInputTest < Test::Unit::TestCase
       create_target_info('test/plugin/data/log/foo/bar.log'),
       create_target_info('test/plugin/data/log/test.log'),
     ]
-    plugin = create_driver(EX_CONFIG, false).instance
+    plugin = create_driver(ex_config, false).instance
     sio = StringIO.new
     plugin.instance_eval do
       @pf = Fluent::Plugin::TailInput::PositionFile.load(sio, EX_FOLLOW_INODES, {}, logger: $log)
@@ -1591,8 +1753,7 @@ class TailInputTest < Test::Unit::TestCase
     end
 
     path = 'test/plugin/data/2010/01/20100102-030405.log'
-    target_info = Fluent::Plugin::TailInput::TargetInfo.new(path, Fluent::FileWrapper.stat(path).ino)
-    mock.proxy(plugin).detach_watcher_after_rotate_wait(plugin.instance_variable_get(:@tails)[target_info], target_info.ino)
+    mock.proxy(plugin).detach_watcher_after_rotate_wait(plugin.instance_variable_get(:@tails)[path], Fluent::FileWrapper.stat(path).ino)
 
     Timecop.freeze(2010, 1, 2, 3, 4, 6) do
       path = "test/plugin/data/2010/01/20100102-030406.log"
@@ -1612,9 +1773,9 @@ class TailInputTest < Test::Unit::TestCase
     test 'type of pos_file_compaction_interval is time' do
       tail = {
         "tag" => "tail",
-        "path" => "#{TMP_DIR}/*.txt",
+        "path" => "#{@tmp_dir}/*.txt",
         "format" => "none",
-        "pos_file" => "#{TMP_DIR}/pos/tail.pos",
+        "pos_file" => "#{@tmp_dir}/pos/tail.pos",
         "refresh_interval" => 1,
         "read_from_head" => true,
         'pos_file_compaction_interval' => '24h',
@@ -1631,7 +1792,7 @@ class TailInputTest < Test::Unit::TestCase
     DummyWatcher = Struct.new("DummyWatcher", :tag)
 
     def test_tag
-      d = create_driver(EX_CONFIG, false)
+      d = create_driver(ex_config, false)
       d.run {}
       plugin = d.instance
       mock(plugin.router).emit_stream('tail', anything).once
@@ -1715,13 +1876,13 @@ class TailInputTest < Test::Unit::TestCase
     test 'max_line_size' do |(label, size)|
       config = config_element("", "", {
                                 "tag" => "max_line_size",
-                                "path" => "#{TMP_DIR}/with_long_lines.txt",
+                                "path" => "#{@tmp_dir}/with_long_lines.txt",
                                 "format" => "none",
                                 "read_from_head" => true,
                                 "max_line_size" => label,
                                 "log_level" => "debug"
                               })
-      File.open("#{TMP_DIR}/with_long_lines.txt", "w+") do |f|
+      File.open("#{@tmp_dir}/with_long_lines.txt", "w+") do |f|
         f.puts "foo"
         f.puts "x" * size # 'x' * size + \n > @max_line_size
         f.puts "bar"
@@ -1747,7 +1908,7 @@ class TailInputTest < Test::Unit::TestCase
   # Ensure that no fatal exception is raised when a file is missing and that
   # files that do exist are still tailed as expected.
   def test_missing_file
-    File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+    File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
       f.puts "test1"
       f.puts "test2"
     }
@@ -1756,16 +1917,16 @@ class TailInputTest < Test::Unit::TestCase
     # since their interactions with the filesystem differ.
     config1 = config_element("", "", {
                                "tag" => "t1",
-                               "path" => "#{TMP_DIR}/non_existent_file.txt,#{TMP_DIR}/tail.txt",
+                               "path" => "#{@tmp_dir}/non_existent_file.txt,#{@tmp_dir}/tail.txt",
                                "format" => "none",
                                "rotate_wait" => "2s",
-                               "pos_file" => "#{TMP_DIR}/tail.pos"
+                               "pos_file" => "#{@tmp_dir}/tail.pos"
                              })
     config2 = config1 + config_element("", "", { "read_from_head" => true })
     [config1, config2].each do |config|
       d = create_driver(config, false)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -1781,19 +1942,19 @@ class TailInputTest < Test::Unit::TestCase
 
   sub_test_case 'inode_processing' do
     def test_should_delete_file_pos_entry_for_non_existing_file_with_follow_inodes
-      config = COMMON_FOLLOW_INODE_CONFIG
+      config = common_follow_inode_config
 
-      path = "#{TMP_DIR}/tail.txt"
+      path = "#{@tmp_dir}/tail.txt"
       ino = 1
       pos = 1234
-      File.open("#{TMP_DIR}/tail.pos", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.pos", "wb") {|f|
         f.puts ("%s\t%016x\t%016x\n" % [path, pos, ino])
       }
 
       d = create_driver(config, false)
       d.run
 
-      pos_file = File.open("#{TMP_DIR}/tail.pos", "r")
+      pos_file = File.open("#{@tmp_dir}/tail.pos", "r")
       pos_file.pos = 0
 
       assert_raise(EOFError) do
@@ -1802,21 +1963,21 @@ class TailInputTest < Test::Unit::TestCase
     end
 
     def test_should_write_latest_offset_after_rotate_wait
-      config = COMMON_FOLLOW_INODE_CONFIG
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      config = common_follow_inode_config
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
 
       d = create_driver(config, false)
       d.run(expect_emits: 2, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
-        FileUtils.move("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail.txt" + "1")
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        FileUtils.move("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail.txt" + "1")
         sleep 1
-        File.open("#{TMP_DIR}/tail.txt" + "1", "ab") {|f| f.puts "test4\n"}
+        File.open("#{@tmp_dir}/tail.txt" + "1", "ab") {|f| f.puts "test4\n"}
       end
 
-      pos_file = File.open("#{TMP_DIR}/tail.pos", "r")
+      pos_file = File.open("#{@tmp_dir}/tail.pos", "r")
       pos_file.pos = 0
       line_parts = /^([^\t]+)\t([0-9a-fA-F]+)\t([0-9a-fA-F]+)/.match(pos_file.readline)
       waiting(5) {
@@ -1833,16 +1994,16 @@ class TailInputTest < Test::Unit::TestCase
     def test_should_remove_deleted_file
       config = config_element("", "", {"format" => "none"})
 
-      path = "#{TMP_DIR}/tail.txt"
+      path = "#{@tmp_dir}/tail.txt"
       ino = 1
       pos = 1234
-      File.open("#{TMP_DIR}/tail.pos", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.pos", "wb") {|f|
         f.puts ("%s\t%016x\t%016x\n" % [path, pos, ino])
       }
 
       d = create_driver(config)
       d.run do
-        pos_file = File.open("#{TMP_DIR}/tail.pos", "r")
+        pos_file = File.open("#{@tmp_dir}/tail.pos", "r")
         pos_file.pos = 0
         assert_equal([], pos_file.readlines)
       end
@@ -1850,8 +2011,8 @@ class TailInputTest < Test::Unit::TestCase
 
     def test_should_mark_file_unwatched_after_limit_recently_modified_and_rotate_wait
       config = config_element("ROOT", "", {
-          "path" => "#{TMP_DIR}/tail.txt*",
-          "pos_file" => "#{TMP_DIR}/tail.pos",
+          "path" => "#{@tmp_dir}/tail.txt*",
+          "pos_file" => "#{@tmp_dir}/tail.pos",
           "tag" => "t1",
           "rotate_wait" => "1s",
           "refresh_interval" => "1s",
@@ -1863,14 +2024,14 @@ class TailInputTest < Test::Unit::TestCase
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
-      target_info = create_target_info("#{TMP_DIR}/tail.txt")
+      target_info = create_target_info("#{@tmp_dir}/tail.txt")
 
       d.run(expect_emits: 1, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
       end
 
 
@@ -1888,8 +2049,8 @@ class TailInputTest < Test::Unit::TestCase
 
     def test_should_read_from_head_on_file_renaming_with_star_in_pattern
       config = config_element("ROOT", "", {
-          "path" => "#{TMP_DIR}/tail.txt*",
-          "pos_file" => "#{TMP_DIR}/tail.pos",
+          "path" => "#{@tmp_dir}/tail.txt*",
+          "pos_file" => "#{@tmp_dir}/tail.pos",
           "tag" => "t1",
           "rotate_wait" => "10s",
           "refresh_interval" => "1s",
@@ -1901,14 +2062,14 @@ class TailInputTest < Test::Unit::TestCase
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
 
       d.run(expect_emits: 2, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
-        FileUtils.move("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail.txt1")
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        FileUtils.move("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail.txt1")
       end
 
       events = d.events
@@ -1917,20 +2078,20 @@ class TailInputTest < Test::Unit::TestCase
     end
 
     def test_should_not_read_from_head_on_rotation_when_watching_inodes
-      config = COMMON_FOLLOW_INODE_CONFIG
+      config = common_follow_inode_config
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
 
       d.run(expect_emits: 1, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
       end
 
-      FileUtils.move("#{TMP_DIR}/tail.txt", "#{TMP_DIR}/tail.txt1")
+      FileUtils.move("#{@tmp_dir}/tail.txt", "#{@tmp_dir}/tail.txt1")
       Timecop.travel(Time.now + 10) do
         sleep 2
         events = d.events
@@ -1941,23 +2102,23 @@ class TailInputTest < Test::Unit::TestCase
     end
 
     def test_should_mark_file_unwatched_if_same_name_file_created_with_different_inode
-      config = COMMON_FOLLOW_INODE_CONFIG
+      config = common_follow_inode_config
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
-      target_info = create_target_info("#{TMP_DIR}/tail.txt")
+      target_info = create_target_info("#{@tmp_dir}/tail.txt")
 
       d.run(expect_emits: 2, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
-        cleanup_file("#{TMP_DIR}/tail.txt")
-        File.open("#{TMP_DIR}/tail.txt", "wb") {|f| f.puts "test4\n"}
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        cleanup_file("#{@tmp_dir}/tail.txt")
+        File.open("#{@tmp_dir}/tail.txt", "wb") {|f| f.puts "test4\n"}
       end
 
-      new_target_info = create_target_info("#{TMP_DIR}/tail.txt")
+      new_target_info = create_target_info("#{@tmp_dir}/tail.txt")
 
       pos_file = d.instance.instance_variable_get(:@pf)
 
@@ -1973,7 +2134,7 @@ class TailInputTest < Test::Unit::TestCase
 
     def test_should_close_watcher_after_rotate_wait
       now = Time.now
-      config = COMMON_FOLLOW_INODE_CONFIG + config_element('', '', {"rotate_wait" => "1s", "limit_recently_modified" => "1s"})
+      config = common_follow_inode_config + config_element('', '', {"rotate_wait" => "1s", "limit_recently_modified" => "1s"})
 
       d = create_driver(config, false)
       d.instance.instance_eval do
@@ -1986,52 +2147,52 @@ class TailInputTest < Test::Unit::TestCase
         @metrics = Fluent::Plugin::TailInput::MetricsInfo.new(opened_file_metrics, closed_file_metrics, rotated_file_metrics)
       end
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
-      target_info = create_target_info("#{TMP_DIR}/tail.txt")
+      target_info = create_target_info("#{@tmp_dir}/tail.txt")
       mock.proxy(Fluent::Plugin::TailInput::TailWatcher).new(target_info, anything, anything, true, true, anything, nil, anything, anything).once
       d.run(shutdown: false)
-      assert d.instance.instance_variable_get(:@tails)[target_info]
+      assert d.instance.instance_variable_get(:@tails)[target_info.path]
 
       Timecop.travel(now + 10) do
         d.instance.instance_eval do
-          sleep 0.1 until @tails[target_info] == nil
+          sleep 0.1 until @tails[target_info.path] == nil
         end
-        assert_nil d.instance.instance_variable_get(:@tails)[target_info]
+        assert_nil d.instance.instance_variable_get(:@tails)[target_info.path]
       end
       d.instance_shutdown
     end
 
     def test_should_create_new_watcher_for_new_file_with_same_name
       now = Time.now
-      config = COMMON_FOLLOW_INODE_CONFIG + config_element('', '', {"limit_recently_modified" => "2s"})
+      config = common_follow_inode_config + config_element('', '', {"limit_recently_modified" => "2s"})
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
-      path_ino = create_target_info("#{TMP_DIR}/tail.txt")
+      path_ino = create_target_info("#{@tmp_dir}/tail.txt")
 
       d.run(expect_emits: 1, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
       end
 
-      cleanup_file("#{TMP_DIR}/tail.txt")
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      cleanup_file("#{@tmp_dir}/tail.txt")
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test3"
         f.puts "test4"
       }
-      new_path_ino = create_target_info("#{TMP_DIR}/tail.txt")
+      new_path_ino = create_target_info("#{@tmp_dir}/tail.txt")
 
       Timecop.travel(now + 10) do
         sleep 3
         d.instance.instance_eval do
-          @tails[path_ino] == nil
-          @tails[new_path_ino] != nil
+          @tails[path_ino.path] == nil
+          @tails[new_path_ino.path] != nil
         end
       end
 
@@ -2043,19 +2204,19 @@ class TailInputTest < Test::Unit::TestCase
     end
 
     def test_truncate_file_with_follow_inodes
-      config = COMMON_FOLLOW_INODE_CONFIG
+      config = common_follow_inode_config
 
       d = create_driver(config, false)
 
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
 
       d.run(expect_emits: 3, shutdown: false) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f| f.puts "test3\n"}
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f| f.puts "test3\n"}
         sleep 2
-        File.open("#{TMP_DIR}/tail.txt", "w+b") {|f| f.puts "test4\n"}
+        File.open("#{@tmp_dir}/tail.txt", "w+b") {|f| f.puts "test4\n"}
       end
 
       events = d.events
@@ -2069,15 +2230,15 @@ class TailInputTest < Test::Unit::TestCase
 
     # issue #3464
     def test_should_replace_target_info
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1\n"
       }
-      target_info = create_target_info("#{TMP_DIR}/tail.txt")
+      target_info = create_target_info("#{@tmp_dir}/tail.txt")
       inodes = []
 
       config = config_element("ROOT", "", {
-                                "path" => "#{TMP_DIR}/tail.txt*",
-                                "pos_file" => "#{TMP_DIR}/tail.pos",
+                                "path" => "#{@tmp_dir}/tail.txt*",
+                                "pos_file" => "#{@tmp_dir}/tail.pos",
                                 "tag" => "t1",
                                 "refresh_interval" => "60s",
                                 "read_from_head" => "true",
@@ -2090,21 +2251,21 @@ class TailInputTest < Test::Unit::TestCase
         while d.events.size < 1 do
           sleep 0.1
         end
-        inodes = d.instance.instance_variable_get(:@tails).keys.collect do |key|
-          key.ino
+        inodes = d.instance.instance_variable_get(:@tails).values.collect do |tw|
+          tw.ino
         end
         assert_equal([target_info.ino], inodes)
 
-        cleanup_file("#{TMP_DIR}/tail.txt")
-        File.open("#{TMP_DIR}/tail.txt", "wb") {|f| f.puts "test2\n"}
+        cleanup_file("#{@tmp_dir}/tail.txt")
+        File.open("#{@tmp_dir}/tail.txt", "wb") {|f| f.puts "test2\n"}
 
         while d.events.size < 2 do
           sleep 0.1
         end
-        inodes = d.instance.instance_variable_get(:@tails).keys.collect do |key|
-          key.ino
+        inodes = d.instance.instance_variable_get(:@tails).values.collect do |tw|
+          tw.ino
         end
-        new_target_info = create_target_info("#{TMP_DIR}/tail.txt")
+        new_target_info = create_target_info("#{@tmp_dir}/tail.txt")
         assert_not_equal(target_info.ino, new_target_info.ino)
         assert_equal([new_target_info.ino], inodes)
       end
@@ -2113,7 +2274,7 @@ class TailInputTest < Test::Unit::TestCase
 
   sub_test_case "tail_path" do
     def test_tail_path_with_singleline
-      File.open("#{TMP_DIR}/tail.txt", "wb") {|f|
+      File.open("#{@tmp_dir}/tail.txt", "wb") {|f|
         f.puts "test1"
         f.puts "test2"
       }
@@ -2121,7 +2282,7 @@ class TailInputTest < Test::Unit::TestCase
       d = create_driver(SINGLE_LINE_CONFIG + config_element("", "", { "path_key" => "path" }))
 
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") {|f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") {|f|
           f.puts "test3"
           f.puts "test4"
         }
@@ -2130,12 +2291,12 @@ class TailInputTest < Test::Unit::TestCase
       events = d.events
       assert_equal(true, events.length > 0)
       events.each do |emit|
-        assert_equal("#{TMP_DIR}/tail.txt", emit[2]["path"])
+        assert_equal("#{@tmp_dir}/tail.txt", emit[2]["path"])
       end
     end
 
     def test_tail_path_with_multiline_with_firstline
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       config = config_element("", "", {
                                 "path_key" => "path",
@@ -2145,7 +2306,7 @@ class TailInputTest < Test::Unit::TestCase
                               })
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "f test1"
           f.puts "s test2"
           f.puts "f test3"
@@ -2160,12 +2321,12 @@ class TailInputTest < Test::Unit::TestCase
       events = d.events
       assert_equal(4, events.length)
       events.each do |emit|
-        assert_equal("#{TMP_DIR}/tail.txt", emit[2]["path"])
+        assert_equal("#{@tmp_dir}/tail.txt", emit[2]["path"])
       end
     end
 
     def test_tail_path_with_multiline_without_firstline
-      File.open("#{TMP_DIR}/tail.txt", "wb") { |f| }
+      File.open("#{@tmp_dir}/tail.txt", "wb") { |f| }
 
       config = config_element("", "", {
                                 "path_key" => "path",
@@ -2176,7 +2337,7 @@ class TailInputTest < Test::Unit::TestCase
                               })
       d = create_driver(config)
       d.run(expect_emits: 1) do
-        File.open("#{TMP_DIR}/tail.txt", "ab") { |f|
+        File.open("#{@tmp_dir}/tail.txt", "ab") { |f|
           f.puts "foo 1"
           f.puts "bar 1"
           f.puts "baz 1"
@@ -2186,7 +2347,7 @@ class TailInputTest < Test::Unit::TestCase
       events = d.events
       assert(events.length > 0)
       events.each do |emit|
-        assert_equal("#{TMP_DIR}/tail.txt", emit[2]["path"])
+        assert_equal("#{@tmp_dir}/tail.txt", emit[2]["path"])
       end
     end
 
@@ -2194,7 +2355,7 @@ class TailInputTest < Test::Unit::TestCase
       if ENV["APPVEYOR"] && Fluent.windows?
         omit "This testcase is unstable on AppVeyor."
       end
-      files = ["#{TMP_DIR}/tail1.txt", "#{TMP_DIR}/tail2.txt"]
+      files = ["#{@tmp_dir}/tail1.txt", "#{@tmp_dir}/tail2.txt"]
       files.each { |file| File.open(file, "wb") { |f| } }
 
       config = config_element("", "", {
@@ -2228,30 +2389,30 @@ class TailInputTest < Test::Unit::TestCase
 
   def test_limit_recently_modified
     now = Time.new(2010, 1, 2, 3, 4, 5)
-    FileUtils.touch("#{TMP_DIR}/tail_unwatch.txt", mtime: (now - 3601))
-    FileUtils.touch("#{TMP_DIR}/tail_watch1.txt", mtime: (now - 3600))
-    FileUtils.touch("#{TMP_DIR}/tail_watch2.txt", mtime: now)
+    FileUtils.touch("#{@tmp_dir}/tail_unwatch.txt", mtime: (now - 3601))
+    FileUtils.touch("#{@tmp_dir}/tail_watch1.txt", mtime: (now - 3600))
+    FileUtils.touch("#{@tmp_dir}/tail_watch2.txt", mtime: now)
 
     config = config_element('', '', {
       'tag' => 'tail',
-      'path' => "#{TMP_DIR}/*.txt",
+      'path' => "#{@tmp_dir}/*.txt",
       'format' => 'none',
       'limit_recently_modified' => '3600s'
     })
 
     expected_files = [
-      create_target_info("#{TMP_DIR}/tail_watch1.txt"),
-      create_target_info("#{TMP_DIR}/tail_watch2.txt")
+      create_target_info("#{@tmp_dir}/tail_watch1.txt"),
+      create_target_info("#{@tmp_dir}/tail_watch2.txt")
     ]
 
     Timecop.freeze(now) do
       plugin = create_driver(config, false).instance
-      assert_equal expected_files, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path }
+      assert_equal(expected_files, plugin.expand_paths.values.sort_by { |path_ino| path_ino.path })
     end
   end
 
   def test_skip_refresh_on_startup
-    FileUtils.touch("#{TMP_DIR}/tail.txt")
+    FileUtils.touch("#{@tmp_dir}/tail.txt")
     config = config_element('', '', {
                               'format' => 'none',
                               'refresh_interval' => 1,
@@ -2259,37 +2420,42 @@ class TailInputTest < Test::Unit::TestCase
                             })
     d = create_driver(config)
     d.run(shutdown: false) {}
-    assert_equal 0, d.instance.instance_variable_get(:@tails).keys.size
+    assert_equal(0, d.instance.instance_variable_get(:@tails).keys.size)
     # detect a file at first execution of in_tail_refresh_watchers timer
     waiting(5) { sleep 0.1 until d.instance.instance_variable_get(:@tails).keys.size == 1 }
     d.instance_shutdown
   end
 
   def test_ENOENT_error_after_setup_watcher
-    path = "#{TMP_DIR}/tail.txt"
+    path = "#{@tmp_dir}/tail.txt"
     FileUtils.touch(path)
     config = config_element('', '', {
                               'format' => 'none',
                             })
     d = create_driver(config)
-    mock.proxy(d.instance).setup_watcher(anything, anything) do |tw|
-      cleanup_file(path)
-      tw
-    end
+    file_deleted = false
+    mock.proxy(d.instance).existence_path do |hash|
+      unless file_deleted
+        cleanup_file(path)
+        file_deleted = true
+      end
+      hash
+    end.twice
     assert_nothing_raised do
       d.run(shutdown: false) {}
     end
-    assert($log.out.logs.any?{|log| log.include?("stat() for #{path} failed with Errno::ENOENT. Drop tail watcher for now.\n") })
+    assert($log.out.logs.any?{|log| log.include?("stat() for #{path} failed. Continuing without tailing it.\n") },
+           $log.out.logs.join("\n"))
   ensure
     d.instance_shutdown if d && d.instance
   end
 
   def test_EACCES_error_after_setup_watcher
     omit "Cannot test with root user" if Process::UID.eid == 0
-    path = "#{TMP_DIR}/noaccess/tail.txt"
+    path = "#{@tmp_dir}/noaccess/tail.txt"
     begin
-      FileUtils.mkdir_p("#{TMP_DIR}/noaccess")
-      FileUtils.chmod(0755, "#{TMP_DIR}/noaccess")
+      FileUtils.mkdir_p("#{@tmp_dir}/noaccess")
+      FileUtils.chmod(0755, "#{@tmp_dir}/noaccess")
       FileUtils.touch(path)
       config = config_element('', '', {
                                 'tag' => "tail",
@@ -2297,25 +2463,26 @@ class TailInputTest < Test::Unit::TestCase
                                 'format' => 'none',
                               })
       d = create_driver(config, false)
-      mock.proxy(d.instance).setup_watcher(anything, anything) do |tw|
-        FileUtils.chmod(0000, "#{TMP_DIR}/noaccess")
-        tw
-      end
+      mock.proxy(d.instance).existence_path do |hash|
+        FileUtils.chmod(0000, "#{@tmp_dir}/noaccess")
+        hash
+      end.twice
       assert_nothing_raised do
         d.run(shutdown: false) {}
       end
-      assert($log.out.logs.any?{|log| log.include?("stat() for #{path} failed with Errno::EACCES. Drop tail watcher for now.\n") })
+      assert($log.out.logs.any?{|log| log.include?("stat() for #{path} failed. Continuing without tailing it.\n") },
+             $log.out.logs.join("\n"))
     end
   ensure
     d.instance_shutdown if d && d.instance
-    if File.exist?("#{TMP_DIR}/noaccess")
-      FileUtils.chmod(0755, "#{TMP_DIR}/noaccess")
-      FileUtils.rm_rf("#{TMP_DIR}/noaccess")
+    if File.exist?("#{@tmp_dir}/noaccess")
+      FileUtils.chmod(0755, "#{@tmp_dir}/noaccess")
+      FileUtils.rm_rf("#{@tmp_dir}/noaccess")
     end
   end unless Fluent.windows?
 
   def test_EACCES
-    path = "#{TMP_DIR}/tail.txt"
+    path = "#{@tmp_dir}/tail.txt"
     FileUtils.touch(path)
     config = config_element('', '', {
                               'format' => 'none',
@@ -2333,8 +2500,8 @@ class TailInputTest < Test::Unit::TestCase
   end
 
   def test_shutdown_timeout
-    path = "#{TMP_DIR}/tail.txt"
-    File.open("#{TMP_DIR}/tail.txt", "wb") do |f|
+    File.open("#{@tmp_dir}/tail.txt", "wb") do |f|
+      # Should be large enough to take too long time to consume
       (1024 * 1024 * 5).times do
         f.puts "{\"test\":\"fizzbuzz\"}"
       end
@@ -2346,18 +2513,127 @@ class TailInputTest < Test::Unit::TestCase
                               'format' => 'json',
                               'skip_refresh_on_startup' => true,
                             })
+    shutdown_start_time = 0
+
     d = create_driver(config)
     mock.proxy(d.instance).io_handler(anything, anything) do |io_handler|
+      mock.proxy(io_handler).ready_to_shutdown(anything) do
+        shutdown_start_time = Fluent::Clock.now
+      end
       io_handler.shutdown_timeout = 0.5
       io_handler
     end
 
-    start_time = Fluent::Clock.now
     assert_nothing_raised do
       d.run(expect_emits: 1)
     end
 
-    elapsed = Fluent::Clock.now - start_time
-    assert_true(elapsed > 0.5 && elapsed < 2.5)
+    elapsed = Fluent::Clock.now - shutdown_start_time
+    assert_true(elapsed > 0.5 && elapsed < 2.0,
+                "elapsed time: #{elapsed}")
+  end
+
+  sub_test_case "throttling logs at in_tail level" do
+    data("file test1.log no_limit 5120 text: msg" => ["test1.log", 5120, "msg"],
+         "file test2.log no_limit 1024 text: test" => ["test2.log", 1024, "test"])
+    def test_lines_collected_with_no_throttling(data)
+      file, num_lines, msg = data
+
+      pattern = "/^#{@tmp_dir}\/(?<file>.+)\.log$/"
+      rule = create_rule_directive({
+        "file" => "/test.*/",
+      }, -1)
+      group = create_group_directive(pattern, "1s", rule)
+      path_element = create_path_element(file)
+
+      conf = ROOT_CONFIG + group + path_element + CONFIG_READ_FROM_HEAD + SINGLE_LINE_CONFIG
+
+      File.open("#{@tmp_dir}/#{file}", 'wb') do |f|
+        num_lines.times do
+          f.puts "#{msg}\n"
+        end
+      end
+
+
+      d = create_driver(conf, false)
+      d.run(timeout: 3) do
+        start_time = Fluent::Clock.now
+
+        assert_equal(num_lines, d.record_count)
+        assert_equal({ "message" => msg }, d.events[0][2])
+
+        prev_count = d.record_count
+        sleep(0.1) while d.emit_count < 1
+        assert_true(Fluent::Clock.now - start_time < 2)
+        ## after waiting for 1 (+ jitter) secs, limit will reset
+        ## Plugin will start reading but it will encounter EOF Error
+        ## since no logs are left to be read
+        ## Hence, d.record_count = prev_count
+        tail_watcher_interval = 1.0 # hard coded value in in_tail
+        safety_ratio = 1.2
+        jitter = tail_watcher_interval * safety_ratio
+        sleep(1.0 + jitter)
+        assert_equal(0, d.record_count - prev_count)
+      end
+    end
+
+    test "lines collected with throttling" do
+      file = "podname1_namespace12_container-123456.log"
+      limit = 1000
+      rate_period = 2
+      num_lines = 3000
+      msg = "a" * 8190 # Total size = 8190 bytes + 2 (\n) bytes
+
+      rule = create_rule_directive({
+        "namespace"=> "/namespace.+/",
+        "podname"=> "/podname.+/",
+      }, limit)
+      path_element = create_path_element(file)
+      conf = ROOT_CONFIG + create_group_directive(tailing_group_pattern, "#{rate_period}s", rule) + path_element + SINGLE_LINE_CONFIG + CONFIG_READ_FROM_HEAD
+
+      d = create_driver(conf, false)
+      file_path = "#{@tmp_dir}/#{file}"
+
+      File.open(file_path, 'wb') do |f|
+        num_lines.times do
+          f.puts msg
+        end
+      end
+
+      d.run(timeout: 15) do
+        sleep_interval = 0.1
+        tail_watcher_interval = 1.0 # hard coded value in in_tail
+        safety_ratio = 1.2
+        lower_jitter = sleep_interval * safety_ratio
+        upper_jitter = (tail_watcher_interval + sleep_interval) * safety_ratio
+        lower_interval = rate_period - lower_jitter
+        upper_interval = rate_period + upper_jitter
+
+        emit_count = 0
+        prev_count = 0
+
+        while emit_count < 3 do
+          start_time = Fluent::Clock.now
+          sleep(sleep_interval) while d.emit_count <= emit_count
+          elapsed_seconds = Fluent::Clock.now - start_time
+          if emit_count > 0
+            assert_true(elapsed_seconds > lower_interval && elapsed_seconds < upper_interval,
+                        "elapsed_seconds #{elapsed_seconds} is out of allowed range:\n" +
+                        "  lower: #{lower_interval} [sec]\n" +
+                        "  upper: #{upper_interval} [sec]")
+          end
+          assert_equal(limit, d.record_count - prev_count)
+          emit_count = d.emit_count
+          prev_count = d.record_count
+        end
+
+        ## When all the lines are read and rate_period seconds are over
+        ## limit will reset and since there are no more logs to be read,
+        ## number_lines_read will be 0
+        sleep upper_interval
+        gw = d.instance.find_group_from_metadata(file_path)
+        assert_equal(0, gw.current_paths[file_path].number_lines_read)
+      end
+    end
   end
 end