Ensure blocks given to subscriber are properly yield

rails · May 17, 2012 · 313f292 · 313f292
1 parent 6057a1d
commit 313f292
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/activesupport/lib/active_support/log_subscriber/test_helper.rb b/activesupport/lib/active_support/log_subscriber/test_helper.rb
@@ -61,11 +61,12 @@ def initialize(level = DEBUG)
           @logged = Hash.new { |h,k| h[k] = [] }
         end
 
-        def debug
-        end
-
         def method_missing(level, message)
-          @logged[level] << message
+           if block_given?
+             @logged[level] << yield
+           else
+             @logged[level] << message
+           end
         end
 
         def logged(level)

diff --git a/activesupport/test/log_subscriber_test.rb b/activesupport/test/log_subscriber_test.rb
@@ -11,7 +11,7 @@ def some_event(event)
 
   def foo(event)
     debug "debug"
-    info "info"
+    info { "info" }
     warn "warn"
   end