fix to wait force_flush to ensure that expected chunks are written

fluent · Oct 5, 2016 · b43fa56 · b43fa56
1 parent a7813a3
commit b43fa56
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/fluent/test/driver/output.rb b/lib/fluent/test/driver/output.rb
@@ -18,6 +18,7 @@
 require 'fluent/test/driver/event_feeder'
 
 require 'fluent/plugin/output'
+require 'timeout'
 
 module Fluent
   module Test
@@ -43,6 +44,7 @@ def run_actual(**kwargs, &block)
           super(**kwargs, &block)
           if @flush_buffer_at_cleanup
             @instance.force_flush
+            Timeout.timeout(10){ sleep 0.1 until @instance.buffer.queue.size == 0 }
           end
         end
 
@@ -52,6 +54,7 @@ def formatted
 
         def flush
           @instance.force_flush
+          Timeout.timeout(10){ sleep 0.1 until @instance.buffer.queue.size == 0 }
         end
 
         def instance_hook_after_started