lib/atp/ast/builder.rb in atp-0.4.3 vs lib/atp/ast/builder.rb in atp-0.5.0

@@ -234,10 +234,11 @@
         end
         if options[:set_run_flag] || options[:set_flag]
           children << set_run_flag(options[:set_run_flag] || options[:set_flag])
         end
         children << continue if options[:continue]
+        children << render(options[:render]) if options[:render]
         n(:on_fail, *children)
       end
 
       def on_pass(options = {})
         children = []
@@ -246,9 +247,10 @@
         end
         if options[:set_run_flag] || options[:set_flag]
           children << set_run_flag(options[:set_run_flag] || options[:set_flag])
         end
         children << continue if options[:continue]
+        children << render(options[:render]) if options[:render]
         n(:on_pass, *children)
       end
 
       def set_result(type, options = {})
         children = []