lib/fluent/test/output_test.rb in fluentd-0.12.19 vs lib/fluent/test/output_test.rb in fluentd-0.12.20

@@ -67,12 +67,12 @@
 
       def expect_format(str)
         (@expected_buffer ||= '') << str
       end
 
-      def run(&block)
+      def run(num_waits = 10, &block)
         result = nil
-        super {
+        super(num_waits) {
           es = ArrayEventStream.new(@entries)
           buffer = @instance.format_stream(@tag, es)
 
           block.call if block