lib/atp/flow.rb in atp-0.3.2 vs lib/atp/flow.rb in atp-0.3.3

@@ -77,9 +77,17 @@
         end
         if options.delete(:continue)
           options[:on_fail] ||= {}
           options[:on_fail][:continue] = true
         end
+        if f = options.delete(:flag_pass)
+          options[:on_pass] ||= {}
+          options[:on_pass][:set_run_flag] = f
+        end
+        if f = options.delete(:flag_fail)
+          options[:on_fail] ||= {}
+          options[:on_fail][:set_run_flag] = f
+        end
         builder.test(instance, options)
       end
       append(t) unless r
       t
     end