Fix test to pass with v0.14

fluent-plugins-nursery · Sep 15, 2017 · 00b427f · 00b427f
1 parent 2d4a705
commit 00b427f
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/test/helper.rb b/test/helper.rb
@@ -17,11 +17,18 @@
 
 class Test::Unit::TestCase
  def capture_log(log)
- tmp = log.out
- log.out = StringIO.new
- yield
- return log.out.string
- ensure
- log.out = tmp
+ if defined?(Fluent::Test::TestLogger) and log.is_a?(Fluent::Test::TestLogger) # v0.14
+ yield
+ log.out.logs.join("\n")
+ else
+ begin
+ tmp = log.out
+ log.out = StringIO.new
+ yield
+ return log.out.string
+ ensure
+ log.out = tmp
+ end
+ end
  end
 end