Buffer and flush messages, nested within example.

Author: ioquatix

lib/rspec/core/formatters/documentation_formatter.rb

@@ -12,6 +12,7 @@ class DocumentationFormatter < BaseTextFormatter
         def initialize(output)
           super
           @group_level = 0
+          @messages = []
         end
 
         def example_group_started(notification)
@@ -23,27 +24,39 @@ def example_group_started(notification)
 
         def example_group_finished(_notification)
           @group_level -= 1 if @group_level > 0
+          flush_messages
         end
 
         def example_passed(passed)
           output.puts passed_output(passed.example)
+          flush_messages
         end
 
         def example_pending(pending)
           output.puts pending_output(pending.example,
                                      pending.example.execution_result.pending_message)
+          flush_messages
         end
 
         def example_failed(failure)
           output.puts failure_output(failure.example)
+          flush_messages
         end
 
         def message(notification)
-          output.puts "#{current_indentation}#{notification.message}"
+          @messages << notification.message
         end
 
       private
 
+        def flush_messages
+          @messages.each do |message|
+            output.puts "#{current_indentation(1)}#{message}"
+          end
+
+          @messages.clear
+        end
+
         def passed_output(example)
           ConsoleCodes.wrap("#{current_indentation}#{example.description.strip}", :success)
         end
@@ -65,8 +78,8 @@ def next_failure_index
           @next_failure_index += 1
         end
 
-        def current_indentation
-          '  ' * @group_level
+        def current_indentation(offset = 0)
+          '  ' * (@group_level + offset)
         end
       end
     end

spec/rspec/core/formatters/documentation_formatter_spec.rb

@@ -89,8 +89,8 @@ def execution_result(values)
 
       expect(formatter_output.string).to eql("
 root
-  message
   example
+    message
 ")
     end