lib/lucid/ast/step_result.rb in lucid-0.0.8 vs lib/lucid/ast/step_result.rb in lucid-0.0.9

@@ -7,12 +7,14 @@
       def initialize(keyword, step_match, multiline_arg, status, exception, source_indent, background, file_colon_line)
         @keyword, @step_match, @multiline_arg, @status, @exception, @source_indent, @background, @file_colon_line = keyword, step_match, multiline_arg, status, exception, source_indent, background, file_colon_line
       end
 
       def accept(visitor)
-        visitor.visit_step_name(@keyword, @step_match, @status, @source_indent, @background, @file_colon_line)
-        visitor.visit_multiline_arg(@multiline_arg) if @multiline_arg
-        visitor.visit_exception(@exception, @status) if @exception
+        visitor.step_result(self) do
+          visitor.visit_step_name(@keyword, @step_match, @status, @source_indent, @background, @file_colon_line)
+          visitor.visit_multiline_arg(@multiline_arg) if @multiline_arg
+          visitor.visit_exception(@exception, @status) if @exception
+        end
       end
 
       def args
         [@keyword, @step_match, @multiline_arg, @status, @exception, @source_indent, @background, @file_colon_line]
       end
\ No newline at end of file